Python中有许多测试框架,但其中最受欢迎的就是PyTest。PyTest是一个强大而灵活的测试框架,它提供了许多先进的功能,可以让你的测试更加简洁、易读。
一、PyTest 简介
PyTest是一个开源的Python测试框架,用于编写简单而丰富的测试套件。它的功能特性包括:参数化、设置/拆解函数、假对象和钩子函数等。PyTest也被设计用来支持大型的测试集合,特别是适合进行回归测试。

二、PyTest 示例
接下来,我将通过以下10个示例,详细介绍如何使用PyTest。
1. 基础测试
PyTest的测试是通过编写函数并使用assert关键字进行断言的。以下是一个简单的测试示例
def test_addition():assert 1 + 1 == 2
2. 测试异常
如果你希望一个特定的异常被抛出,可以使用pytest.raises上下文管理器:
import pytestdef test_zero_division():with pytest.raises(ZeroDivisionError):1 / 0
3. 参数化测试
如果你有多组输入和期望的输出,可以使用pytest.mark.parametrize装饰器:
import pytest@pytest.mark.parametrize("a,b,expected", [(1, 2, 3), (4, 5, 9), (7, 8, 15)])def test_addition(a, b, expected):assert a + b == expected
4. 使用setup和teardown
对于需要前置和后置操作的测试,你可以使用setup和teardown函数:
def setup_function(function):if function == test_needs_resource_a:setup_resource_a()def teardown_function(function):if function == test_needs_resource_a:teardown_resource_a()def test_needs_resource_a():assert use_resource_a()
5. 使用fixtures
PyTest的fixtures功能允许你定义一些设置代码,这些代码可以被多个测试函数重用:
import pytest@pytest.fixturedef resource_a():setup_resource_a()yield use_resource_a()teardown_resource_a()def test_needs_resource_a(resource_a):assert resource_a
6. 使用mocks
你可以使用PyTest的unittest.mock模块创建假的对象,模拟函数或对象的行为:
from unittest.mock import MagicMockdef test_with_mock():my_mock = MagicMock(return_value=3)assert my_mock(1, 2, 3) == 3my_mock.assert_called_once_with(1, 2, 3)
7. 测试私有函数
虽然一般来说我们应该避免测试私有函数,但有时这也是有用的:
def test__private_function():from my_module import _private_functionassert _private_function(1, 2) == 3
在这个示例中,我们测试了一个名为_private_function的私有函数。这个函数接受两个参数并返回它们的和。我们使用了assert语句来确认返回的结果是否与预期的结果相符。注意,这个例子仅作为示例,实际上我们应尽量避免对私有函数进行单独测试,因为它们的实现可能会随着时间而改变。在大多数情况下,我们应该专注于测试公共接口,这样可以保证即使实现细节发生改变,测试仍然有效。
8. 跳过某些测试
如果有些测试你不希望执行,你可以使用pytest.mark.skip或pytest.mark.skipif来跳过这些测试:
import pytest@pytest.mark.skip(reason="This test is currently being debugged.")def test_in_debugging():assert 1 + 1 == 2
9. 使用pytest的命令行参数
PyTest提供了许多命令行参数,可以帮助你更有效地运行测试。例如,你可以使用-k参数来运行名称匹配某个模式的测试:
pytest -k "addition"
10. 分组测试
你可以使用pytest.mark来给你的测试分组:
import pytest@pytest.mark.slowdef test_slow_function():# This is a slow test...pass@pytest.mark.fastdef test_fast_function():# This is a fast test...pass
然后你可以只运行某个分组的测试:
pytest -m slow
11. 使用assert进行多种条件检查
PyTest的assert可以被用来进行多种条件检查。例如,我们可以使用assert来检查是否一个值在一个列表中:
def test_in_list():assert 1 in [1, 2, 3]
12. 使用xfail标记预期失败的测试
如果你知道某个测试目前不能通过,但你仍然希望执行它并记录结果,可以使用pytest.mark.xfail:
import pytest@pytest.mark.xfail(reason="This test is expected to fail.")def test_failing():assert 1 == 2
13. 使用capsysfixture捕获标准输出
如果你的函数会打印到标准输出,你可以使用capsysfixture来捕获这些输出,然后在你的测试中进行检查:
def test_print_output(capsys):print("Hello")captured = capsys.readouterr()assert captured.out == "Hello\n"
14. 使用tmp_pathfixture创建临时文件
如果你的测试需要创建临时文件,你可以使用tmp_pathfixture:
def test_create_file(tmp_path):d = tmp_path / "sub"d.mkdir()p = d / "hello.txt"p.write_text("Hello, World!")assert p.read_text() == "Hello, World!"
15. 与doctest结合
如果你的函数或者模块包含doctest,PyTest可以直接运行这些doctest:
def add(a, b):"""This function adds two numbers.>>> add(1, 2)3"""return a + b
然后你可以使用pytest --doctest-modules命令来运行这些doctest。
三、结论
以上就是PyTest的一些基本功能和示例。总的来说,PyTest是一个非常强大的测试框架,无论你是在编写简单的单元测试,还是复杂的集成测试,都可以使用PyTest来简化你的工作。通过学习和掌握PyTest,你将能够更有效地编写和维护你的测试代码。
END配套学习资源分享
最后: 为了回馈铁杆粉丝们,我给大家整理了完整的软件测试视频学习教程,朋友们如果需要可以自行免费领取 【保证100%免费】

加入我的软件测试交流群:110685036免费获取~(同行大佬一起学术交流,每晚都有大佬直播分享技术知识点)
软件测试面试文档
我们学习必然是为了找到高薪的工作,下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料,并且有字节大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。

全套资料获取方式:




















