如何使用unittest为同一Python测试用例传入多组不同参数运行？

方案1：原生unittest实现（无需第三方依赖） #

你要的装饰器写法可以自己手写实现，完全兼容unittest原生逻辑，代码如下：

装饰器实现代码 #

import unittest
from functools import wraps

def inputs(*args):
    def decorator(test_func):
        if not hasattr(test_func, 'params'):
            test_func.params = []
        test_func.params.insert(0, args)
        @wraps(test_func)
        def wrapper(self, *inner_args, **inner_kwargs):
            for param in test_func.params:
                with self.subTest(case=param):
                    test_func(self, *param, *inner_args, **inner_kwargs)
        return wrapper
    return decorator

使用示例（完全匹配你要求的写法） #

class TestAddCalculation(unittest.TestCase):
    @inputs(1, 2, 3)
    @inputs(4, 5, 9)
    def test_add_result(self, a, b, c):
        self.assertEqual(a + b, c)

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

实现说明 #

• 利用unittest原生的subTest上下文管理器，单组参数失败不会终止其他参数的测试，会单独标记失败的参数组合
• 装饰器会自动收集所有@inputs传入的参数，遍历执行测试逻辑

方案2：第三方库实现（功能更丰富） #

如果需要支持参数命名、参数组合、独立用例统计等进阶功能，可以使用parameterized库，和你要求的写法高度契合。

安装方式 #

执行pip install parameterized完成安装。

使用示例 #

import unittest
from parameterized import parameterized

class TestAddCalculation(unittest.TestCase):
    # 写法1：单装饰器传入多组参数
    @parameterized.expand([
        (1, 2, 3),
        (4, 5, 9)
    ])
    # 写法2：多装饰器分开传参，和你要求的写法完全一致
    # @parameterized.expand([(1, 2, 3)])
    # @parameterized.expand([(4, 5, 9)])
    def test_add_result(self, a, b, c):
        self.assertEqual(a + b, c)

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

功能优势 #

• 每组参数会被识别为独立的测试用例，运行时单独统计成功/失败状态
• 支持自定义用例命名、参数过滤、参数笛卡尔积组合等进阶功能

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