python - 一个 pytest 函数中的多个测试

Question

我有一个关于 pytest 以及我应该如何处理多个测试的“概念性”问题。

假设我需要测试一堆字符串:names = ["", "a ", "a", "a "]。

我可以创建多个 test_ 函数来单独测试它们，但它们会完全相同，但输入不同。

我正在考虑做类似的事情:

def test_names():
    names = [" ", "a ", " a", " a "]
    for name in names:
        with pytest.raises(errors.MyFooError) as ex:
            random_method(name)

问题是:这样做，如果一个元素没有引发 errors.MyFooError，我会收到这个:

Failed: DID NOT RAISE <class 'errors.MyFooError'

消息是通用的，我不知道这是在哪个元素中发生的。

我可能会分成 4 个测试，一个用于列表中的每个元素，我想这是正确的方法吗？如果列表很大会怎样？

Answer 1

您应该使用 pytest 的parametrize 功能，如图所示 here .

你的代码看起来像

import pytest

@pytest.mark.parametrize('input', [" ", "a ", " a", " a "])
def test_names(input):
    with pytest.raises(errors.MyFooError) as ex:
        random_method(input)