20、pytest中的参数化

白菜兔2023-12-09 2:40

官方实例

python 复制代码

# content of test_expectation.py

import pytest

@pytest.mark.parametrize("test_input, expected",[("3+5",8),("2+4",6),("6*9",42)])
def test_eval(test_input, expected):
    assert eval(test_input) == expected

python 复制代码

# content of test_expectation_class
import pytest

@pytest.mark.parametrize("n, expected",[(1,2),(3,4)])
class TestClass:
    def test_simple_case(self, n, expected):
        assert n + 1 == expected
        
    def test_weird_simple_case(self, n, expected):
        assert (n * 1) + 1 == expected

python 复制代码

# content of test_parametrize_glob.py
import pytest

pytestmark = pytest.mark.parametrize("n,expected",[(1,2),(3,4)])

class TestClass:
    def test_simple_case(self, n, expected):
        assert n + 1 == expected

    def test_weird_simple_case(self, n, expected):
        assert (n * 1) + 1 == expected

解读与实操

内置的pytest.mark.parametrize装饰器允许对测试函数的参数进行参数化。

注意：参数值按原样传递给测试（没有任何副本），例如，如果你传递一个列表或字典作为参数值，并且测试用例代码对它进行了修改，那么这些修改将在随后的测试用例调用中反映出来。

请注意，你也可以在类或模块上使用参数化标记，类中的函数将调用带有参数集内容。

要将模块中的所有测试参数化，可以将其赋值给pytestmark全局变量。

场景应用

测试函数相同，只是传入参数不同，就可以用这种方法扩展测试用例。特别是针对列表的搜索功能，同一个搜索函数，但是存在多个搜索项。