本文共 2754 字，大约阅读时间需要 9 分钟。

PyTest 测试框架（二）：Pytest 的 setup/teardown 方法

本文作者：hiyo

AlertDialog

Pytest是一个强大的测试框架，它支持基于xUnit风格的测试结构。setup()和teardown()方法用于初始化和清理测试环境，这对于确保测试用例的独立性至关重要。Pytest的setup/teardown方法包括模块级(setup_module/teardown_module)、函数级(setup_function/teardown_function)、类级(setup_class/teardown_class)及方法级(setup_method/teardown_method)等不同的作用范围。

模块级

在模块中，Pytest会自动执行setup_module方法，作为模块中的第一个测试用例前所做的准备工作；而teardown_module方法则在模块中的最后一个测试用例之后运行，用于清理模块级别的资源。以下是一个模块级的示例：

def setup_module():    print(" 初始化... ")def teardown_module():    print(" 清理... ")class Test_Demo:    def test_case1(self):        print(" 开始执行测试用例1 ")        assert 1 + 1 == 2    def test_case2(self):        print(" 开始执行测试用例2 ")        assert 2 + 8 == 10    def test_case3(self):        print(" 开始执行测试用例3 ")        assert 99 + 1 == 100

以下是执行结果的一部分：

模块初始化... PASSED [33%] 开始执行测试用例1 PASSED [66%] 开始执行测试用例2 PASSED [100%] 开始执行测试用例3 模块清理...

函数级

对于函数级来说，Pytest会在每个测试函数前调用setup_function，测试函数执行之后再调用teardown_function。这使得每个测试用例都能独立维护自身的环境。以下是一个函数级的示例：

def setup_function():    print(" 初始化... ")def teardown_function():    print(" 清理... ")def test_case1():    print(" 开始执行测试用例1 ")    assert 1 + 1 == 2def test_case2():    print(" 开始执行测试用例2 ")    assert 2 + 8 == 10def test_case3():    print(" 开始执行测试用例3 ")    assert 99 + 1 == 100

执行结果示例：

test_setup_teardown2.py::test_case1 初始化... PASSED [33%] 开始执行测试用例1 清理... test_setup_teardown2.py::test_case2 初始化... PASSED [66%] 开始执行测试用例2 清理... test_setup_teardown2.py::test_case3 初始化... PASSED [100%] 开始执行测试用例3 清理...

类级

在类级中，Pytest会在类的前后各执行一次setup_class和teardown_class方法。这样可以为整个类级的测试用例提供统一的环境准备和清理流程。以下是一个类级的示例：

class Test_Demo:    def setup_class(self):        print(" 初始化... ")    def teardown_class(self):        print(" 清理... ")    def test_case1(self):        print(" 开始执行测试用例1 ")        assert 1 + 1 == 2    def test_case2(self):        print(" 开始执行测试用例2 ")        assert 2 + 8 == 10    def test_case3(self):        print(" 开始执行测试用例3 ")        assert 99 + 1 == 100

执行结果示例：

初始化... PASSED [33%] 开始执行测试用例1 PASSED [66%] 开始执行测试用例2 PASSED [100%] 开始执行测试用例3 清理...

方法级

在方法级，Pytest会为每个测试方法执行setup_method和teardown_method。这与类级的效果相似，只不过是针对每个单独的测试方法。以下是一个方法级的示例：

class Test_Demo:    def setup_method(self):        print(" 初始化... ")    def teardown_method(self):        print(" 清理... ")    def test_case1(self):        print(" 开始执行测试用例1 ")        assert 1 + 1 == 2    def test_case2(self):        print(" 开始执行测试用例2 ")        assert 2 + 8 == 10    def test_case3(self):        print(" 开始执行测试用例3 ")        assert 99 + 1 == 100

执行结果示例：

初始化... PASSED [33%] 开始执行测试用例1 清理... 初始化... PASSED [66%] 开始执行测试用例2 清理... 初始化... PASSED [100%] 开始执行测试用例3 清理...

总结

Pytest的setup/teardown方法为开发者提供了高效且灵活的方式来管理测试环境的初始化和清理。从模块级到函数级、类级，再到方法级，Pytest支持多层级的setup/teardown操作，确保每组测试用例都能在独立的环境下运行。这使得Pytest成为现代测试开发中的一个非常强大的工具。

转载地址：http://qysmz.baihongyu.com/