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扩展

关于pytest更多的内容，可以持续关注小鱼的博客。

我是小鱼：

• CSDN 博客专家；
• 阿里云 专家博主；
• 51CTO 博客专家；
• 51认证讲师；
• 金牌面试官；
• 职业规划师；
  

关注我，带你学习更多更有趣的Python知识。