如何在Python unittest中通过程序获取测试模块名称？

当然可以获取测试模块名称，但你直接遍历discover返回的顶层测试套件是不行的——因为顶层套件里包含的是子套件（对应测试目录或测试文件），而非具体的测试用例，所以你看到的__module__全是unittest.suite。要拿到真正的测试模块名，得递归遍历所有层级的测试套件，定位到最底层的测试用例对象，再从用例中提取模块信息。

解决代码 #

先修正你原代码的导入问题（正确导入TestLoader的方式），再用递归函数遍历所有测试用例：

from unittest import TestLoader

def get_unique_test_modules(suite):
    module_names = set()
    for item in suite:
        # 判断当前项是否是子套件（可迭代）
        if hasattr(item, '__iter__'):
            # 递归处理子套件，合并结果
            module_names.update(get_unique_test_modules(item))
        else:
            # 当前项是测试用例，提取所属模块名
            module_name = item.__module__
            module_names.add(module_name)
    return module_names

# 加载指定路径下的所有测试
loader = TestLoader()
test_suite = loader.discover('path/to/tests')

# 获取所有唯一的测试模块名称
modules = get_unique_test_modules(test_suite)
for mod in modules:
    print(mod)

关键说明 #

• 嵌套结构处理：discover返回的测试套件是嵌套的（目录→测试文件→测试类→测试用例），必须递归遍历才能触达底层的测试用例。
• 去重处理：同一个测试模块下的多个用例会重复返回模块名，用集合set可以自动去重，得到唯一的模块列表。
• 测试用例的__module__：每个测试用例实例（比如TestCase的子类实例）的__module__属性，就是它所属测试模块的完整名称（比如tests.user.test_register）。

可选：直接打印所有用例的模块名（不去重） #

如果不需要去重，只想看到每个测试用例对应的模块名，可以用这个简化版：

def print_all_test_modules(suite):
    for item in suite:
        if hasattr(item, '__iter__'):
            print_all_test_modules(item)
        else:
            print(item.__module__)

print_all_test_modules(test_suite)

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