问通过命令行选项从Python程序运行unittest

EN

您必须确保始终遵循一些命名约定(您似乎正在这样做)：

1. 所有测试都使用相同的前缀命名(test_是标准的)，后面跟着要测试的模块的名称。 prog.py => test_prog.py
2. 测试驻留在test/目录中。

然后你可以做这样的事情：

prog.py

import sys
...
... do module stuff here...
...

if __name__ == "__main__":

    # Check if we want to run the tests for this file
    if "--unittest" in sys.argv:
        import unittest

        test_filename = 'test_' + __file__
        test_directory = 'test'

        suite = unittest.TestLoader().discover(test_directory, pattern=test_filename)
        unittest.TextTestRunner(verbosity=2).run(suite)

我们正在做的是：

• 检查命令参数以查看是否存在--unittest (因为这是您唯一希望运行测试的时候)。
• 如果是，那么我们创建test_prog.py -遵循我们设置的命名约定。
• 然后我们将其传递给TestLoader().discover函数。 发现(.)从指定的目录开始，并找到与所提供的模式匹配的所有测试模块(递归到子目录)。 在我们的示例中，它将查看test/目录中任何名为test_prog.py的模块。当它这样做时，它会加载它，并使用我们想要运行的TestSuite创建一个TestCases。
• 最后，我们手动测试unittest以运行在前一步中获得的suite。

通常，unittest会在后台为我们完成所有这些工作，但是由于我们试图运行一个特定的测试模块，所以我们必须准确地说明如何从哪里获得它。

另外，请注意，您必须为执行此操作--每个文件都要执行此操作。