summaryrefslogtreecommitdiffstats
path: root/Lib/unittest
diff options
context:
space:
mode:
authorAntoine Pitrou <solipsis@pitrou.net>2013-09-13 22:37:18 (GMT)
committerAntoine Pitrou <solipsis@pitrou.net>2013-09-13 22:37:18 (GMT)
commite1ba8df182923c103fd595a1a0c5a8be9b381dbb (patch)
treeac42ea00b2b7187bcf31e9dee716c2e3bd3705fb /Lib/unittest
parent356bdeb0a173d026beb614050c84c38f73b88db6 (diff)
downloadcpython-e1ba8df182923c103fd595a1a0c5a8be9b381dbb.zip
cpython-e1ba8df182923c103fd595a1a0c5a8be9b381dbb.tar.gz
cpython-e1ba8df182923c103fd595a1a0c5a8be9b381dbb.tar.bz2
Issue #19013: add a __main__ to unittest.test to ease CLI invocation
Diffstat (limited to 'Lib/unittest')
-rw-r--r--Lib/unittest/test/__main__.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/Lib/unittest/test/__main__.py b/Lib/unittest/test/__main__.py
new file mode 100644
index 0000000..44d0591
--- /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()