pytest的参数化

1. 什么是pytest的参数化

pytest是一个用于编写Python测试的框架，它提供了许多方便的功能和特性来简化测试过程。其中一个非常有用的功能就是参数化（Parameterization），它允许我们在测试用例中使用不同的参数进行多次运行，从而更全面地覆盖不同的测试场景。参数化使得我们可以更高效地编写测试用例，减少冗余代码的编写和维护工作，提高测试的覆盖率和可读性。

2. 如何使用pytest的参数化

在pytest中使用参数化非常简单。我们可以使用@pytest.mark.parametrize装饰器来定义参数化的测试用例函数，然后在函数内部使用参数进行测试。

2.1 参数化示例

下面是一个简单的示例，我们通过参数化来测试一个计算平方的函数：

import pytest

def square(num):
    return num ** 2
@pytest.mark.parametrize("input,expected", [(2, 4), (3, 9), (4, 16)])
def test_square(input, expected):
    assert square(input) == expected

在上面的示例中，我们使用了@pytest.mark.parametrize装饰器来定义了一个参数化的测试用例函数test_square，它接受两个参数：input和expected。然后在装饰器的参数中，我们传递了一个参数列表，每个参数列表包含一个输入值和预期的输出值。

当我们运行上述代码时，pytest会为每个参数列表生成一个独立的测试用例，并将输入值作为参数传递给test_square函数，然后对比输出值和预期值。

2.2 参数化的高级用法

除了基本的参数列表外，pytest的参数化还支持更高级的用法，如使用fixtures、元组解包等。

2.2.1 使用fixtures

在pytest中，我们可以使用fixtures来准备测试环境，为测试用例提供依赖的数据或对象。参数化和fixtures结合使用可以更好地组织和管理测试用例。

下面是一个使用fixtures的参数化示例：

import pytest

@pytest.fixture
def setup_input():
    return 2
@pytest.mark.parametrize("input,expected", [(setup_input, 4), (3, 9), (4, 16)])
def test_square(input, expected):
    if callable(input):
        input = input()
    assert square(input) == expected

在上面的示例中，我们定义了一个setup_input的fixture，它返回了一个固定的输入值2。然后我们在参数化的测试用例中，使用setup_input作为输入值之一。

当我们运行上述代码时，pytest会首先调用setup_inputfixture来获取输入值2，然后将其作为参数传递给test_square函数。