如何在pytest.mark.parametrize迭代中执行不同断言？

嗨，这个需求我太懂了——不想靠len(file_paths)这种容易失效的判断来区分测试用例的断言逻辑，毕竟后续输入列表随时可能变，对吧？我给你几个实用的方案，都能完美解决这个问题：

方案1：给每个测试用例绑定专属标识 #

直接在parametrize里给每组输入搭配一个明确的标识（比如用枚举或者字符串），这样测试函数里不用依赖输入内容，直接根据标识执行对应断言。用枚举的话还能避免字符串拼写错误，更健壮：

import pytest
from enum import Enum

# 定义枚举来标识不同测试场景
class TestScenario(Enum):
    SINGLE_SCRIPT = "single_script"
    MULTIPLE_SCRIPTS = "multiple_scripts"
    EMPTY_AND_DIRECTORY = "empty_and_directory"

@pytest.mark.parametrize('file_paths, scenario', [
    (['script.py'], TestScenario.SINGLE_SCRIPT),
    (['script.py', 'script1.py', 'testdir/script2.py'], TestScenario.MULTIPLE_SCRIPTS),
    (['', 'testdir'], TestScenario.EMPTY_AND_DIRECTORY)
])
def test_save_file(project_explorer_with_files, file_paths, scenario):
    # 你的业务逻辑代码...
    # codes codes codes ...

    if scenario == TestScenario.SINGLE_SCRIPT:
        # 第一个场景的断言逻辑
        assert ...
    elif scenario == TestScenario.MULTIPLE_SCRIPTS:
        # 第二个场景的断言逻辑
        assert ...
    elif scenario == TestScenario.EMPTY_AND_DIRECTORY:
        # 第三个场景的断言逻辑
        assert ...

后续如果要新增或修改测试用例，只需要在枚举里加新项，再补充对应的分支就行，完全不用关心输入列表的长度或内容变化。

方案2：拆分测试函数（更符合pytest最佳实践） #

如果每个测试场景的断言差异比较大，不如直接把它们拆成单独的测试函数，把公共逻辑抽成一个辅助函数，这样每个测试的职责更单一，可读性也更强：

def _process_save_file(project_explorer_with_files, file_paths):
    # 把所有测试用例共享的业务逻辑放到这里，避免重复代码
    # codes codes codes ...
    return result  # 返回需要断言的结果

def test_save_single_script(project_explorer_with_files):
    result = _process_save_file(project_explorer_with_files, ['script.py'])
    # 单文件场景的断言
    assert ...

def test_save_multiple_scripts(project_explorer_with_files):
    result = _process_save_file(project_explorer_with_files, ['script.py', 'script1.py', 'testdir/script2.py'])
    # 多文件场景的断言
    assert ...

def test_save_empty_and_directory(project_explorer_with_files):
    result = _process_save_file(project_explorer_with_files, ['', 'testdir'])
    # 空路径+目录场景的断言
    assert ...

这个方案的好处是，测试报告里的函数名能直接体现测试场景，出问题时定位更快；而且修改某个场景的逻辑或断言，完全不会影响其他测试用例。

方案3：利用pytest的ids参数和request对象 #

如果你不想添加额外参数，也可以给每个用例命名，然后通过request对象获取当前用例的ID来判断：

@pytest.mark.parametrize('file_paths', [
    ['script.py'],
    ['script.py', 'script1.py', 'testdir/script2.py'],
    ['', 'testdir']
], ids=["single_script", "multiple_scripts", "empty_and_directory"])
def test_save_file(project_explorer_with_files, file_paths, request):
    # 你的业务逻辑代码...
    # codes codes codes ...

    case_id = request.node.callspec.id
    if case_id == "single_script":
        assert ...
    elif case_id == "multiple_scripts":
        assert ...
    elif case_id == "empty_and_directory":
        assert ...

不过这个方案要注意，如果修改了ids里的命名，必须同步修改判断逻辑里的字符串，相比枚举方案容错性稍差一点。

个人更推荐方案1或方案2，方案2尤其适合测试逻辑差异较大的场景，方案1则适合想保持单个测试函数的情况。

内容的提问来源于stack exchange，提问作者Khalil Al Hooti