unittest: Allow passing module name or instance into unittest.main()

pull/488/head
Andrew Leech 2018-11-01 16:25:38 +11:00 zatwierdzone przez Andrew Leech
rodzic d64557a211
commit a9cd99ce2d
1 zmienionych plików z 1 dodań i 1 usunięć

Wyświetl plik

@ -219,7 +219,7 @@ def main(module="__main__"):
if isinstance(c, object) and isinstance(c, type) and issubclass(c, TestCase): if isinstance(c, object) and isinstance(c, type) and issubclass(c, TestCase):
yield c yield c
m = __import__(module) m = __import__(module) if isinstance(module, str) else module
suite = TestSuite() suite = TestSuite()
for c in test_cases(m): for c in test_cases(m):
suite.addTest(c) suite.addTest(c)