pytest测试函数内部使用parametrize无效，如何正确复用测试逻辑？

这种在测试函数内部用with语句使用pytest.mark.parametrize的写法完全不可行，pytest根本不支持这种用法。

原因 #

pytest.mark.parametrize是一个装饰器，它的作用时机是测试用例收集阶段（也就是pytest扫描测试文件、解析测试函数的时候），而测试函数内部的代码是在测试执行阶段才会运行。当你在函数里用with包裹parametrize时，pytest已经完成了测试用例的收集工作，根本不会识别到这个参数化配置，自然不会生效。

另外你原代码里还有个低级错误：即使parametrize能生效，你发起请求时还是写死了/api/color_names/{color.name}/，根本没用到参数化的endpoint变量，这样也达不到测试两个接口的目的。

正确的优化方式 #

直接把parametrize作为装饰器作用在测试函数上，通过模板字符串来动态生成接口地址：

import pytest
from your_app.models import Color

@pytest.mark.parametrize("endpoint_template", [
    "/api/colors/{pk}/",
    "/api/color_names/{name}/",
])
def test_get_color(client, endpoint_template):
    color = Color.objects.create(name="Green")
    # 替换模板里的占位符，生成实际接口地址
    endpoint = endpoint_template.format(pk=color.pk, name=color.name)
    response = client.get(endpoint)
    
    assert response.status == 200
    assert response.json() == {"id": color.pk, "name": color.name}

这样pytest会在收集用例时自动生成两个独立的测试：一个测试通过ID查询颜色，另一个通过名称查询，完全复用了创建颜色、断言逻辑的代码，达到了减少重复的目的。

补充说明 #

如果你的接口地址规则更复杂，也可以用parametrize传递生成endpoint的函数，或者结合fixture来复用颜色对象，但上面的模板方式是最直接简洁的方案。

内容的提问来源于stack exchange，提问作者Johnny Metz