pytest 測試參數化覆蓋

Question

我目前正在使用pytest_generate_tests參數化我的所有測試用例，這很好用。

我現在想做的是為特定測試覆蓋此行為。 如果我嘗試在測試本身上使用pytest.mark.parametrize裝飾器， pytest.mark.parametrize收到ValueError: duplicate錯誤，這是可以理解的，因為我現在試圖在兩個地方對測試進行參數化。

有沒有辦法可以覆蓋這個測試用例的“默認”參數化？

我可以通過執行以下操作來實現這一點，但這是一種非常hacky的方法：

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)

有一個更好的方法嗎？

Answer 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]