pytest：參數化測試與參數笛卡爾積

Question

只是想知道，是否有任何（更）優雅的方式參數化與笛卡爾積？這是我想通了，至今：

numbers = [1,2,3,4,5] 
vowels  = ['a','e','i','o','u'] 
consonants = ['x','y','z'] 

cartesian = [elem for elem in itertools.product(*[numbers,vowels,consonants])] 

@pytest.fixture(params=cartesian) 
def someparams(request): 
    return request.param 

def test_something(someparams): 
    pass 

至少我想封裝的數字，元音，輔音和笛卡兒在夾具功能。

Answer 1

我可以想到兩種方法來做到這一點。一個使用參數化燈具，一個參數化測試功能。這取決於你哪一個你更優雅。

下面是參數化的測試功能：

import itertools 
import pytest 

numbers = [1,2,3,4,5] 
vowels = ['a','e','i','o','u'] 
consonants = ['x','y','z'] 


@pytest.mark.parametrize('number,vowel,consonant', 
    itertools.product(numbers, vowels, consonants) 
) 
def test(number, vowel, consonant): 
    pass 

值得注意的是，第二個參數參數多態裝飾可以是一個迭代，不只是一個列表。

這裏是你如何設定參數，每個燈具做到這一點：

import pytest 

numbers = [1,2,3,4,5] 
vowels = ['a','e','i','o','u'] 
consonants = ['x','y','z'] 


@pytest.fixture(params=numbers) 
def number(request): 
    return request.param 

@pytest.fixture(params=vowels) 
def vowel(request): 
    return request.param 

@pytest.fixture(params=consonants) 
def consonant(request): 
    return request.param 


def test(number, vowel, consonant): 
    pass 

你的直覺是正確的。通過參數化多個燈具中的每一個，pytest負責創建所有出現的排列。

測試輸出是相同的。下面是一個示例（我跑py.test與-vv選項）：

test_bar.py:22: test[1-a-x] PASSED 
test_bar.py:22: test[1-a-y] PASSED 
test_bar.py:22: test[1-a-z] PASSED 
test_bar.py:22: test[1-e-x] PASSED 
test_bar.py:22: test[1-e-y] PASSED 
test_bar.py:22: test[1-e-z] PASSED 
test_bar.py:22: test[1-i-x] PASSED 

Answer 2

可以將多個parametrize參數，在這種情況下，他們將產生的所有參數的產品：

import pytest 

numbers = [1,2,3,4,5] 
vowels = ['a','e','i','o','u'] 
consonants = ['x','y','z'] 


@pytest.mark.parametrize('number', numbers) 
@pytest.mark.parametrize('vowel', vowels) 
@pytest.mark.parametrize('consonant', consonants) 
def test(number, vowel, consonant): 
    pass