pytest测试失败时如何控制fixture报告仅输出seed而非stimulus值

提供两种改造成本极低的实现方式，都可以满足测试失败时只打印seed、不输出完整stimulus数据的需求：

方案1：封装stimulus自定义打印内容 #

将随机数据和对应seed封装为自定义类，重写__repr__方法仅输出seed信息，pytest测试失败打印fixture参数时会直接展示seed，不会输出冗长的测试数据：

import random
import pytest
from dataclasses import dataclass

@dataclass
class Stimulus:
    # 实际测试用的随机数据
    data: object
    # 对应随机种子
    seed: int

    def __repr__(self):
        # 自定义pytest打印的内容，仅暴露seed
        return f"Stimulus(seed={self.seed!r})"

@pytest.fixture
def seed():
    return random.getrandbits(32)

@pytest.fixture
def stimulus(seed):
    # 种随机种子
    random.seed(seed)
    # 替换为你实际的随机数据生成逻辑
    test_data = [random.randint(0, 100) for _ in range(1000)]
    return Stimulus(data=test_data, seed=seed)

def test_foo(stimulus):
    # 测试时取stimulus.data使用即可
    assert False # 模拟测试失败

方案2：通过pytest钩子统一输出seed #

如果不想改动现有fixture的返回结构，可以在测试目录的conftest.py中添加钩子逻辑，所有用到seed fixture的测试失败时，自动在错误日志中追加seed值：

# conftest.py
def pytest_runtest_makereport(item, call):
    # 仅处理测试执行阶段的失败
    if call.when == "call" and call.excinfo is not None:
        if "seed" in item.fixturenames:
            # 获取当前测试用的seed值
            used_seed = item.funcargs["seed"]
            # 追加到错误信息末尾
            call.excinfo.value.args = (
                *call.excinfo.value.args,
                f"\n本次测试使用的随机seed为: {used_seed}"
            )

额外优化：支持命令行传seed复现问题 #

可以给pytest加自定义命令行参数，不用每次修改fixture代码就能复现失败用例：

# conftest.py
def pytest_addoption(parser):
    parser.addoption("--fixed-seed", type=int, default=None, help="指定固定随机种子用于复现测试失败")

@pytest.fixture
def seed(request):
    input_seed = request.config.getoption("--fixed-seed")
    if input_seed is not None:
        return input_seed
    return random.getrandbits(32)

后续复现问题时直接执行命令 pytest --fixed-seed=失败用例对应的seed值 即可。

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