如何使用pytest fixture参数化创建自定义Person实例？

先看原固定属性的Fixture示例 #

假设你原来的代码是这样的：

class Person:
    def __init__(self, name="Alice", age=30):
        self.name = name
        self.age = age

import pytest

@pytest.fixture
def a_person():
    # 只能生成固定属性的Person实例
    return Person()

def test_default_person(a_person):
    assert a_person.name == "Alice"
    assert a_person.age == 30

方法1：让Fixture自身参数化（批量生成实例） #

如果需要批量测试多组固定参数的Person，可以给Fixture添加params属性，配合内置的request对象获取参数：

@pytest.fixture(params=[("Bob", 25), ("Charlie", 35), ("Diana", 45)])
def a_person(request):
    # request.param会依次取params列表里的每个元素
    name, age = request.param
    return Person(name=name, age=age)

def test_multiple_persons(a_person):
    # 这个测试会自动运行3次，分别对应三组参数
    assert isinstance(a_person, Person)
    # 可以针对不同实例做断言
    if a_person.name == "Bob":
        assert a_person.age == 25

方法2：结合@pytest.mark.parametrize自定义测试参数（更灵活） #

如果需要在单个测试用例里自定义Person的属性，用indirect=True告诉pytest把参数传递给Fixture，而不是直接作为测试参数：

方式1：传递元组参数

@pytest.fixture
def a_person(request):
    # 从request.param获取测试用例传递的参数
    name, age = request.param
    return Person(name=name, age=age)

@pytest.mark.parametrize("a_person", [("Eve", 28), ("Frank", 50)], indirect=True)
def test_custom_persons(a_person):
    assert isinstance(a_person, Person)
    assert a_person.age in [28, 50]

方式2：传递字典参数（更直观）

@pytest.fixture
def a_person(request):
    # 直接用字典拆包传递给Person构造函数
    return Person(**request.param)

@pytest.mark.parametrize("a_person", [
    {"name": "Grace", "age": 22},
    {"name": "Henry", "age": 33}
], indirect=True)
def test_person_with_kwargs(a_person):
    assert a_person.name in ["Grace", "Henry"]

兼容默认值的写法 #

如果需要Fixture同时支持默认实例和自定义实例，可以在内部判断是否有传递参数：

@pytest.fixture
def a_person(request):
    # 有参数就用参数，没参数就返回默认实例
    if hasattr(request, "param"):
        return Person(**request.param)
    return Person()

# 测试默认实例
def test_default_person(a_person):
    assert a_person.name == "Alice"

# 测试自定义实例
@pytest.mark.parametrize("a_person", [{"name": "Ivy", "age": 20}], indirect=True)
def test_custom_ivy(a_person):
    assert a_person.name == "Ivy"
    assert a_person.age == 20

核心知识点解释 #

• request：pytest内置的Fixture参数，用来获取当前测试的上下文，包括传递给Fixture的参数。
• indirect=True：当用@pytest.mark.parametrize给Fixture传参时，需要加这个参数，告诉pytest把参数传递给对应的Fixture处理，而不是直接作为测试用例的入参。
• Fixture的params属性：用来定义Fixture的批量参数，pytest会自动为每个参数生成一个测试实例。

内容的提问来源于stack exchange，提问作者Simon Ernesto Cardenas Zarate