如何让python unittest.TestCase的类下的setUp和tearDown只执行一次

转自:https://www.cnblogs.com/UncleYong/p/7076872.html

知道unittest单元测试框架的朋友应该都知道,

执行继承了unittest.TestCase的类下每个test开头的方法(就是用例)时,都会执行setUp和tearDown,如下面的例子所示: 


import unittest
 
class TestSetupTeardown(unittest.TestCase):
    def setUp(self):
        print('连接数据库成功...')
    def tearDown(self):
        print('关闭数据库。')
 
    def test_a(self):
        print('test_a')
 
    def test_b(self):
        print('test_b')
 
if __name__ == '__main__':
    unittest.main()

setUp连接数据库,tearDown关闭数据库,这样反复执行,无疑是会增加数据库服务器资源的损耗,且浪费时间

能不能只执行一次呢?

答案是肯定的,通过装饰器就可以实现

import unittest
 
class TestSetupTeardown(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print('连接数据库成功...')
    @classmethod
    def tearDownClass(cls):
        print('关闭数据库。')
 
    def test_a(self):
        print('test_a')
 
    def test_b(self):
        print('test_b')
 
if __name__ == '__main__':
    unittest.main()


### 解释 Python 单元测试代码 #### setUp tearDown 方法的作用 `setUp` 方法会在每一个测试方法执行前被自动调用,主要用于准备测试环境,比如初始化对象或者设置一些必要的参数。这有助于确保每个测试都在相同且独立的状态下运行[^5]。 相反,`tearDown` 方法则是在每个测试方法执行完毕后立即调用,通常用来清理资源或重置状态,防止不同测试之间相互影响。 对于只需要一次性创建的对象或资源,在级别上可以使用 `setUpClass` `tearDownClass` 来替代 `setUp` `tearDown`,这样可以在整个测试加载时只做一次准备工作,并在所有测试完成后统一处理善后事宜。 #### 测试方法 (test methods) 测试方法是以 `test_` 开头的方法名定义的函数,这些方法包含了具体的断言逻辑来验证程序的行为是否符合预期。例如: ```python import unittest class SimpleMathTestCase(unittest.TestCase): @classmethod def setUpClass(cls): cls.test_data = { 'add': [(1, 2), (3, 4)], 'subtract': [(7, 5), (-1, -9)] } def setUp(self): self.calculator = Calculator() def test_addition(self): for a, b in self.test_data['add']: result = self.calculator.add(a, b) expected_result = a + b self.assertEqual(result, expected_result, f"{a} + {b}") def test_subtraction(self): for a, b in self.test_data['subtract']: result = self.calculator.subtract(a, b) expected_result = a - b self.assertEqual(result, expected_result, f"{a} - {b}") def tearDown(self): del self.calculator if __name__ == '__main__': unittest.main() ``` 在这个例子中,`SimpleMathTestCase` 继承自 `unittest.TestCase` 并实现了两个测试方法:`test_addition()` `test_subtraction()`。这两个方法分别用于检验加法减法操作的结果是否正确。每次当某个特定的测试被执行之前都会先调用 `setUp()` 初始化计算器实例;而在该次测试结束后,则会调用 `tearDown()` 删除此实例以释放内存或其他资源。 此外,还展示了如何利用 `@classmethod` 装饰器配合 `setUpClass` 实现仅需构建一次的数据集共享给所有的测试案例使用。 #### 使用 TestSuite 组织多个测试 除了单个文件内的测试外,有时还需要组合来自不同源码位置上的若干组测试形成更大的集合来进行批量执行。这时就可以借助于 `TestLoader().loadTestsFromTestCase()` 或者直接手动添加到 `TestSuite` 对象里去实现这一点[^4]。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值