Pytest单元测试框架
1、引言
小屌丝：鱼哥，最近为啥不讲一讲测试开发的技术，例如…
小鱼：我去~ ~ 你们这是商量好的，一起来提这个事情？
小屌丝：为啥这么说呢？
小鱼：因为最近有妹子跟我抱怨，说为啥不讲讲测试开发的知识，都是整的一些开发的…
小屌丝：难道不是吗？
小鱼：兄弟，我也有苦衷啊~ ~
小屌丝：别整那些没用的，今天就整一个吧！！！
小鱼：整整整~~
小鱼的博客讲过unittest框架，也有很多的例子在里面，不了解的话，可以看一下。
而今天，我们还是来分享框架，只不过这次的是单元测试框架：Pytest。
小鱼记得写过一篇测试框架的组成，
这里面包含了功能、性能、自动化等各个阶段的测试框架组成已经必须会的知识，
不了解的话，可以看一下小鱼的这篇博文《测试开发必会知识之测试框架组成》。
先截个图，看下：
在这里，咱俩就单纯的聊聊Pytest这个单元测试框架，别的咱们不聊！！
2、Pytest简介
2.1 入门阶段
2.1.1 运行测试用例
一切的理论，都是为了实战准备的，
所以，我们第一步，先上例子，直接实战！
编写简单的测试用例，文件名称为"test_sample.py"
# pytest test_sample.py
def inc(x):
return x + 1
def test_answer():
assert inc(3) == 5
这两个方法都非常的简单，
- test_answer()对目标函数inc进行断言；
- pytest test_sample.py 运行测试用例
我们看向运行结果
测试结果显示，运行了一个测试用例，结果是红色，表示失败。错误信息显示，在代码的第7行抛出AssertionError。可以更改一下代码，让测试用例绿色通过。
这个测试用例涉及pytest的3个简单的规则:
- 测试模块以 test_ 前缀命名
- 测试用例(函数)同样以 test_ 前缀命名
- 结果判断使用 assert 断言即可
2.1.2. 异常处理
捕获异常，及异常处理，是每段代码中必要条件，
当然，pytest也支持异常的捕获，
方法为： 使用 with + pytest.raises 捕获目标函数的异常:
# -*- coding:utf-8 -*-
# @Time : 2021-09-05
# @Author : Carl_DJ
import pytest
def f():
raise SystemExit(1)
def test_mytest():
#捕获异常
with pytest.raises(SystemExit):
f()
2.1.3 测试类
当然，pytest也支持测试类，
测试类的作用：用来做测试用来分组
# -*- coding:utf-8 -*-
# @Time : 2021-09-05
# @Author : Carl_DJ
class TestClass:
#test_开头
def test_one(self):
x = "this"
assert "h" in x
#test_开头
def test_two(self):
x = "hello"
assert hasattr(x, "check")
这里，都是以test_开头， 跟unittest都是一样。
如果不是test_开头，则无法被调用的。
2.1.4 自动运行测试脚本
如果一个suit文件夹有多个测试用例脚本，
我们只需要输入一个 pytest，即可运行全部测试脚本。
如下图
这就是开启了懒人模式。
2.2 进阶技巧
2.2.1 parametrize
俗话说，代码不参数，变参泪两行！
所以，能参数的时候，就尽量参数，不管重构苦不苦。
我们先看下parametrize的用法，如下：
在测试用例的前面加上：
@pytest.mark.parametrize(“参数名”,列表数据)
参数名：用来接收每一项数据，并作为测试用例的参数。
列表数据：一组测试数据。
看例子
不添加parametrize，看看咋写测试用例
def test_eval():
assert eval("3+5") == 8
assert eval("'2'+'4'") == "24"
assert eval("6*9") == 54
看着很麻烦，
我们再用一下parametrize优化一下，看看是否简洁很多
#使用parametrize对测试用例参数化
@pytest.mark.parametrize("test_input,expected", [
("3+5", 8),
("'2'+'4'", "24"),
("6*9", 54)
])
def test_eval_1(test_input, expected):
assert eval(test_input) == expected
看了这段代码，就很简洁了。
我们在来看看，parametrize做了什么。
- 先调整测试函数的参数为输入和期望；
- 然后在parametrize填写参数值；
- 运行时候会自动进行函数参数赋值。
这样再增加测试条件，不需要改动test_eval_1的函数体, 增加条件数组就可以了。
2.2.2. mark
mark就是一个标签，标记那些测试用例执行，哪些不执行。
#标签
@pytest.mark.slow
def test_mark():
print("test mark")
# 模拟运行很耗时的测试用例
time.sleep(10)
assert 5 == 5
然后目录下增加 pytest.ini 文件，对pytest进行配置:
[pytest]
markers =
slow: marks tests as slow (deselect with '-m "not slow"')
使用下面命令可以跳过标记的函数，加快测试速度：
pytest test_sample.py -m "not slow"
也可以仅仅运行标记的函数
pytest -m slow
2.2.3 fixture
fixture 就类似于unittest的 setup/teardown，但是功能比这个强大一些。
举个例子
# -*- coding:utf-8 -*-
# @Time : 2021-09-05
# @Author : Carl_DJ
import pytest
#设置fixture
@pytest.fixture
def first_entry():
return "a"
#设置fixture
@pytest.fixture
def second_entry():
return 2
#设置fixture
@pytest.fixture
def order(first_entry, second_entry):
return [first_entry, second_entry]
#设置fixture
@pytest.fixture
def expected_list():
return ["a", 2, 3.0]
def test_string(order, expected_list):
order.append(3.0)
# 断言
assert order == expected_list
当然了，fixture还可以嵌套，order嵌套了first_entry和second_entry。
小屌丝：那测试数据库写入的用例，需要一个数据库链接，怎么办呢？
小鱼这个，也不难，fixture也可以搞定。
在测试用例目录编写conftest.py
@pytest.fixture
def database_connection():
# coding...
...
2.2.4 plugin&&hook
可以编写pytest的插件plugin和hook对pytest进行扩展。
先创建一个目录a，然后再目录a中创建conftest.py和test_sub.py 两个文件。
#在目录a下创建conftest.py
def pytest_runtest_setup(item):
# 在目录a下运行每个用例
print("setting up", item)
# 在目录a下创建test_sub.py
def test_sub():
pass
使用 pytest a/test_sub.py --capture=no 会加载我们编写的pluging和hook，在console中可以看到下面字样:
...
a/test_sub.py setting up <Function test_sub>
敲黑板：
使用pytest_runtest_setup可以实现测试框架中的setup类似功能。
3、总结
以上就是小鱼总结的一些pytest常用的功能，是不是也很简单呢。
我们在回顾一下，今天都讲了那些姿势 知识！
- 测试目录一般使用 tests 命名和src同层级
- 测试模块使用 test_ 前缀
- 测试类使用 Test 前缀，不需要继承其它父类
- 测试用例也使用 test_ 前缀
- 可以使用parametrize进行参数化处理
- 可以使用mark给测试用例加标签
- 可以使用fixture模拟测试条件
- 使用pytest.ini文件对pytest进行配置
- 可以编写插件和hoo对pytest扩展