From ef84b8ffb253b0b3cd2ba2f5ca6b140ab5f04ba8 Mon Sep 17 00:00:00 2001 From: Thomas Waldmann Date: Sun, 27 Mar 2011 00:11:54 +0100 Subject: make tests work on python 2.5 --- test/test_argparse.py | 46 +++++++++++++++++++++------------------------- 1 file changed, 21 insertions(+), 25 deletions(-) diff --git a/test/test_argparse.py b/test/test_argparse.py index 4d1a6ec..dde6d16 100644 --- a/test/test_argparse.py +++ b/test/test_argparse.py @@ -15,7 +15,22 @@ from StringIO import StringIO class StdIOBuffer(StringIO): pass -from test import test_support + +# silence some warnings - these are expected +import warnings +warnings.filterwarnings( + action='ignore', + message='The "version" argument to ArgumentParser is deprecated.', + category=DeprecationWarning) +warnings.filterwarnings( + action='ignore', + message='The format_version method is deprecated', + category=DeprecationWarning) +warnings.filterwarnings( + action='ignore', + message='The print_version method is deprecated', + category=DeprecationWarning) + class TestCase(unittest.TestCase): @@ -28,15 +43,6 @@ class TestCase(unittest.TestCase): print(obj2) super(TestCase, self).assertEqual(obj1, obj2) - def setUp(self): - # The tests assume that line wrapping occurs at 80 columns, but this - # behaviour can be overridden by setting the COLUMNS environment - # variable. To ensure that this assumption is true, unset COLUMNS. - env = test_support.EnvironmentVarGuard() - env.unset("COLUMNS") - #this only works on python >= 2.7: - #self.addCleanup(env.__exit__) - class TempDirMixin(object): @@ -4247,8 +4253,11 @@ class TestEncoding(TestCase): def _test_module_encoding(self, path): path, _ = os.path.splitext(path) path += ".py" - with codecs.open(path, 'r', 'utf8') as f: + f = codecs.open(path, 'r', 'utf8') + try: f.read() + finally: + f.close def test_argparse_module_encoding(self): self._test_module_encoding(argparse.__file__) @@ -4332,19 +4341,6 @@ class TestImportStar(TestCase): ] self.assertEqual(sorted(items), sorted(argparse.__all__)) -def test_main(): - # silence warnings about version argument - these are expected - with test_support.check_warnings( - ('The "version" argument to ArgumentParser is deprecated.', - DeprecationWarning), - ('The (format|print)_version method is deprecated', - DeprecationWarning)): - test_support.run_unittest(__name__) - # Remove global references to avoid looking like we have refleaks. - RFile.seen = {} - WFile.seen = set() - - if __name__ == '__main__': - test_main() + unittest.main() -- cgit v1.2.1