从不同的文件运行单元测试

问题描述

我有一个包含单元测试的文件TestProtocol.py。我可以运行该脚本并获得预期的30个测试的测试结果。现在，我想从同一目录中的另一个文件tester.py运行这些测试。在tester.py内部我尝试了import TestProtocol，但它运行了0个测试。

然后我找到了文档，上面说我应该这样做：

suite = unittest.TestLoader().discover(".", pattern = "*")
unittest.run(suite)

这应该遍历当前目录.中与模式*匹配的所有文件，因此所有文件中的所有测试。遗憾的是，它再次运行0个测试。

有一个related QA建议

import TestProtocol
suite = unittest.findTestCases(TestProtocol)
unittest.run(suite)

但这也找不到任何测试。

如何导入和运行测试？

---

解决方案

您可以尝试使用以下

# preferred module name would be test_protol as CamelCase convention are used for class name
import TestProtocol

# try to load all testcases from given module, hope your testcases are extending from unittest.TestCase
suite = unittest.TestLoader().loadTestsFromModule(TestProtocol)
# run all tests with verbosity 
unittest.TextTestRunner(verbosity=2).run(suite)

以下是完整的示例

文件1：test_me.py

# file 1: test_me.py 
import unittest

class TestMe(unittest.TestCase):
    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

if __name__ == '__main__':
    unittest.main()

文件2：test_ther.py，将其放在同一目录下

# file 2: test_other.py, put this under same directory
import unittest
import test_me

suite = unittest.TestLoader().loadTestsFromModule(test_me)
unittest.TextTestRunner(verbosity=2).run(suite)

运行每个文件，它将显示相同的结果

# python test_me.py - Ran 1 test in 0.000s
# python test_other.py - Ran 1 test in 0.000s