diff options
Diffstat (limited to 'Lib/test/test_binhex.py')
-rwxr-xr-x | Lib/test/test_binhex.py | 78 |
1 files changed, 38 insertions, 40 deletions
diff --git a/Lib/test/test_binhex.py b/Lib/test/test_binhex.py index a2b2a2c..c774200 100755 --- a/Lib/test/test_binhex.py +++ b/Lib/test/test_binhex.py @@ -2,46 +2,44 @@ """Test script for the binhex C module Uses the mechanism of the python binhex module - Roger E. Masse + Based on an original test by Roger E. Masse. """ import binhex +import os import tempfile -from test_support import verbose, TestSkipped - -def test(): - - try: - fname1 = tempfile.mktemp() - fname2 = tempfile.mktemp() - f = open(fname1, 'w') - except: - raise TestSkipped, "Cannot test binhex without a temp file" - - start = 'Jack is my hero' - f.write(start) - f.close() - - binhex.binhex(fname1, fname2) - if verbose: - print 'binhex' - - binhex.hexbin(fname2, fname1) - if verbose: - print 'hexbin' - - f = open(fname1, 'r') - finish = f.readline() - f.close() # on Windows an open file cannot be unlinked - - if start != finish: - print 'Error: binhex != hexbin' - elif verbose: - print 'binhex == hexbin' - - try: - import os - os.unlink(fname1) - os.unlink(fname2) - except: - pass -test() +import test_support +import unittest + + +class BinHexTestCase(unittest.TestCase): + + def setUp(self): + self.fname1 = tempfile.mktemp() + self.fname2 = tempfile.mktemp() + + def tearDown(self): + try: os.unlink(self.fname1) + except OSError: pass + + try: os.unlink(self.fname2) + except OSError: pass + + DATA = 'Jack is my hero' + + def test_binhex(self): + f = open(self.fname1, 'w') + f.write(self.DATA) + f.close() + + binhex.binhex(self.fname1, self.fname2) + + binhex.hexbin(self.fname2, self.fname1) + + f = open(self.fname1, 'r') + finish = f.readline() + f.close() + + self.assertEqual(self.DATA, finish) + + +test_support.run_unittest(BinHexTestCase) |