Python 测试框架之 Unittest & Pytest

Wu_Candy

发布于 2022-07-04 17:07:48

3330

发布于 2022-07-04 17:07:48

文章被收录于专栏：无量测试之道

Unittest

Unittest是Python自带的一个单元测试框架

Unittest中包含了对一些常规的测试用例体系性的支持，主要包括：

test fixture
test case
test suite
test runner

创建.py文件，并选择文件类型如下图所示：

点击OK后默认生成的文件内容如下：

可以在此文件的内容基础上进行改造，写自己的测试case，改造后的内容如下：

import unittest  # 导入unittest模块

def number_sum(a, b):
    return a + b

"""
setUp():每个测试case运行之前运行
tearDown():每个测试case运行完之后执行
setUpClass():必须使用@classmethod 装饰器,  所有case运行之前只运行一次
tearDownClass():必须使用@classmethod装饰器, 所有case运行完之后只运行一次

"""

class MyTestCase(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("setupClass")

    def setUp(self):
        print("setUp")

    def test_sum_int(self):
        self.assertEqual(number_sum(1, 2), 3)
        self.assertEqual(number_sum(100, 300), 400)

    def test_sum_number(self):
        self.assertEqual(number_sum(1.1, 2.2), 3.3)

    def tearDown(self):
        print("tearDown")

    @classmethod
    def tearDownClass(cls):
        print("tearDownClass")


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

以上代码是标准的unittest测试框架中的基础结构。

Pytest

Pytest中定义测试用例主要有三种方式：

兼容Unittest，已有的Unittest库和文件都可以直接进行调用
可以基于最简单的类进行定义，如果一个类里面没有初始化方法并且以Test开头，系统便会认为这是一个测试用例
可以直接定义测试函数来定义测试用例

安装Pytest： 1、使用命令：pip install pytest 2、在Pycharm编译器里的配置中添加，如下截图所示

Pytest安装成功后，将之前默认的runner由unittest修改为：pytest，修改步骤如下截图所示：

将Pytest配置成默认的runner后，新建一个.py文件为：test_demos.py 内容如下：

点击如上图所示的绿色小箭头，可查看到第一行提示：使用Pytest来运行test_demos.py文件，运行结果正常

将Python集成工具由Unittest切换成Pytest后，再来运行之前写的使用Unittest框架编写的代码，如图所示：

类名前的绿色小箭头提示可以使用：Pytest来运行test_unittest.py文件，Pytest兼容Unittest，可以直接使用Pytest直接调用Unittest库和文件。运行结果如下所示：

Pytest的用例识别规则：

Test类包含的所有test_方法
不在class中的所有test_*函数
类中不能初始化方法
断言使用基本的assert即可

文件范围：

test_*.py
*_test.py

用例执行顺序的控制方法如下图所示：

完整代码如下：

import pytest #导入pytest模块

def func(x):
    return x + 1

def test_answer():
    assert func(3) == 5


class TestFunc:
    @classmethod
    def setup_class(self):
        print("setup_class")

    def setup(self):
        print("setup")

    @pytest.mark.fail
    def test_answer1(self):
        print("test_answer1")
        assert func(3) == 5

    @pytest.mark.success
    @pytest.mark.parametrize("input,expect",{
        (5, 6),
        (7, 8),
        (0, 1),
        (2, 2)
    })
    def test_answer2(self, input, expect):
        print("test_answer2")
        assert func(input) == expect

    def test_answer3(self):
        print("test_answer3")
        assert func(7) == 8

    def teardown(self):
        print("teardown")

    @classmethod
    def teardown_class(cls):
        print("teardown_class")

‍

Pytest支持分组： 1、@pytest.mark.webtest 2、@pytest.mark.sec 3、Pytest -m “webtest and not sec”

调用方法如下图所示：

在Terminal里输入命令：pytest -m fail来实现只执行指定的用例case的目的。参数化用例的使用方法的实例代码如下所示：

@pytest.mark.parametrize("input,expect",{
    (5, 6),
    (7, 8),
    (0, 1),
    (2, 2)
})
def test_answer2(self, input, expect):
    print("test_answer2")
    assert func(input) == expect

参数化用例方法可以达到代码重复利用的效果。单独来执行test_answer2这个用例方法时会发现实际共执行了4次该方法，input与expect的值分别是参数化中的(5, 6),(7, 8),(0, 1),(2, 2)，结果如下：