Extend last change to cover TestSuites as well as TestCases.

This commit is contained in:
Raymond Hettinger 2003-07-16 04:29:42 +00:00
parent 6c85bcb2aa
commit f3590623e9
1 changed files with 1 additions and 1 deletions

View File

@ -251,7 +251,7 @@ def run_unittest(*classes):
"""Run tests from unittest.TestCase-derived classes.""" """Run tests from unittest.TestCase-derived classes."""
suite = unittest.TestSuite() suite = unittest.TestSuite()
for cls in classes: for cls in classes:
if isinstance(cls, unittest.TestCase): if isinstance(cls, (unittest.TestSuite, unittest.TestCase)):
suite.addTest(cls) suite.addTest(cls)
else: else:
suite.addTest(unittest.makeSuite(cls)) suite.addTest(unittest.makeSuite(cls))