pytest 测试框架学习（11）：pytest.raises

Mokwing

发布于 2020-09-08 11:15:55

2.6K0

发布于 2020-09-08 11:15:55

文章被收录于专栏：Mokwing MokwingMokwing

pytest.raises

含义

raises：在断言一些代码块或者函数时会引发意料之中的异常或者其他失败的异常，导致程序无法运行时，使用 raises 捕获匹配到的异常，可以继续让代码正常运行。源码：

expected_exception：预期的异常（可以输入一个元组）。

使用

预期内异常

import pytest

def test_raises():
    with pytest.raises(ZeroDivisionError):
        2 / 0
    assert eval("1 + 2") == 3

raises 可以捕获到该异常，并继续下面断言代码。

如果我们不知道预期异常的是什么，我们可以使用 match 和 raise 进行自定义异常

import pytest

def exc(x):
    if x == 0:
        raise ValueError("value not 0 or None")
    return 2 / x

def test_raises():
    with pytest.raises(ValueError, match="value not 0 or None"):
        exc(0)
    assert eval("1 + 2") == 3

match 还可以使用正则表达式进行匹配异常：

with pytest.raises(ValueError, match=r"value not \d+$"):
	raise ValueError("value not 0")

Tips: 使用正则时，等号后面有个 r 。

在捕获异常后，可以从上下文管理器中获取异常的一些详细信息，可以辅助我们更好的去断言。

import pytest

def exc(x):
    if x == 0:
        raise ValueError("value not 0")
    return 2 / x

def test_raises():
    with pytest.raises(ValueError) as exec_info:
        exc(0)

    print("exec_info.type = ", exec_info.type)
    print("exec_info.value.args = ", exec_info.value.args)

    assert exec_info.type == ValueError
    assert exec_info.value.args[0] == "value not 0"

执行结果：

注意：官方提示， raise 的异常应该是当前代码块最后一行，如果在其后面还有代码，那么将不会被执行。比如：

修改后代码

参数化 pytest.mark.parametrize 使用参数化的话，可能会存在一部分用例可能会抛出异常，一部分可能会没有异常导致失败。如果想要其正常执行，则需要一个上下文管理器。这里官方指导使用 does_not_raise 例：

from contextlib import contextmanager
import pytest

@contextmanager
def does_not_raise():
    yield

@pytest.mark.parametrize(
    "example_input,expectation",
    [
        (3, does_not_raise()),
        (2, does_not_raise()),
        (1, does_not_raise()),
        (0, pytest.raises(ZeroDivisionError)),
    ],
)
def test_division(example_input, expectation):
    """Test how much I know division."""
    with expectation:
        assert (6 / example_input) is not None

注意：如果使用的python版本不一致，导入对应依赖也是不一样的。 python3.7+： from contextlib import nullcontext as does_not_raise python3.3+： from contextlib import ExitStack as does_not_raise 或者使用 pip install contextlib2 from contextlib2 import nullcontext as does_not_raise

以下两个为拓展形式，但官方建议不使用。因为上面形式的可读性会更强。

lambda 表达式

import pytest

pytest.raises(ZeroDivisionError, lambda: 1/0)

指定函数，并调用可用参数

import pytest

def exc(x):
    return 2 / x

def test_raises_param():
    pytest.raises(ZeroDivisionError, exc, x=0)
    assert 1 == 1

说明：本篇参考官网并加入自己些许理解翻译而来，觉得有用，可以点赞和赞赏哦(^ v ^)，谢谢支持；如果有不足地方，可留言评论。后续将继续更新。

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2020-07-10 ，如有侵权请联系 cloudcommunity@tencent.com 删除

import

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度