From e1ba8df182923c103fd595a1a0c5a8be9b381dbb Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Sat, 14 Sep 2013 00:37:18 +0200 Subject: [PATCH] Issue #19013: add a __main__ to unittest.test to ease CLI invocation --- Lib/unittest/test/__main__.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 Lib/unittest/test/__main__.py diff --git a/Lib/unittest/test/__main__.py b/Lib/unittest/test/__main__.py new file mode 100644 index 00000000000..44d0591e847 --- /dev/null +++ b/Lib/unittest/test/__main__.py @@ -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(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()