pytest之fixture参数化：轻松生成多种测试数据，提高测试效率

2023-10-15 09:16:51

fixture参数化

众所周知，pytest支持测试用例参数化，但它还支持fixture参数化，相比于测试用例参数化，fixture参数化使用起来更简便，并且可以生成多种测试数据，从而提高测试效率。

本文将详细介绍pytest中fixture参数化的使用方式，包括如何定义参数化fixture、如何使用fixture参数化测试用例以及一些常见的fixture参数化技巧。

如何定义参数化fixture

要定义一个参数化fixture，我们需要使用pytest.fixture()装饰器。在装饰器中，我们可以使用params参数来指定fixture的参数值。例如，以下代码定义了一个名为“my_fixture”的参数化fixture：

import pytest

@pytest.fixture(params=[1, 2, 3])
def my_fixture():
    return 10 * value

在这个例子中，my_fixture()函数将返回10、20和30这三个值。

如何使用fixture参数化测试用例

要使用fixture参数化测试用例，我们需要在测试用例中使用fixture作为参数。例如，以下代码使用my_fixture()参数化fixture来测试一个名为“test_my_fixture”的测试用例：

import pytest

@pytest.fixture(params=[1, 2, 3])
def my_fixture():
    return 10 * value

def test_my_fixture(my_fixture):
    assert my_fixture in [10, 20, 30]

在这个例子中，test_my_fixture()函数将分别执行三次，每次都会使用my_fixture()函数返回的一个值。因此，test_my_fixture()函数将分别断言10、20和30是否在[10, 20, 30]列表中。

一些常见的fixture参数化技巧

使用yield来生成fixture参数值

有时候，我们可能需要使用yield来生成fixture参数值。例如，以下代码使用yield来生成一个范围为1到10的fixture参数值：

import pytest

@pytest.fixture(params=range(1, 11))
def my_fixture():
    yield value

在这个例子中，my_fixture()函数将生成10个参数值，分别是1、2、3、4、5、6、7、8、9和10。

使用request.param来访问fixture参数值

在测试用例中，我们可以使用request.param来访问fixture参数值。例如，以下代码使用request.param来访问my_fixture()函数的参数值：

import pytest

@pytest.fixture(params=[1, 2, 3])
def my_fixture():
    return 10 * value

def test_my_fixture(my_fixture):
    assert my_fixture == request.param