Issue #19013: add a __main__ to unittest.test.testmock to ease CLI invocation

This commit is contained in:
Antoine Pitrou 2013-09-14 00:39:15 +02:00
parent e1ba8df182
commit 4b3c58c8a0

View file

@ -0,0 +1,18 @@
import os
import unittest
def load_tests(loader, standard_tests, pattern):
# top level directory cached on loader instance
this_dir = os.path.dirname(__file__)
pattern = pattern or "test*.py"
# We are inside unittest.test, so the top-level is two notches up
top_level_dir = os.path.dirname(os.path.dirname(os.path.dirname(this_dir)))
package_tests = loader.discover(start_dir=this_dir, pattern=pattern,
top_level_dir=top_level_dir)
standard_tests.addTests(package_tests)
return standard_tests
if __name__ == '__main__':
unittest.main()