python-在pytest中捕获异常后如何打印消息

假定此示例代码：

def test_foo():
    dict = load_dict()
    try:
        value = dict[foo][bar]
    except KeyError:
        print('missing foo or bar')

如果由于foo或bar不存在而引发KeyError,则测试不会失败,因为捕获了异常.如果我添加了一个SystemExit(1),它将失败,打印消息并显示所有回溯.

我的问题是,我如何告诉pytest如果发生KeyError,则意味着测试失败,因此我不需要引发SystemExit？

解决方法:

有一个功能pytest.fail明确使测试失败：

import pytest

def test_foo():
    d1 = {'foo': 'bar'}
    try:
        value = d1['baz']
    except KeyError as err:
        pytest.fail('this was unexpected: {}'.format(err))

但是,惯用的方式是使用pytest.raises上下文管理器来验证是否引发了异常,并使用一些方便的实用程序将其捕获以进行分析：

import pytest

def test_foo():
    d1 = {'foo': 'bar'}
    with pytest.raises(KeyError) as excinfo:
        value = d1['baz']
    assert excinfo.type == KeyError
    assert excinfo.match('baz')

查看文档以获取更多示例.如果您熟悉unittest,则pytest.raises是unittest.TestCase.assertRaises的项,而pytest.fail是unittest.TestCase.fail的项.