如何在Django测试中实现参数化测试？

针对你在Django应用测试中遇到的参数化测试复用setUp初始化变量的问题，以下是两种可行的优化方案，解决之前思路的痛点：

方案1：使用Django自带的subTest解决循环测试的定位问题 #

直接利用Django TestCase内置的subTest上下文管理器，给循环中的每个测试用例添加标识，失败时能明确看到出错的参数：

from django.test import TestCase
from app.models import Example

class TestClass(TestCase):
    def setUp(self):
        # 在setUp中初始化需要复用的实例
        self.param1 = Example.objects.create(att="foo")
        self.param2 = Example.objects.create(att="bar")
        # 定义测试用例集合，关联实例与预期值
        self.test_cases = [
            (self.param1, "foo"),
            (self.param2, "bar"),
        ]

    def test_example(self):
        for param, expected in self.test_cases:
            # 用subTest标记每个子测试，失败时会显示对应的参数
            with self.subTest(param_id=param.id, expected=expected):
                self.assertEqual(param.att, expected)

优势： #

• 无需额外安装第三方库，依赖Django原生功能
• 子测试失败时会输出param_id和expected的值，精准定位出错用例
• 代码结构简单，易读易维护

方案2：结合parameterized库与setUpClass提升测试速度 #

针对之前parameterized库参数为None、测试速度慢的问题，改用setUpClass一次性初始化类属性，再通过属性名获取实例：

from django.test import TestCase
from app.models import Example
from parameterized import parameterized

class TestClass(TestCase):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        # 仅在类初始化时创建一次实例，所有测试方法复用
        cls.param1 = Example.objects.create(att="foo")
        cls.param2 = Example.objects.create(att="bar")

    @classmethod
    def tearDownClass(cls):
        super().tearDownClass()
        # 测试结束后清理数据
        cls.param1.delete()
        cls.param2.delete()

    # 参数化用例传递属性名而非直接传实例
    @parameterized.expand([
        ("param1", "foo"),
        ("param2", "bar"),
    ])
    def test_example(self, param_name, expected):
        # 通过属性名从类实例中获取初始化好的对象
        param = getattr(self, param_name)
        self.assertEqual(param.att, expected)

优势： #

• parameterized会生成独立的测试方法（如test_example_0_param1、test_example_1_param2），失败时直接定位到具体用例
• 实例仅创建一次，避免重复初始化数据，大幅提升测试速度
• 完全避免硬编码主键，复用setUpClass中创建的实例

之前思路的问题解析 #

• 思路1（循环）：未使用subTest标记子测试，导致失败时无法区分具体出错的用例
• 思路2（parameterized直接传实例）：装饰器在类定义阶段就会解析参数，此时setUp还未执行，所以参数值为初始的None
• 思路3（getattr）：使用了setUp而非setUpClass，每个测试方法都会重新创建实例，导致重复操作数据库，测试速度变慢

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