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):
yield c
m = __import__(module)
m = __import__(module) if isinstance(module, str) else module
suite = TestSuite()
for c in test_cases(m):
suite.addTest(c)