前言

单元测试框架基本上都具有setup和teardown的功能；

setup用来实现用例执行前的一些操作(如：数据准备、打开浏览器等操作)；

teardown用来实现用例执行完成之后的一些操作(如：数据清理、关闭浏览器等操作)。

pytest作为一款强大的测试框架同样也有setup和teardown语法。

• 函数级（setup_function/teardown_function）
• 方法级（setup_method/teardown_method）
• 类级（setup_class/teardown_class）
• 模块级（setup_module/teardown_module）

一、函数级（setup_function/teardown_function）

用于函数形式的测试用例，每个测试用例执行前调用一次setup_function，执行结束后调用一次teardown_function，如下：

def setup_function():
    print("准备条件")


def teardown_function():
    print("销毁条件")


def test_001():
    print("执行第一个测试用例")


def test_002():
    print("执行第二个测试用例")

运行结果：

二、方法级（setup_method/teardown_method）

用于类中方法形式的测试用例，每个测试用例执行前调用一次setup_method，执行结束后调用一次teardown_method，如下：

class TestOne:
    def setup_method(self):
        print("准备条件")

    def teardown_method(self):
        print("销毁条件")

    def test_001(self):
        print("执行第一个测试用例")

    def test_002(self):
        print("执行第二个测试用例")

运行结果：

三、类级（setup_class/teardown_class）

用于类中，在一个类中只调用一次，用例开始前调用一次setup_class，用例结束后调用一次teardown_class，如下：

class TestOne:
    def setup_class(self):
        print("准备条件")

    def teardown_class(self):
        print("销毁条件")

    def test_001(self):
        print("执行第一个测试用例")

    def test_002(self):
        print("执行第二个测试用例")

运行结果：

四、模块级（setup_module/teardown_module）

用于当前模块，当前模块执行前调用一次setup_module，当前模块执行结束后调用一次teardown_module，如下：

def setup_module():
    print("开始执行当前模块")


def teardown_module():
    print("结束执行当前模块")


class TestOne:
    def setup_class(self):
        print("准备条件")

    def teardown_class(self):
        print("销毁条件")

    def test_001(self):
        print("执行第一个测试用例")

    def test_002(self):
        print("执行第二个测试用例")