33

这是我的设置 -

project/
    __init__.py
    prog.py
    test/
        __init__.py
        test_prog.py

我希望能够通过调用prog.py中的命令行选项来运行我的单元测试。这样，当我部署我的项目时，我可以随时部署运行单元测试的能力。

python prog.py --unittest

我在prog.py或我的项目的其余部分需要什么才能使其正常工作？

4

4 回答 4

69

Pythonunittest模块包含自己的测试发现功能，您可以从命令行运行该功能：

$ python -m unittest discover

要从您的模块中运行此命令，您可以使用该subprocess模块：

#!/usr/bin/env python

import sys
import subprocess

# ...
# the rest of your module's code
# ...

if __name__ == '__main__':
    if '--unittest' in sys.argv:
        subprocess.call([sys.executable, '-m', 'unittest', 'discover'])

如果您的模块有其他命令行选项，您可能想要查看argparse更高级的选项。

于 2013-05-31T21:21:05.000 回答
15

也许这就是你要找的。实现一个load_tests函数test_prog.py并使用以下代码prog.py来加载和运行测试：

import unittest
import test.test_prog
suite = unittest.TestLoader().loadTestsFromModule(test.test_prog)
unittest.TextTestRunner().run(suite)
于 2013-05-31T20:28:22.327 回答
11

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

  1. 所有测试都以相同的前缀命名（test_是规范），后跟您要测试的模块的名称。

    prog.py=>test_prog.py

  2. 测试驻留在test/目录中。

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

程序.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函数。

    discover(...)从指定目录开始，查找与提供的模式匹配的所有测试模块（递归到子目录）。

    在我们的例子中，它将在test/目录中查找任何名为test_prog.py. 当它完成时，它会加载它并使用我们要运行的 TestCases 创建一个 TestSuite。

  • 最后，我们手动测试unittest运行suite上一步中获得的。

通常，unittest它将在后台为我们完成所有这些工作，但是由于我们正在尝试运行特定的测试模块，因此我们必须准确说明如何以及从何处获取它。

另外，请注意，您必须对要在其中执行此操作的每个文件执行此操作。

于 2013-05-31T21:13:36.147 回答
0

我刚刚删除.idea了文件系统中项目下的隐藏目录。

我重新启动PyCharm，重新打开项目，这个问题得到了解决。

于 2020-12-02T22:48:55.183 回答