Commit b3f60837 authored by Raymond Hettinger's avatar Raymond Hettinger

Extend last change to cover TestSuites as well as TestCases.

parent e9a636db
...@@ -251,7 +251,7 @@ def run_unittest(*classes): ...@@ -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))
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment