传递给pytest fixture的params可以作为变量传入吗?

时间:2012-12-18 02:18:53

标签: python pytest

我有两个简单的测试设置,我试图将它们分组在一个夹具中 并希望测试功能将'params'传递给灯具。

这是一个人为的例子,用来解释我的问题。 假设我有以下pytest夹具:

@pytest.fixture(scope="module", params=['param1','param2'])
def myFixture(request):
    if request.param == 'param1':
        p = 5
    elif request.param == 'param2':
        p = 10
    return p

# would like to set request.param = ['param1'] for myFixture
def test_madeup(myFixture):
    assert myFixture == 5

# would like to set request.param = ['param2'] for myFixture
def test_madeup2(myFixture):
    assert myFixture == 10

我是否可以将上面的参数作为test_madeup函数的输入传入? 所以,如下所示:

@pytest.fixture(scope="module", params=fixtureParams)
def myFixture(request):
    if request.param == 'param1':
        return 5
    elif request.param == 'param2':
        return 10


def test_madeup(myFixture, ['param1']):
    assert myFixture == 5

以上,当然不起作用。真实案例有点复杂, 但我只是想知道我是否可以通过params=['param1','param2'] 来自test_madeup函数的夹具。

2 个答案:

答案 0 :(得分:14)

如果我正确理解你的问题,你基本上想要通过提供测试的一些信息来选择参数化夹具的一个实例来执行测试。尽管我们可能想到一种机制,但这是不可能的。我不确定以下解决方案是否映射到您的整个问题,但这是解决上述具体案例的一种方法:

import pytest

@pytest.fixture(scope="module")
def myFixture1():
    return 5

@pytest.fixture(scope="module")
def myFixture2():
    return 2

@pytest.fixture(scope="module", params=["param1", "param2"])
def myFixture(request):
    if request.param == 'param1':
        return request.getfuncargvalue("myFixture1")
    elif request.param == 'param2':
        return request.getfuncargvalue("myFixture2")

def test_1(myFixture1):
    assert myFixture1 == 5

def test_2(myFixture2):
    assert myFixture2 == 2

def test_all(myFixture):
    assert myFixture in (2,5)

这会运行四个测试,因为test_all会在两个灯具中执行两次。

如果你的灯具设置不重,你可能还有一个灯具可以产生一个列表和一个“迭代”参数化的灯具。然后,测试可以获取整个列表并将其编入索引。

答案 1 :(得分:4)

不确定这是否是您想要的,但示例案例可以实现如下:

@pytest.mark.parametrize(('param', 'expected'), [('param1', 5), ('param2', 10)])
def test_madeup(param, expected):
    assert param == expected