Python 测试框架之 Unittest & Pytest

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)，结果如下：

总结：

今天分享的内容是Python测试框架中常用的框架Unittest与Pytest相关内容，在自动化测试过程中可以进行实操，提高测试效率。
后面将分享使用allure来实现测试报告的可视化展示，敬请期待~