修改pytest Fixture参数默认值时遭遇KeyError: arg1问题求助

首先注意：你的原代码缺少@装饰器，正确的fixture定义应该以@pytest.fixture(...)开头，否则无法被pytest识别。

你的核心问题出在fixture的params参数是两个结构不一致的字典：第一个字典包含arg1键，第二个只包含arg2键。当pytest遍历到第二个参数时，直接通过request.param['arg1']访问不存在的键，就会抛出KeyError。

以下是几种可行的解决方法：

方法1：使用字典get()方法安全访问键 #

get()方法在键不存在时会返回None（或指定的默认值），不会抛出异常。修改代码如下：

@pytest.fixture(scope='function', params=[{'arg1': value1}, {'arg2': Value2}])
def some_fixture(request):
    if request.param.get('arg1') == value1: 
        # 执行对应操作
    if request.param.get('arg2') == Value2:
        # 执行对应操作

方法2：先检查键是否存在再访问 #

在取值前先判断键是否在字典中，避免直接访问不存在的键：

@pytest.fixture(scope='function', params=[{'arg1': value1}, {'arg2': Value2}])
def some_fixture(request):
    if 'arg1' in request.param and request.param['arg1'] == value1: 
        # 执行对应操作
    if 'arg2' in request.param and request.param['arg2'] == Value2:
        # 执行对应操作

方法3：统一参数字典的结构（推荐） #

如果每个测试用例逻辑上需要覆盖所有可能的参数键，可以调整params的结构，让每个字典都包含所有必要的键，不存在的参数用None或默认值填充：

@pytest.fixture(scope='function', params=[{'arg1': value1, 'arg2': None}, {'arg1': None, 'arg2': Value2}])
def some_fixture(request):
    if request.param['arg1'] == value1: 
        # 执行对应操作
    if request.param['arg2'] == Value2:
        # 执行对应操作

这样每个参数字典的结构一致，就不会出现键不存在的问题，代码逻辑也更清晰。

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