有没有办法使用多返回值夹具运行测试函数？

Question

我有一个想要参数化的测试：

@pytest.mark.parametrize("param", [1,2])
def test(param):
    assert param==1

现在，如果我想让“param”具有复杂的逻辑，我可以这样做：

def f():
    return 1

def g(): # something dumb just to make a point
    for _ in range(100):
        pass
    return 1

@pytest.mark.parametrize("param", [f(), g()])
def test(param):
    assert param==1

但是，这将在测试运行之前运行“setup”函数，即

1. f()
2. g ()
3. f() 的测试值
4. g() 的测试值

我尝试将其作为多返回值固定装置：

@pytest.fixture
def param():
    for i in "12":
        yield i

def test(param):
    assert param==1

但这是非法的（拆卸时中断），所以现在我想象类似我展示的参数化的东西，但是与步骤2 & 3 交换顺序，这样我就有了 2 个独立的测试。

是否可以？

Answer 1

一种可能性是使用间接参数化来推迟对运行时调用的固定装置的实际函数调用（与加载时评估的参数相反）。在您的情况下，您可以使用可调用对象作为参数：

@pytest.fixture
def param(request):
    fct = request.param  # f or g
    yield fct()


@pytest.mark.parametrize("param", [f, g], indirect=True)
def test(param):
    assert param == 1

在这种情况下，每个函数将在执行相应的测试之前被调用。