Python中有许多测试框架,但其中最受欢迎的就是PyTest。PyTest是一个强大而灵活的测试框架,它提供了许多先进的功能,可以让你的测试更加简洁、易读。

一、PyTest 简介

PyTest是一个开源的Python测试框架,用于编写简单而丰富的测试套件。它的功能特性包括:参数化、设置/拆解函数、假对象和钩子函数等。PyTest也被设计用来支持大型的测试集合,特别是适合进行回归测试。

二、PyTest 示例

接下来,我将通过以下10个示例,详细介绍如何使用PyTest。

1. 基础测试

PyTest的测试是通过编写函数并使用assert关键字进行断言的。以下是一个简单的测试示例

def test_addition():    assert 1 + 1 == 2

2. 测试异常

如果你希望一个特定的异常被抛出,可以使用pytest.raises上下文管理器:

import pytest
def 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 MagicMock
def test_with_mock():    my_mock = MagicMock(return_value=3)    assert my_mock(1, 2, 3) == 3    my_mock.assert_called_once_with(1, 2, 3)

7. 测试私有函数

虽然一般来说我们应该避免测试私有函数,但有时这也是有用的:​​​​​​​

def test__private_function():    from my_module import _private_function    assert _private_function(1, 2) == 3

在这个示例中,我们测试了一个名为_private_function的私有函数。这个函数接受两个参数并返回它们的和。我们使用了assert语句来确认返回的结果是否与预期的结果相符。注意,这个例子仅作为示例,实际上我们应尽量避免对私有函数进行单独测试,因为它们的实现可能会随着时间而改变。在大多数情况下,我们应该专注于测试公共接口,这样可以保证即使实现细节发生改变,测试仍然有效。

8. 跳过某些测试

如果有些测试你不希望执行,你可以使用pytest.mark.skippytest.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免费获取~(同行大佬一起学术交流,每晚都有大佬直播分享技术知识点)

软件测试面试文档

我们学习必然是为了找到高薪的工作,下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料,并且有字节大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。

在这里插入图片描述

全套资料获取方式:

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐