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_coverage.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_coverage.py discover
  18 #
  19 # It may be possible to use other coverage commands with this, as
  20 # well...
  21
  22 """Main entry point"""
  23
  24 import sys
  25 if sys.argv[0].endswith("__main__.py"):
  26     sys.argv[0] = "python -m unittest"
  27
  28 __unittest = True
  29
  30 from unittest.main import main, TestProgram, USAGE_AS_MAIN
  31 TestProgram.USAGE = USAGE_AS_MAIN
  32
  33 main(module=None)