在 pytest 参数化中使用类型错误消息

Question

我有一个函数会在满足某些条件时引发 TypeError。

def myfunc(..args here...):
    ... 
    raise TypeError('Message')

我想使用 pytest 参数化测试这条消息。

但是，因为我正在使用其他参数，所以我也希望有这样的设置：

testdata = [
        (..args here..., 'Message'), # Message is the expected output
    ]

    @pytest.mark.parametrize(
        "..args here..., expected_output", testdata)
    def test_myfunc(
       ..args here..., expected_output):

        obs = myfunc()
        assert obs == expected_output

简单地将Message 作为参数化测试数据中的预期输出，给我一个失败的测试。

Answer 1

您不能期望消息错误是myfunc 的正常输出。为此有一个特殊的上下文管理器 - pytest.raises。

For example，如果你想知道一些错误和它的消息

def test_raises():
    with pytest.raises(Exception) as excinfo:   
        raise Exception('some info')   
    assert str(excinfo.value) == 'some info'

所以，就你而言，这将类似于

testdata = [
    (..args here..., 'Message')
]

@pytest.mark.parametrize("..args here..., expected_exception_message", testdata)
    def test_myfunc(..args here..., expected_exception_message):
        with pytest.raises(TypeError) as excinfo: 
            obs = myfunc(..args here...)
        assert str(excinfo.value) == expected_exception_message