单元测试|Unittest setup 前置初始化和 teardown 后置操作

2023-09-15
北京
本文字数：655 字
阅读完需：约 2 分钟

Unittest setup 前置初始化和 teardown 后置操作

针对类中每个函数执行时进行前置后置的操作

setUp(self)：测试前的初始化操作

tearDown(self)：测试后的操作

代码示例：

import unittest  # 导入unittest模块
class test_case(unittest.TestCase):  # 集成TestCase类
    def setUp(self):        print("我是setUp方法，进行测试前的初始化工作")    def tearDown(self) -> None:  # 表示该方法没有返回值        print("我是tearDown方法，执行测试后的清除工作")
    # 必须以test开头的测试方法才能被执行    def test_aaa(self):        print("我是case: test_aaa")    def test_bbb(self):        print("我是case: test_bbb")    def test_ccc(self):        print("我是case: test_ccc")

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

复制代码

针对整个类进行前置后置的操作

setUpClass(cls)：测试前的初始化操作

tearDownClass(cls)：测试后的操作


import unittest

class TestCase(unittest.TestCase):
    @classmethod    def setUpClass(cls) -> None:        print('setUpClass')
    @classmethod    def tearDownClass(cls) -> None:        print('tearDownClass')
    def setUp(self) -> None:        print('setUp')
    def tearDown(self) -> None:        print('tearDown')
    def test_01(self):        print('test_01')
    def test_02(self):        print('test_02')

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