Pytest单元测试框架有多简单，几分钟就能入门

Pytest单元测试框架

1、引言

在这里，咱俩就单纯的聊聊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更多的内容， 可以持续关注， 也可以阅读官方文档。

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

---

在此特意为大家准备了一份13G的超实用干货学习资源，涉及的内容非常全面，涵盖功能测试、Python编程语言，接口测试、UI自动化测试、性能测试......包括软件学习路线图，50多天的上课视频、16个突击实战项目，80余个软件测试用软件，37份测试文档，70个软件测试相关问题，40篇测试经验级文章，上千份测试真题分享,还有2022软件测试面试宝典，还有软件测试求职的各类精选简历，希望对大家有所帮助…..关注下方公众号免费获取~