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

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()

执行结果：