Python unittest如何实现setUp与tearDown针对每个subtest自动执行

原生unittest框架中，setUp()/tearDown()默认仅在每个test_开头的测试方法执行前后各运行一次，同个测试方法内的多个subTest不会重复触发前后置逻辑。

方案1：封装subTest上下文管理器（最小侵入，保留subTest原生特性） #

在自定义的测试基类中封装带前后置逻辑的子测试上下文，仅需替换原有subTest的调用语句即可，无需修改其他业务逻辑：

自定义基类实现 #

import unittest

class BaseTestCase(unittest.TestCase):
    def run_subtest(self, **kwargs):
        self.setUp()
        try:
            with self.subTest(**kwargs):
                yield
        finally:
            self.tearDown()

使用示例 #

class TestDemo(BaseTestCase):
    def setUp(self):
        # 你的前置逻辑
        self.conn = get_db_connection()
    
    def tearDown(self):
        # 你的后置逻辑
        self.conn.close()
    
    def test_multi_cases(self):
        test_params = [
            {"input": 1, "expect": 2},
            {"input": 2, "expect": 3},
            {"input": 3, "expect": 4}
        ]
        for item in test_params:
            # 替换原有 with self.subTest(...) 即可
            with self.run_subtest(input=item["input"]):
                res = item["input"] + 1
                self.assertEqual(res, item["expect"])

• 优势：完全兼容原生subTest的错误统计规则，单个子测试失败不会中断整个测试方法的执行，改造量极小。

方案2：参数化生成独立测试用例（符合unittest原生设计） #

如果你的子测试本质是同一逻辑的多参数校验，直接将每个参数生成为独立测试用例是更符合框架设计的方案，框架会自动为每个独立用例执行setUp/tearDown，无需手动封装。

无第三方依赖实现（动态生成测试方法） #

import unittest

class TestDemo(unittest.TestCase):
    def setUp(self):
        self.conn = get_db_connection()
    
    def tearDown(self):
        self.conn.close()
    
    def case_logic(self, input_val, expect_val):
        res = input_val + 1
        self.assertEqual(res, expect_val)

# 动态绑定测试方法
test_params = [
    {"input": 1, "expect": 2},
    {"input": 2, "expect": 3},
    {"input": 3, "expect": 4}
]
for idx, item in enumerate(test_params):
    def test_func(self, item=item):
        self.case_logic(item["input"], item["expect"])
    setattr(TestDemo, f"test_case_{idx+1}", test_func)

第三方参数化库实现（语法更简洁） #

使用parameterized库的装饰器实现，代码更简洁易读：

import unittest
from parameterized import parameterized

class TestDemo(unittest.TestCase):
    def setUp(self):
        self.conn = get_db_connection()
    
    def tearDown(self):
        self.conn.close()
    
    @parameterized.expand([
        (1, 2),
        (2, 3),
        (3, 4)
    ])
    def test_case(self, input_val, expect_val):
        res = input_val + 1
        self.assertEqual(res, expect_val)

• 优势：每个参数用例都会被unittest识别为独立测试用例，测试报告展示更清晰，无需手动维护前后置逻辑。

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