test.support: run_unittest: Accept string test module name as param.

This is used e.g. by datetimetester.py from CPython.
pull/211/head
Paul Sokolovsky 2017-09-22 22:21:24 +03:00
rodzic a953735f3d
commit 32060ab025
1 zmienionych plików z 8 dodań i 1 usunięć

Wyświetl plik

@ -10,7 +10,14 @@ TESTFN = '@test'
def run_unittest(*classes):
suite = unittest.TestSuite()
for c in classes:
suite.addTest(c)
if isinstance(c, str):
c = __import__(c)
for name in dir(c):
obj = getattr(c, name)
if isinstance(obj, type) and issubclass(obj, unittest.TestCase):
suite.addTest(obj)
else:
suite.addTest(c)
runner = unittest.TestRunner()
result = runner.run(suite)