如何在pytest堆叠参数化中排除指定参数组合？

下面提供三种实用方案，满足你不想重复指定参数、批量复用的需求：

方案一：自定义参数生成函数（推荐复用场景） #

提前生成所有合法的参数组合，过滤掉不需要的项，然后把函数放在公共模块里，所有测试用例直接调用即可。

示例代码：

# 可放在公共utils模块或conftest.py中
def get_valid_test_params():
    x_values = [1, 2, 3, 4]
    y_values = [True, False]
    # 生成所有组合并排除x=3且y=False的情况
    valid_params = [
        (x, y) for x in x_values 
        for y in y_values 
        if not (x == 3 and y == False)
    ]
    return valid_params

# 测试用例中直接使用
@pytest.mark.parametrize("x, y", get_valid_test_params())
def test_get_something(x, y):
    get_something(x, y)

方案二：测试函数内直接跳过（快速修改） #

如果不想改动参数化结构，直接在测试函数内部判断并跳过特定组合，写法简单直接。

示例代码：

@pytest.mark.parametrize("x", [1, 2, 3, 4])
@pytest.mark.parametrize("y", [True, False])
def test_get_something(x, y):
    # 跳过目标组合
    if x == 3 and y == False:
        pytest.skip("排除x=3且y=False的测试组合")
    get_something(x, y)

方案三：全局钩子函数（全局统一配置） #

通过pytest的pytest_generate_tests钩子函数，在全局层面为所有用到x、y参数的测试自动生成过滤后的参数组合，无需在每个测试用例中写parametrize。

在项目根目录的conftest.py中添加如下代码：

def pytest_generate_tests(metafunc):
    # 检查当前测试函数是否需要x和y参数
    if "x" in metafunc.fixturenames and "y" in metafunc.fixturenames:
        x_values = [1, 2, 3, 4]
        y_values = [True, False]
        # 生成过滤后的参数组合
        filtered_params = [
            (x, y) for x in x_values 
            for y in y_values 
            if not (x == 3 and y == False)
        ]
        # 为测试函数参数化x和y
        metafunc.parametrize("x,y", filtered_params)

之后测试用例只需直接声明参数即可：

def test_get_something(x, y):
    get_something(x, y)

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