Factor out common boilerplate for test_support

rhettinger · rhettinger · commit b8237bb73eeb · 2003-04-27T07:54:23.000Z
git-svn-id: http://svn.python.org/projects/python/trunk@32501 6015fed2-1504-0410-9fe1-9d1591cc4771
diff --git a/Lib/test/test_bisect.py b/Lib/test/test_bisect.py
@@ -192,24 +192,14 @@ def test_backcompatibility(self):
 
 """
 
-#==============================================================================
-
-def makeAllTests():
-    suite = unittest.TestSuite()
-    for klass in (TestBisect,
-                  TestInsort
-                  ):
-        suite.addTest(unittest.makeSuite(klass))
-    return suite
-
 #------------------------------------------------------------------------------
 
 __test__ = {'libreftest' : libreftest}
 
 def test_main(verbose=None):
     from test import test_bisect
-    suite = makeAllTests()
-    test_support.run_suite(suite)
+    test_support.run_classtests(TestBisect,
+                                TestInsort)
     test_support.run_doctest(test_bisect, verbose)
 
 if __name__ == "__main__":
diff --git a/Lib/test/test_bool.py b/Lib/test/test_bool.py
@@ -331,9 +331,7 @@ def test_picklevalues(self):
         self.assertEqual(cPickle.dumps(False, True), "I00\n.")
 
 def test_main():
-    suite = unittest.TestSuite()
-    suite.addTest(unittest.makeSuite(BoolTest))
-    test_support.run_suite(suite)
+    test_support.run_classtests(BoolTest)
 
 if __name__ == "__main__":
     test_main()
diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
@@ -9,7 +9,7 @@
 warnings.filterwarnings("ignore", "tempnam", RuntimeWarning, __name__)
 warnings.filterwarnings("ignore", "tmpnam", RuntimeWarning, __name__)
 
-from test.test_support import TESTFN, run_suite
+from test.test_support import TESTFN, run_classtests
 
 class TemporaryFileTests(unittest.TestCase):
     def setUp(self):
@@ -282,14 +282,10 @@ def test_traversal(self):
         os.rmdir(TESTFN)
 
 def test_main():
-    suite = unittest.TestSuite()
-    for cls in (TemporaryFileTests,
-                StatAttributeTests,
-                EnvironTests,
-                WalkTests,
-               ):
-        suite.addTest(unittest.makeSuite(cls))
-    run_suite(suite)
+    run_classtests(TemporaryFileTests,
+                   StatAttributeTests,
+                   EnvironTests,
+                   WalkTests)
 
 if __name__ == "__main__":
     test_main()
diff --git a/Lib/test/test_support.py b/Lib/test/test_support.py
@@ -233,6 +233,12 @@ def run_unittest(testclass):
     """Run tests from a unittest.TestCase-derived class."""
     run_suite(unittest.makeSuite(testclass), testclass)
 
+def run_classtests(*classnames):
+    suite = unittest.TestSuite()
+    for cls in classnames:
+        suite.addTest(unittest.makeSuite(cls))
+    run_suite(suite)
+
 
 #=======================================================================
 # doctest driver.
