如何让Unittest用例共享类实例？解决setup重复执行问题

刚接触TDD踩这个坑太正常了！我当初用unittest的时候也纠结过这个问题——为啥每个测试方法都要跑一遍setup，实例不能共享呢？其实是unittest的默认机制在搞鬼：它会给每个test_开头的方法单独创建一个测试类的实例，所以setup自然会执行两次。针对你提到的workSq.convertToMonths类共享实例的需求，这里有几个实用的解决方案：

方案1：用类级别的初始化方法setUpClass #

这是最常用的方式，setUpClass是类方法，在整个测试类的所有测试方法执行前只运行一次，初始化的资源会绑定在测试类上，所有测试实例都能访问：

import unittest
from your_module import workSq

class TestWorkSheetOperations(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # 这里只初始化一次convertToMonths实例
        cls.converter = workSq.convertToMonths()
        # 提前调用getFileData，结果也存在类属性里共享
        cls.file_data = cls.converter.getFileData()

    def test_getFileData_validity(self):
        # 测试getFileData的结果，直接用共享的类属性
        self.assertIsNotNone(self.file_data)
        # 可以加更多断言，比如检查数据格式等

    def test_getWorkSheet_validity(self):
        # 使用共享的converter实例调用getWorkSheet
        worksheet = self.converter.getWorkSheet()
        self.assertIsNotNone(worksheet)
        # 比如检查工作表名称是否符合预期

    @classmethod
    def tearDownClass(cls):
        # 如果convertToMonths实例需要清理资源（比如关闭文件/连接），在这里处理
        # 示例：cls.converter.close()
        pass

这个方案的好处是逻辑清晰，资源初始化和清理都集中在类方法里，适合所有测试方法都需要共享同一个实例的场景。

方案2：把相关测试合并到同一个方法，用subTest区分场景 #

如果只是getWorkSheet和getFileData这两个方法需要共享实例，也可以把它们放到同一个测试方法里，用subTest标记不同的测试场景，这样整个方法只会创建一个测试实例：

import unittest
from your_module import workSq

class TestWorkSheetOperations(unittest.TestCase):
    def setUp(self):
        # 这里只执行一次，因为只有一个test方法
        self.converter = workSq.convertToMonths()

    def test_worksheet_and_file_data(self):
        # 测试getFileData
        with self.subTest(scenario="验证getFileData返回值"):
            file_data = self.converter.getFileData()
            self.assertIsNotNone(file_data)
        
        # 测试getWorkSheet，共享同一个converter实例
        with self.subTest(scenario="验证getWorkSheet返回值"):
            worksheet = self.converter.getWorkSheet()
            self.assertIsNotNone(worksheet)

这种方式的优点是不需要修改初始化逻辑，缺点是测试报告里会显示这是一个测试用例，但subTest会把每个场景的结果单独列出来，方便排查问题。

方案3：模块级别的全局资源（适合多测试类共享） #

如果你的测试模块里有多个测试类都需要用到同一个convertToMonths实例，可以用模块级别的初始化方法：

import unittest
from your_module import workSq

# 模块级别的全局变量，所有测试类共享
global_converter = None

def setUpModule():
    global global_converter
    global_converter = workSq.convertToMonths()

def tearDownModule():
    # 模块结束后清理资源
    # global_converter.close()
    pass

class TestFileData(unittest.TestCase):
    def test_getFileData(self):
        data = global_converter.getFileData()
        self.assertIsNotNone(data)

class TestWorkSheet(unittest.TestCase):
    def test_getWorkSheet(self):
        worksheet = global_converter.getWorkSheet()
        self.assertIsNotNone(worksheet)

这个方案适合跨测试类共享资源的场景，但要注意如果用多线程执行测试，可能会有线程安全问题，单线程测试完全没问题。

总结一下：优先用setUpClass，它是unittest官方推荐的类级共享资源方式；如果只是少量测试方法共享，用subTest更简单；跨类共享才考虑模块级别的资源。

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