python – 发现后过滤测试

假设你有一个带有这个单元测试之王的测试目录：

import unittest


class MyTest(unittest.TestCase):
    def test_suite_1(self):
        self.assertFalse("test_suite_1")

    def test_suite_2(self):
        self.assertFalse("test_suite_2")

    def test_other(self):
        self.assertFalse("test_other")

您可以编写自己的发现函数,仅发现以“test_suite_”开头的测试函数,例如：

import unittest


def run_suite():
    loader = unittest.TestLoader()
    loader.testMethodPrefix = "test_suite_"
    suite = loader.discover("tests")
    result = unittest.TestResult()
    suite.run(result)
    for test,info in result.failures:
        print(info)


if __name__ == '__main__':
    run_suite()

注意：discover方法中的参数“tests”是一个目录路径,因此您可能需要编写一个完整路径.

结果,你会得到：

Traceback (most recent call last):
  File "/path/to/tests/test_my_module.py",line 8,in test_suite_1
    self.assertFalse("test_suite_1")
AssertionError: 'test_suite_1' is not false

Traceback (most recent call last):
  File "/path/to/tests/test_my_module.py",line 11,in test_suite_2
    self.assertFalse("test_suite_2")
AssertionError: 'test_suite_2' is not false