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相关内容，在自动化测试过程中可以进行实操，提高测试效率。

---

学习资源分享

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走

这些资料，对于想【进阶自动化测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！凡事要趁早，特别是技术行业，一定要提升技术功底。希望对大家有所帮助…….