如何在Python所有单元测试前仅执行一次预热函数？

在Python的unittest框架中，setUp()方法会在每个测试方法执行前被调用一次，这不符合你只初始化一次的需求。你可以通过以下两种方式实现仅在所有测试前执行一次预热操作：

方法一：使用setUpClass类方法 #

setUpClass是unittest.TestCase提供的类方法，会在整个测试类的所有测试方法执行前仅运行一次，需要用@classmethod装饰器修饰，参数为cls（代表测试类本身）。

修改后的代码示例：

import unittest

def addnumbers(a,b,c):
    return a+b+c

class Testaddnumbers(unittest.TestCase):
    
    @classmethod
    def setUpClass(cls):
        cls.a = 10  # 设置为类属性，所有测试实例共享
        print(f'a set up to {cls.a}')
        
    def test1(self):
        self.assertEqual(addnumbers(self.a,2,3), 15)

    def test2(self):
        self.assertEqual(addnumbers(self.a,5,6), 21)
        
if __name__ == '__main__':
    unittest.main()

运行这段代码会发现，print语句只会输出一次，test1和test2共享同一个cls.a的值。

方法二：使用模块级别的setUpModule函数 #

如果你需要在整个模块的所有测试类执行前只运行一次预热操作，可以定义模块级别的setUpModule函数：

import unittest

def addnumbers(a,b,c):
    return a+b+c

# 模块级预热函数，所有测试执行前仅运行一次
def setUpModule():
    global a
    a = 10
    print(f'a set up to {a}')

class Testaddnumbers(unittest.TestCase):
    
    def test1(self):
        self.assertEqual(addnumbers(a,2,3), 15)

    def test2(self):
        self.assertEqual(addnumbers(a,5,6), 21)
        
if __name__ == '__main__':
    unittest.main()

这种方式适合多个测试类共享同一预热数据的场景。

注意：如果预热操作涉及资源初始化（如数据库连接、文件打开），记得对应实现tearDownClass（类级清理）或tearDownModule（模块级清理）函数，避免资源泄漏。

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