Change the PyUnit-based tests to use the test_main() approach. This

allows using the tests with unittest.py as a script.  The tests will
still run when run as a script themselves.

diff --git a/Lib/test/test_binhex.py b/Lib/test/test_binhex.py
index c774200fd81c7ef5a13b7c8be24fa2d763abb04f..2580fb21e28372ed54b5699b15f66d50e19d8897 100755 (executable)
--- a/Lib/test/test_binhex.py
+++ b/Lib/test/test_binhex.py
@@ -42,4 +42,9 @@ class BinHexTestCase(unittest.TestCase):
         self.assertEqual(self.DATA, finish)
 
 
-test_support.run_unittest(BinHexTestCase)
+def test_main():
+    test_support.run_unittest(BinHexTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_binop.py b/Lib/test/test_binop.py
index c64e3a84ea9250bf912d5f449de2c1e97e00b8c4..976e7ffa106602e9e750456e055ea1de3a30c8b1 100644 (file)
--- a/Lib/test/test_binop.py
+++ b/Lib/test/test_binop.py
@@ -320,4 +320,9 @@ self.assertEqual(Rat(1, 2) * 3.0, 1.5)
 self.assertEqual(eval('1/2'), 0.5)
 """
 
-test_support.run_unittest(RatTestCase)
+def test_main():
+    test_support.run_unittest(RatTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_call.py b/Lib/test/test_call.py
index 529f15b190b1fc01cf8e1f59ab036317efee55f4..886bf18c2ba3f440b4cafbef457a24da63fc824c 100644 (file)
--- a/Lib/test/test_call.py
+++ b/Lib/test/test_call.py
@@ -122,4 +122,10 @@ class CFunctionCalls(unittest.TestCase):
     def test_oldargs1_2_kw(self):
         self.assertRaises(TypeError, {}.update, x=2, y=2)
 
-run_unittest(CFunctionCalls)
+
+def test_main():
+    run_unittest(CFunctionCalls)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_codecs.py b/Lib/test/test_codecs.py
index 6d9e31d672700f33906706af8b6a219650027fe9..28d84ce4a63c7e5a233fa8f7735727bf434e93aa 100644 (file)
--- a/Lib/test/test_codecs.py
+++ b/Lib/test/test_codecs.py
@@ -22,4 +22,10 @@ class UTF16Test(unittest.TestCase):
         f = reader(s)
         self.assertEquals(f.read(), u"spamspam")
 
-test_support.run_unittest(UTF16Test)
+
+def test_main():
+    test_support.run_unittest(UTF16Test)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_codeop.py b/Lib/test/test_codeop.py
index 7847fb3eef35e4812914740a934e0d093070a049..fbe4a318533d0081730d759a374fb26b3e35eb87 100644 (file)
--- a/Lib/test/test_codeop.py
+++ b/Lib/test/test_codeop.py
@@ -87,4 +87,10 @@ class CodeopTests(unittest.TestCase):
         self.assertNotEquals(compile_command("a = 1\n", "abc").co_filename,
                              compile("a = 1\n", "def", 'single').co_filename)
 
-run_unittest(CodeopTests)
+
+def test_main():
+    run_unittest(CodeopTests)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_commands.py b/Lib/test/test_commands.py
index 54ae676d77d996630e4bc16e4c883f2fe3626c64..77372c52433770382a3e0f1af206cae71b1d0f36 100644 (file)
--- a/Lib/test/test_commands.py
+++ b/Lib/test/test_commands.py
@@ -42,4 +42,10 @@ class CommandTests(unittest.TestCase):
 
         self.assert_(re.match(pat, getstatus("/bin/ls"), re.VERBOSE))
 
-run_unittest(CommandTests)
+
+def test_main():
+    run_unittest(CommandTests)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_copy_reg.py b/Lib/test/test_copy_reg.py
index 51ec60b5958176e0d974ff3debad0f8f19820042..0f5c96f10dfd5bddd16510544ca6e707b31b1e99 100644 (file)
--- a/Lib/test/test_copy_reg.py
+++ b/Lib/test/test_copy_reg.py
@@ -22,4 +22,9 @@ class CopyRegTestCase(unittest.TestCase):
                           type(1), int, "not a callable")
 
 
-test_support.run_unittest(CopyRegTestCase)
+def test_main():
+    test_support.run_unittest(CopyRegTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_dircache.py b/Lib/test/test_dircache.py
index 6d57dcfde5491ce1915aec4e05624199e8766b13..95dc2af43786dffe14e37336d2ca5d8fe2d85b33 100644 (file)
--- a/Lib/test/test_dircache.py
+++ b/Lib/test/test_dircache.py
@@ -65,4 +65,10 @@ class DircacheTests(unittest.TestCase):
         dircache.annotate(self.tempdir, lst)
         self.assertEquals(lst, ['A/', 'test2', 'test_nonexistent'])
 
-run_unittest(DircacheTests)
+
+def test_main():
+    run_unittest(DircacheTests)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_dospath.py b/Lib/test/test_dospath.py
index 74bdd4a12a9a344694b6f75a2251ddaf29eaa264..3cca1597ebd7b3b74797bf78535a6aab4f810f5b 100644 (file)
--- a/Lib/test/test_dospath.py
+++ b/Lib/test/test_dospath.py
@@ -53,4 +53,9 @@ class DOSPathTestCase(unittest.TestCase):
         self.assertEquals(splitdrive("c:"), ('c:', ''))
 
 
-test_support.run_unittest(DOSPathTestCase)
+def test_main():
+    test_support.run_unittest(DOSPathTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_fnmatch.py b/Lib/test/test_fnmatch.py
index 9f089267ff39f4b58e22f12581e474cdddb54ba3..0bd7d6a96cd442de9065d15ab2bffb88591a2cc6 100644 (file)
--- a/Lib/test/test_fnmatch.py
+++ b/Lib/test/test_fnmatch.py
@@ -38,4 +38,9 @@ class FnmatchTestCase(unittest.TestCase):
         check('\\', r'[!\]', 0)
 
 
-test_support.run_unittest(FnmatchTestCase)
+def test_main():
+    test_support.run_unittest(FnmatchTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_fpformat.py b/Lib/test/test_fpformat.py
index 9d20dc991fc7afce99bef9732dac32f5507679de..d4a5eb25ab1ff919fd1dae3e9b10b2eb9499f61f 100644 (file)
--- a/Lib/test/test_fpformat.py
+++ b/Lib/test/test_fpformat.py
@@ -66,4 +66,10 @@ class FpformatTest(unittest.TestCase):
         else:
             self.fail("No exception on non-numeric sci")
 
-run_unittest(FpformatTest)
+
+def test_main():
+    run_unittest(FpformatTest)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_glob.py b/Lib/test/test_glob.py
index 804ae1d68f4abeeeb92aed928f322a243d5b7daf..8cf6ecf47120d22b0b1a9c3c37c98798656682e0 100644 (file)
--- a/Lib/test/test_glob.py
+++ b/Lib/test/test_glob.py
@@ -106,4 +106,10 @@ class GlobTests(unittest.TestCase):
         eq(self.glob('?a?', '*F'), map(self.norm, [os.path.join('aaa', 'zzzF'),
                                                    os.path.join('aab', 'F')]))
 
-run_unittest(GlobTests)
+
+def test_main():
+    run_unittest(GlobTests)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_grp.py b/Lib/test/test_grp.py
index 1edb4173d4f5051a3b9c1e47c45a79971edbcb0c..a16a922956afeabf3eeeb40721ed89d9e8d04e99 100755 (executable)
--- a/Lib/test/test_grp.py
+++ b/Lib/test/test_grp.py
@@ -19,4 +19,9 @@ class GroupDatabaseTestCase(unittest.TestCase):
         entry = grp.getgrnam(self.groups[0][0])
 
 
-test_support.run_unittest(GroupDatabaseTestCase)
+def test_main():
+    test_support.run_unittest(GroupDatabaseTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_hash.py b/Lib/test/test_hash.py
index 20ec9a4649acd1d3f845ef01ccc859e825aad221..f8a1523cc689c65b6689215e0d9625849d6d0ddd 100644 (file)
--- a/Lib/test/test_hash.py
+++ b/Lib/test/test_hash.py
@@ -28,4 +28,9 @@ class HashEqualityTestCase(unittest.TestCase):
         self.same_hash(float(0.5), complex(0.5, 0.0))
 
 
-test_support.run_unittest(HashEqualityTestCase)
+def test_main():
+    test_support.run_unittest(HashEqualityTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_iter.py b/Lib/test/test_iter.py
index 37fab7ced7680972e58f1626721a9e0f36281dad..f6084cb9a33caf9d543024c16809cbceafa9c527 100644 (file)
--- a/Lib/test/test_iter.py
+++ b/Lib/test/test_iter.py
@@ -694,4 +694,10 @@ class TestCase(unittest.TestCase):
         (a, b), (c,) = IteratingSequenceClass(2), {42: 24}
         self.assertEqual((a, b, c), (0, 1, 42))
 
-run_unittest(TestCase)
+
+def test_main():
+    run_unittest(TestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_mailbox.py b/Lib/test/test_mailbox.py
index 57a831631e11dce0d73d66f8bde94f877a6373b7..4e2339858c12e2c4a22409cc392e275ad3a0204c 100644 (file)
--- a/Lib/test/test_mailbox.py
+++ b/Lib/test/test_mailbox.py
@@ -96,4 +96,9 @@ class MaildirTestCase(unittest.TestCase):
     # XXX We still need more tests!
 
 
-test_support.run_unittest(MaildirTestCase)
+def test_main():
+    test_support.run_unittest(MaildirTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_mhlib.py b/Lib/test/test_mhlib.py
index 30f8ca3f10fa3456cf776fac2c12c1667327b157..9d5f101f345d9044bbd998ed803bcd6ccb5d475f 100644 (file)
--- a/Lib/test/test_mhlib.py
+++ b/Lib/test/test_mhlib.py
@@ -331,4 +331,10 @@ class MhlibTests(unittest.TestCase):
         msg.fp.close()
         del msg
 
-run_unittest(MhlibTests)
+
+def test_main():
+    run_unittest(MhlibTests)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_mimetypes.py b/Lib/test/test_mimetypes.py
index 9b9d3396ef47594271d9553efcf62919e1c9805e..8735e278ab2cefe22846290cadf06f139ed798f8 100644 (file)
--- a/Lib/test/test_mimetypes.py
+++ b/Lib/test/test_mimetypes.py
@@ -39,4 +39,9 @@ class MimeTypesTestCase(unittest.TestCase):
                          ".pyunit")
 
 
-test_support.run_unittest(MimeTypesTestCase)
+def test_main():
+    test_support.run_unittest(MimeTypesTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_operator.py b/Lib/test/test_operator.py
index 6312d9a295d9ca555b3251fd421c48afc88c7dea..439d340ce3c86ffbb4e826901c0927202fa7a086 100644 (file)
--- a/Lib/test/test_operator.py
+++ b/Lib/test/test_operator.py
@@ -210,4 +210,9 @@ class OperatorTestCase(unittest.TestCase):
         self.failUnless(operator.xor(0xb, 0xc) == 0x7)
 
 
-test_support.run_unittest(OperatorTestCase)
+def test_main():
+    test_support.run_unittest(OperatorTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
index 9cddb5f7004a7a12b92670cfeb635ed182c3f28d..3f26f2f8890508ffc8d2897ef4df42e2e9697ad8 100644 (file)
--- a/Lib/test/test_os.py
+++ b/Lib/test/test_os.py
@@ -62,5 +62,9 @@ class TemporaryFileTests(unittest.TestCase):
         self.check_tempfile(os.tmpnam())
 
 
+def test_main():
+    run_unittest(TemporaryFileTests)
 
-run_unittest(TemporaryFileTests)
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_parser.py b/Lib/test/test_parser.py
index 9e994750c916fb7a9f419e6fd8fd5a9258c82373..966f3e99342357dc09ac2cf3a6dab8b6ec7d18d7 100644 (file)
--- a/Lib/test/test_parser.py
+++ b/Lib/test/test_parser.py
@@ -353,5 +353,13 @@ class IllegalSyntaxTestCase(unittest.TestCase):
         self.check_bad_tree(tree, "a $= b")
 
 
-test_support.run_unittest(RoundtripLegalSyntaxTestCase)
-test_support.run_unittest(IllegalSyntaxTestCase)
+def test_main():
+    loader = unittest.TestLoader()
+    suite = unittest.TestSuite()
+    suite.addTest(loader.loadTestsFromTestCase(RoundtripLegalSyntaxTestCase))
+    suite.addTest(loader.loadTestsFromTestCase(IllegalSyntaxTestCase))
+    test_support.run_suite(suite)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_pkgimport.py b/Lib/test/test_pkgimport.py
index 2f1950922f50cf2f9d1cdb578e8f07036785b649..72889f97087dc8c4e0eb4ccf954ac86368571ed3 100644 (file)
--- a/Lib/test/test_pkgimport.py
+++ b/Lib/test/test_pkgimport.py
@@ -75,4 +75,10 @@ class TestImport(unittest.TestCase):
         reload(module)
         self.assertEqual(getattr(module, var), 1)
 
-run_unittest(TestImport)
+
+def test_main():
+    run_unittest(TestImport)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_pprint.py b/Lib/test/test_pprint.py
index 34c7a84f02a8fde2a55ecb244d73040be6a0acba..167b4ac24760a364a83c670a80e740891650a38a 100644 (file)
--- a/Lib/test/test_pprint.py
+++ b/Lib/test/test_pprint.py
@@ -76,4 +76,10 @@ class QueryTestCase(unittest.TestCase):
                 verify(native == got, "expected %s got %s from pprint.%s" %
                                       (native, got, function))
 
-test_support.run_unittest(QueryTestCase)
+
+def test_main():
+    test_support.run_unittest(QueryTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_pyclbr.py b/Lib/test/test_pyclbr.py
index ce4d8acb341dfb1b51f8abc66e92117a7d86c735..179b3b5968fee7500181bd83c67f0d55fe73af54 100644 (file)
--- a/Lib/test/test_pyclbr.py
+++ b/Lib/test/test_pyclbr.py
@@ -153,4 +153,10 @@ class PyclbrTest(unittest.TestCase):
         # pdb plays too many dynamic games
         # cm('pdb', pdb)
 
-run_unittest(PyclbrTest)
+
+def test_main():
+    run_unittest(PyclbrTest)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_quopri.py b/Lib/test/test_quopri.py
index 348f4bc95212a4ac2f475992c74aefc9ed8f42cf..0e997271fc48e5f19955538dd85b2d78fcf0a857 100644 (file)
--- a/Lib/test/test_quopri.py
+++ b/Lib/test/test_quopri.py
@@ -136,4 +136,9 @@ zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz''')
             self.assert_(decodestring(e) == p)
 
 
-test_support.run_unittest(QuopriTestCase)
+def test_main():
+    test_support.run_unittest(QuopriTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_repr.py b/Lib/test/test_repr.py
index cd9675a839d2a223fbfd7b4cfa2932aabf605adc..b7d937481ddd6a404b585dfcd6ee5a095a9cceab 100644 (file)
--- a/Lib/test/test_repr.py
+++ b/Lib/test/test_repr.py
@@ -265,6 +265,11 @@ class ClassWithFailingRepr:
         raise Exception("This should be caught by Repr.repr_instance")
 
 
-run_unittest(ReprTests)
-if os.name != 'mac':
-    run_unittest(LongReprTest)
+def test_main():
+    run_unittest(ReprTests)
+    if os.name != 'mac':
+        run_unittest(LongReprTest)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_rfc822.py b/Lib/test/test_rfc822.py
index 4040cdb859385815c66a23a7478f27dea4fcd891..ab1746b126e7d6cb8c23bbab5c91adbb437e91fa 100644 (file)
--- a/Lib/test/test_rfc822.py
+++ b/Lib/test/test_rfc822.py
@@ -177,4 +177,10 @@ class MessageTestCase(unittest.TestCase):
         self.check('To: User J. Person <person@dom.ain>\n\n',
                    [('User J. Person', 'person@dom.ain')])
 
-test_support.run_unittest(MessageTestCase)
+
+def test_main():
+    test_support.run_unittest(MessageTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_sha.py b/Lib/test/test_sha.py
index bc768095df771e474271b55959e55dff09f37aeb..cad7780467fff7c3d06e14b562bfb0772dded06e 100644 (file)
--- a/Lib/test/test_sha.py
+++ b/Lib/test/test_sha.py
@@ -27,4 +27,9 @@ class SHATestCase(unittest.TestCase):
                    "34aa973cd4c4daa4f61eeb2bdbad27316534016f")
 
 
-test_support.run_unittest(SHATestCase)
+def test_main():
+    test_support.run_unittest(SHATestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_strop.py b/Lib/test/test_strop.py
index 7a32e08707f3ed1e9a82ed3593160c3b92a1c8f2..e26f08fc94f8d57fabcc4b7159842e12ed1cc6d0 100644 (file)
--- a/Lib/test/test_strop.py
+++ b/Lib/test/test_strop.py
@@ -125,4 +125,9 @@ class Sequence:
     def __getitem__(self, i): return self.seq[i]
 
 
-test_support.run_unittest(StropFunctionTestCase)
+def test_main():
+    test_support.run_unittest(StropFunctionTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_time.py b/Lib/test/test_time.py
index 160ce4dbaae98d4dbc7f1673c258ad01e6a318a4..b2a3a2878804f074cd17794adc682f0e3c5486bd 100644 (file)
--- a/Lib/test/test_time.py
+++ b/Lib/test/test_time.py
@@ -48,4 +48,9 @@ class TimeTestCase(unittest.TestCase):
                                         999999))
 
 
-test_support.run_unittest(TimeTestCase)
+def test_main():
+    test_support.run_unittest(TimeTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
index 9f818e1607879742376457e4ae3b31cb476c2f02..1b0a1245445a54335fa582592166c4fcd6e85931 100644 (file)
--- a/Lib/test/test_traceback.py
+++ b/Lib/test/test_traceback.py
@@ -37,4 +37,10 @@ class TracebackCases(unittest.TestCase):
         self.assert_(len(err) == 3)
         self.assert_(err[1].strip() == "[x for x in x] = x")
 
-run_unittest(TracebackCases)
+
+def test_main():
+    run_unittest(TracebackCases)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_unary.py b/Lib/test/test_unary.py
index 01c5f01f9b3a18839f138a67f824c60c9cc3fdd4..0c5dd94ceef2e164ad8dc9ba04e9f9ed09715944 100644 (file)
--- a/Lib/test/test_unary.py
+++ b/Lib/test/test_unary.py
@@ -49,4 +49,10 @@ class UnaryOpTestCase(unittest.TestCase):
         self.assertRaises(TypeError, eval, "~2j")
         self.assertRaises(TypeError, eval, "~2.0")
 
-run_unittest(UnaryOpTestCase)
+
+def test_main():
+    run_unittest(UnaryOpTestCase)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_weakref.py b/Lib/test/test_weakref.py
index fa12a6e8f85f09fc43bcdc1539d57a64d407f553..5dfa6ce6573a5020e6adda6b651b86d498484563 100644 (file)
--- a/Lib/test/test_weakref.py
+++ b/Lib/test/test_weakref.py
@@ -2,7 +2,7 @@ import sys
 import unittest
 import weakref
 
-from test_support import run_unittest
+import test_support
 
 
 class C:
@@ -434,5 +434,13 @@ class MappingTestCase(TestBase):
         self.assert_(d.items() == [('something else', o2)])
 
 
-run_unittest(ReferencesTestCase)
-run_unittest(MappingTestCase)
+def test_main():
+    loader = unittest.TestLoader()
+    suite = unittest.TestSuite()
+    suite.addTest(loader.loadTestsFromTestCase(ReferencesTestCase))
+    suite.addTest(loader.loadTestsFromTestCase(MappingTestCase))
+    test_support.run_suite(suite)
+
+
+if __name__ == "__main__":
+    test_main()

diff --git a/Lib/test/test_xmllib.py b/Lib/test/test_xmllib.py
index a1c5057e773b900be8224ed1c8e225c27c0cad81..9ea50420948320026f951839c11709df904e4bc9 100644 (file)
--- a/Lib/test/test_xmllib.py
+++ b/Lib/test/test_xmllib.py
@@ -27,4 +27,9 @@ class XMLParserTestCase(unittest.TestCase):
         parser.close()
 
 
-test_support.run_unittest(XMLParserTestCase)
+def test_main():
+    test_support.run_unittest(XMLParserTestCase)
+
+
+if __name__ == "__main__":
+    test_main()