Pytest学习（三） - setup和teardown的使用

一、前言

从文章标题可以看出，就是初始化和释放的操作，根据我的java习惯来学习pytest，个人感觉没差太多，理解上也不是很难。

哦，对了，差点跑题了，这个框架是基于Python语言的，在学习的时候难免总会用java的类比思想来学习，下面言归正传哈。

我们还从 unittest与pytest来对比学习吧

二、unittest用法

unittest有两个前置方法，两个后置方法，分别是:

• setup()
• setupClass()
• teardown()
• teardownClass()

个人始终觉得unittest和Junit像极了。

三、pytest用法

当然，Pytest也提供了类似setup、teardown的方法，分别是：

• 模块级别：setup_module、teardown_module
• 函数级别：setup_function、teardown_function，不在类中的方法
• 类级别：setup_class、teardown_class
• 方法级别：setup_method、teardown_method
• 方法细化级别：setup、teardown

我总感觉学习pytest像是在学习testng一样，难道是我的错觉吗，啊啊啊啊，不能吧。

四、unittest示例

unittest的setupClass和teardownClass，需要配合@classmethod装饰器一起使用，也就是我们java说的注解呀，这块是翻译给java学Python的同学的，可忽略哈。

示例代码如下：

# -*- coding: utf-8 -*-
# @Time    : 2020/10/21 20:09
# @Author  : longrong.lang
# @FileName: test_setup_teardown_unittest.py
# @Software: PyCharm
# @Cnblogs ：https://www.cnblogs.com/longronglang
'''
unittest代码示例
'''
import unittest

class TestUnitTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("所有用例执行前执行")

    def setUp(self):
        print("每个用例开始前执行")

    def tearDown(self):
        print("每个用例结束后执行")

    @classmethod
    def tearDownClass(cls):
        print("所有用例执行后执行")

    def testA(self):
        '''用例A'''
        print("用例A执行了")
        self.assertEquals(1, 1)

    def testB(self):
        '''用例B'''
        print("用例B执行了")
        self.assertTrue(True)

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

执行结果

可以看出执行顺序为：

setUpClass
setUp
testA
tearDown
setUp
testB
tearDown
tearDownClass
用例之间按用例名称ASCII码的顺序加载，数字与字母顺序为0~9，A~Z，a~z,  所以testA会在testB之前运行。

五、pytest示例

函数级的setup_function、teardown_function只对函数用例生效，而且不在类中使用

依旧还是把类和函数都有的情况放在一起，示例代码如下：

# -*- coding: utf-8 -*-
# @Time    : 2020/10/21 20:27
# @Author  : longrong.lang
# @FileName: test_setup_teardown_pytest.py
# @Software: PyCharm
# @Cnblogs ：https://www.cnblogs.com/longronglang
'''
pyetest示例
'''

import pytest

def setup_module():
    print("setup_module():在模块最之前执行，且只执行一次")

def teardown_module():
    print("teardown_module：在模块之后执行，且只执行一次")

def setup_function():
    print("setup_function():每个方法之前执行")

def teardown_function():
    print("teardown_function():每个方法之后执行")

def test_1():
    print("正在执行用例1")
    x = "this"
    assert 'h' in x

def test_2():
    print("正在执行用例2")
    assert 1 == 1

class TestClass(object):

    def setup_class(self):
        print("setup_class(self)：每个类之前执行一次,只执行一次")

    def teardown_class(self):
        print("teardown_class(self)：每个类之后执行一次,只执行一次")

    def test_A(self):
        print("正在执行用例A")
        x = "this"
        assert 'h' in x

    def test_B(self):
        print("正在执行B")
        assert 1 == 1

if __name__ == "__main__":
    pytest.main(["-q", "test_setup_teardown_pytest.py"])

执行结果

可以看出来，互不影响，执行顺序为：

setup_module()
setup_function()
test_1
teardown_function()
setup_function()
test_2
teardown_function()
setup_class(self)
test_A
test_B
teardown_class(self)
teardown_module

main方法中的-q，为pytest打印测试用例的执行结果级别。

如不清楚，请移步到《Pytest学习（一）- 入门及基础》。