如何自定义python - unitest.py？

Question

我知道这是可能的，我是Python的新手。看着鼻子。但是鼻子不是我们需要的。当我自己运行python unittest.py --help时，应该有一些方法可以自定义。有没有python人能给我们提供一些关于我们可以如何或在哪里改变的信息。我使用的是python 2.6.6

Answer 1

好的，你现在看到的是定制unittest.TestProgram。您将希望创建类似myunittest.py的内容，并在该模块中执行以下操作：

# myunittest.py
from unittest import TestProgram

class MyTestProgram(TestProgram):
    USAGE = """\
Usage: %(progName)s [options] [test] [...]

Options:
  -h, --help       Show this message
  -v, --verbose    Verbose output
  -q, --quiet      Minimal output

Examples:
  %(progName)s                               - run default set of tests
  %(progName)s MyTestSuite                   - run suite 'MyTestSuite'
  %(progName)s MyTestCase.testSomething      - run MyTestCase.testSomething
  %(progName)s MyTestCase                    - run all 'test*' test methods
                                               in MyTestCase
    """
    # additional changes here...

main = MyTestProgram

if __name__ == "__main__":
    main(module=None)

仅仅改变用法可能是你真正关心的，因为这是python unittest.py --help打印的文本，但显然你可以定制更多。

注意这是Python2.6特有的。Python 2.7+使用unittest2，目前我对它还不是很熟悉。