如何为Python unittest测试用例的setUp方法传递参数？

Python标准库的unittest框架中，setUp方法由框架自动调用，无法直接传入自定义参数。要实现每个测试用例对应不同的初始化种子数据，可通过自定义装饰器绑定参数到测试方法，再在setUp中读取该参数的方式实现，具体方案如下：

1. 实现自定义参数装饰器 #

编写一个装饰器，将需要的参数绑定到测试方法的自定义属性上：

def param(**kwargs):
    def decorator(test_func):
        # 将参数存储在测试方法的专属属性中
        setattr(test_func, "_test_params", kwargs)
        return test_func
    return decorator

2. 修改测试类的setUp方法 #

在setUp中，通过self._testMethodName获取当前执行的测试方法，取出绑定的参数后传入初始化函数：

import unittest

def init_db(user=None):
    # 模拟数据库初始化逻辑：根据传入的user创建对应种子数据
    print(f"初始化数据库，用户: {user}")

def test_something():
    # 模拟测试逻辑
    pass

class TestLogin(unittest.TestCase):
    def setUp(self) -> None:
        # 获取当前测试方法的对象
        test_method = getattr(self, self._testMethodName)
        # 取出绑定的参数，无参数时使用默认空字典
        params = getattr(test_method, "_test_params", {})
        # 传入参数完成数据库初始化
        init_db(params.get("user"))

    @param(user='jim')
    def test_login1(self):
        test_something()

    @param(user='bob')
    def test_login2(self):
        test_something()

if __name__ == '__main__':
    unittest.main()

补充说明 #

• 若测试用例无需自定义参数，直接省略@param装饰器即可，setUp会使用默认参数调用init_db。
• 该方案完全基于标准unittest框架实现，无需额外安装第三方依赖。

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