zl程序教程

您现在的位置是:首页 >  Python

当前栏目

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

2023-02-26 10:15:14 时间

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()
在这里插入图片描述

执行结果:

在这里插入图片描述