如何用pytest fixture生成可预测无限datetime序列？解决多yield报错

先搞定fixture的"多yield"错误 #

你遇到的问题核心是：pytest的fixture设计里，yield是用来分隔初始化逻辑和清理逻辑的，一个fixture只能有一个yield语句——yield之前是setup，之后是teardown，多次yield会破坏这个结构，直接触发报错。

不用放弃你的策略，只需要调整fixture的写法：不要在fixture里直接多次yield值，而是让fixture返回一个生成器对象或者可调用函数，用它们来实现无限生成可预测的datetime。

实现可预测的无限datetime生成器（两种方案） #

方案1：返回生成器对象（session级，全局递增）

这个方案会在整个测试会话中维持一个持续递增的datetime序列，所有测试用例共用同一个生成器状态：

import pytest
from datetime import datetime, timedelta

@pytest.fixture(scope="session")
def datetime_generator():
    # 固定起始时间和步长，保证可预测性
    current_dt = datetime(2023, 1, 1, 0, 0, 0)
    time_step = timedelta(hours=1)
    
    def _generator():
        nonlocal current_dt
        while True:
            yield current_dt
            current_dt += time_step
    
    return _generator()

测试用例里调用方式：

def test_table_sorting_asc(datetime_generator):
    # 生成递增的datetime值
    dt1 = next(datetime_generator)
    dt2 = next(datetime_generator)
    dt3 = next(datetime_generator)
    
    # 插入测试数据，验证排序逻辑
    # 比如：插入三条记录，created_at分别为dt1、dt2、dt3，查询后验证顺序正确

方案2：返回可调用函数（更灵活，支持重置）

如果需要在某些测试用例里重置datetime序列，可以用这个方案：

import pytest
from datetime import datetime, timedelta

@pytest.fixture(scope="session")
def datetime_generator():
    start_dt = datetime(2023, 1, 1, 0, 0, 0)
    current_dt = start_dt
    time_step = timedelta(hours=1)
    
    def get_next():
        nonlocal current_dt
        current_dt += time_step
        return current_dt - time_step  # 先返回当前值，再递增
    
    def reset():
        nonlocal current_dt
        current_dt = start_dt
    
    return {"get_next": get_next, "reset": reset}

测试用例里调用方式：

def test_first_sorting(datetime_generator):
    dt1 = datetime_generator["get_next"]()
    dt2 = datetime_generator["get_next"]()
    # 测试逻辑...

def test_second_sorting_with_reset(datetime_generator):
    datetime_generator["reset"]()  # 重置回起始时间
    dt1 = datetime_generator["get_next"]()
    assert dt1 == datetime(2023, 1, 1, 0, 0, 0)
    # 测试逻辑...

关键注意点 #

1. 可预测性：固定起始时间和步长，确保每次测试运行生成的datetime序列完全一致，方便问题复现。
2. session级作用域：整个测试会话中生成器状态持续保留，所有测试用例的datetime会按顺序递增，避免重复值干扰排序测试。
3. 无需调整数据集：不管你需要生成10条还是100条测试数据，直接调用生成器即可，不用手动修改数据集。

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