如何遍历unittest测试套件并逐个执行测试？解决id属性报错问题

错误原因 #

unittest.defaultTestLoader.discover()返回的是层级化的TestSuite结构——顶层TestSuite中可能包含子TestSuite，而非直接的TestCase对象。你之前的嵌套循环只处理了两层，当遇到子TestSuite时调用test.id()就会触发AttributeError，因为TestSuite没有id()方法，只有TestCase才有。

解决方案：递归遍历测试套件 #

通过递归遍历TestSuite的所有层级，提取出所有TestCase对象，就能安全获取每个测试的id()，并实现逐个执行的需求。

完整代码示例 #

import unittest

def extract_all_test_cases(suite):
    """递归遍历TestSuite，提取所有TestCase对象"""
    test_cases = []
    for element in suite:
        if isinstance(element, unittest.TestCase):
            test_cases.append(element)
        else:
            # 递归处理子TestSuite
            test_cases.extend(extract_all_test_cases(element))
    return test_cases

# 生成测试套件
suite = unittest.defaultTestLoader.discover("./tests")

# 提取所有TestCase
all_tests = extract_all_test_cases(suite)

# 逐个打印测试id并执行测试
for test in all_tests:
    print(f"正在执行测试: {test.id()}")
    # 单独执行当前测试
    runner = unittest.TextTestRunner()
    runner.run(test)

代码说明 #

1. extract_all_test_cases函数：递归遍历传入的TestSuite，判断每个元素是TestCase还是子TestSuite，将所有TestCase收集到列表中。
2. 遍历提取出的TestCase列表：可以直接调用test.id()获取测试的唯一标识（格式为测试模块.测试类.测试方法），同时用TextTestRunner逐个执行测试。

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