python/unittest_main.py

   1 #!/usr/bin/env python
   2 #
   3 # Source:
   4 # http://blog.wearpants.org/integrating-coverage-and-unittest-discovery
   5 #
   6 # Run via::
   7 #
   8 #   python -m coverage run unittest_main.py [args]
   9 #
  10 # What was done: Python/Lib/unittest/__main__.py was copied VERBATIM
  11 # here.  This means that running this script via coverage effectively
  12 # provides a wrapper around "python -m unittest".  Pass arguments
  13 # afterwards like normal.
  14 #
  15 # Unit test coverage with automatic discovery::
  16 #
  17 #   python -m coverage run unittest_main.py discover
  18
  19 """Main entry point"""
  20
  21 import sys
  22 if sys.argv[0].endswith("__main__.py"):
  23     sys.argv[0] = "python -m unittest"
  24
  25 __unittest = True
  26
  27 from unittest.main import main, TestProgram, USAGE_AS_MAIN
  28 TestProgram.USAGE = USAGE_AS_MAIN
  29
  30 main(module=None)