如何在Python unittest subTest运行过程中实时检测失败？

官方unittest的subTest默认会跑完所有子测试后才统一输出失败信息，当存在数千个subTest时，等待过程无法及时得知失败情况。以下是两种Python 3下的解决方案：

方法一：在测试用例中手动捕获断言异常 #

直接在subTest块内捕获断言错误，实时打印失败信息，同时保留测试失败的记录：

import unittest

class NumbersTest(unittest.TestCase):
    def test_even(self):
        """Test that numbers between 0 and 5 are all even."""
        for i in range(0, 6):
            with self.subTest(i=i):
                try:
                    self.assertEqual(i % 2, 0)
                except AssertionError as e:
                    # 实时打印失败详情
                    print(f"SubTest 失败: i={i}, 错误信息: {str(e)}")
                    # 抛出异常让unittest记录该subTest失败，不终止整个测试循环
                    raise

这种方式的优点是代码改动小，仅针对特定测试用例生效；缺点是需要逐个修改需要实时反馈的测试函数。

方法二：自定义TestResult全局监听 #

通过重写TextTestResult的addSubTest方法，全局监听所有subTest的失败事件，实时打印信息，无需修改测试用例代码：

import unittest

class RealTimeSubTestResult(unittest.TextTestResult):
    def addSubTest(self, test, subtest, err):
        # 先调用父类方法，确保测试结果被正常记录
        super().addSubTest(test, subtest, err)
        if err is not None:
            # 提取测试名称和子测试名称
            test_desc = self.getDescription(test)
            subtest_name = subtest._subtest_name()
            # 实时打印失败信息
            print(f"\n=== 实时子测试失败 ===")
            print(f"测试用例: {test_desc}")
            print(f"子测试参数: {subtest_name}")
            print(f"错误类型: {err[0].__name__}")
            print(f"错误详情: {err[1]}")

if __name__ == '__main__':
    # 使用自定义的结果类运行测试
    runner = unittest.TextTestRunner(resultclass=RealTimeSubTestResult)
    unittest.main(testRunner=runner)

这种方式的优点是一次配置全局生效，所有subTest的失败都会实时输出，适合大量测试用例的场景；缺点是需要自定义测试运行器。

内容的提问来源于stack exchange，提问作者dfrankow