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):
"""Run tests from unittest.TestCase-derived classes."""
suite = unittest.TestSuite()
for cls in classes:
if isinstance(cls, unittest.TestCase):
if isinstance(cls, (unittest.TestSuite, unittest.TestCase)):
suite.addTest(cls)
else:
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