如何用所有特性标志组合运行Python单元测试并处理初始化问题

问题背景 #

我们有多个模块级的特性/回退标志（如foo、bar），需要测试它们所有True/False组合下的功能逻辑。使用mock.patch和importlib.reload时遇到问题：用装饰器批量生成测试用例时，测试多次运行间setUp/tearDown未正确执行，导致断言失败，且现有TestRunner无法轻易替换。

基础代码示例 #

enabled.py（标志定义模块） #

# 模块级特性标志
foo = False
bar = False

module.py（业务逻辑模块） #

from enabled import foo, bar

def baz():
    # 根据标志返回对应结果
    if foo and bar:
        return "both enabled"
    elif foo:
        return "foo enabled"
    elif bar:
        return "bar enabled"
    else:
        return "none enabled"

解决方法：手动遍历组合+独立模块加载 #

避免装饰器批量生成用例的生命周期问题，手动遍历所有标志组合，通过subTest拆分独立场景，确保每次测试都重置模块状态。

unittest测试代码 #

import unittest
import importlib
import sys
from unittest.mock import patch
import enabled
import module

class TestBaz(unittest.TestCase):
    def setUp(self):
        # 每次测试前重置标志模块的原始状态
        importlib.reload(enabled)
        # 重新加载业务模块，确保读取最新标志值
        importlib.reload(module)

    def tearDown(self):
        # 清理模块引用，彻底避免测试间状态污染
        for mod_name in ['enabled', 'module']:
            if mod_name in sys.modules:
                del sys.modules[mod_name]

    def test_baz_all_combinations(self):
        # 定义所有True/False组合及对应预期结果
        test_cases = [
            ((True, True), "both enabled"),
            ((True, False), "foo enabled"),
            ((False, True), "bar enabled"),
            ((False, False), "none enabled"),
        ]

        for (foo_val, bar_val), expected in test_cases:
            # 用subTest标记每个组合，方便定位失败场景
            with self.subTest(foo=foo_val, bar=bar_val):
                with patch('enabled.foo', foo_val), patch('enabled.bar', bar_val):
                    # 重新加载业务模块以应用新标志
                    importlib.reload(module)
                    self.assertEqual(module.baz(), expected)

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

更优方案：用pytest参数化简化测试 #

如果可以引入pytest（兼容现有unittest用例，无需替换TestRunner），参数化测试能更优雅地处理组合场景，自动维护测试独立性。

pytest测试代码示例 #

import pytest
import importlib
import sys
from unittest.mock import patch
import enabled
import module

# 直接参数化所有组合，自动生成独立测试用例
@pytest.mark.parametrize("foo_val, bar_val, expected", [
    (True, True, "both enabled"),
    (True, False, "foo enabled"),
    (False, True, "bar enabled"),
    (False, False, "none enabled"),
])
def test_baz(foo_val, bar_val, expected):
    # 重置标志模块并加载新值
    importlib.reload(enabled)
    with patch('enabled.foo', foo_val), patch('enabled.bar', bar_val):
        importlib.reload(module)
        assert module.baz() == expected

@pytest.fixture(autouse=True)
def clean_modules():
    # 自动清理模块引用，避免测试间污染
    yield
    for mod_name in ['enabled', 'module']:
        if mod_name in sys.modules:
            del sys.modules[mod_name]

核心优势 #

• 自动拆分每个组合为独立测试用例，报告展示更清晰，便于定位问题
• 无需手动维护subTest，代码更简洁
• 自动处理测试前后的模块清理，彻底避免状态交叉污染

关键注意事项 #

• 每次测试必须重新加载模块，避免Python模块缓存的旧值影响结果
• 测试后清理sys.modules中的模块引用，确保每个测试都是完全独立的环境
• 优先用参数化（unittest.subTest或pytest.parametrize）拆分组合场景，不要用装饰器批量生成用例导致生命周期冲突

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