pytest测试参数化覆盖

时间:2019-04-30 11:18:37

标签: python testing pytest

我目前正在使用pytest_generate_tests参数化我的所有测试用例,并且效果很好。

我现在想做的是针对特定测试覆盖此行为。 如果我尝试在测试本身上使用pytest.mark.parametrize装饰器,则会收到ValueError: duplicate错误,这是可以理解的,因为我现在试图在两个地方对测试进行参数化。

有没有一种方法可以覆盖此测试用例的“默认”参数设置?

我可以通过执行以下类似操作来实现此目的,但这是一种非常棘手的方式:

def pytest_generate_tests(metafunc):
    fixture_modes = ['mode1', 'mode2']
    if 'fixture' in metafunc.fixturenames:
        fixture  = metafunc.config.getoption('fixture')
        if fixture:
            fixture_modes = [fixture]
        if metafunc.function.__name__ != 'func_to_skip':
            metafunc.parametrize('fixture_mode', fixture_modes, indirect=True)

有更好的方法吗?

1 个答案:

答案 0 :(得分:1)

例如,您可以检查测试是否为fixture_mode定义了自己的参数标记。

def pytest_generate_tests(metafunc):
    fixture_modes = ['spam', 'eggs']
    mark = metafunc.definition.get_closest_marker('parametrize')
    if not mark or 'fixture_mode' not in mark.args[0]:
        metafunc.parametrize('fixture_mode', fixture_modes, indirect=True)

现在,显式参数化将覆盖默认参数:

def test_spam(fixture_mode):
    ...

@pytest.mark.parametrize('fixture_mode', (1, 2, 3))
def test_eggs(fixture_mode):
    ...

test_spam将获得默认参数化,test_eggs将获得自定义参数:

test_mod.py::test_spam[spam]
test_mod.py::test_spam[eggs]
test_mod.py::test_eggs[1]
test_mod.py::test_eggs[2]
test_mod.py::test_eggs[3]