当前位置：首页 > news >正文

【pytest】pytest` 中几种常用的参数化方法

news 2025/9/12 19:23:00

pytest 是一个强大的 Python 测试框架，它提供了多种参数化测试的方法。参数化测试允许你使用不同的输入集来运行相同的测试逻辑，从而确保代码在各种条件下都能正常工作。以下是 pytest 中几种常用的参数化方法：

1. 使用 `@pytest.mark.parametrize` 装饰器

这是 pytest 中最常用的参数化方法。你可以使用 @pytest.mark.parametrize 装饰器来指定测试函数的参数和对应的值。

示例：

import pytest@pytest.mark.parametrize("input, expected", [(1, 2),(2, 4),(3, 6),
])
def test_multiplication(input, expected):assert input * 2 == expected

在这个例子中，test_multiplication 函数会被调用三次，每次使用不同的 (input, expected) 对作为参数。

2. 使用 pytest-cases 插件

pytest-cases 是一个第三方插件，它提供了更强大和灵活的参数化方法。你可以使用它来定义和组合测试用例，并将它们作为参数传递给测试函数。

安装：

pip install pytest-cases

示例：

import pytest
from pytest_cases import case, parametrize_with_cases@case(id="case1")
def case_data1():return 1, 2@case(id="case2")
def case_data2():return 2, 4@parametrize_with_cases("input, expected", cases=[case_data1, case_data2])
def test_multiplication(input, expected):assert input * 2 == expected

在这个例子中，我们使用了 pytest-cases 插件来定义了两个测试用例 case_data1 和 case_data2，并将它们作为参数传递给 test_multiplication 函数。

3. 使用 pytest-subtests 插件

pytest-subtests 插件允许你在一个测试函数中运行多个子测试，每个子测试都有自己的输入和预期输出。这对于测试具有多个分支或条件的函数非常有用。

安装：

pip install pytest-subtests

示例：

import pytest
from pytest_subtests import SubTestsdef test_multiplication():with SubTests() as subtests:for input, expected in [(1, 2), (2, 4), (3, 6)]:with subtests.test(input=input, expected=expected):assert input * 2 == expected