测试一个功能,使用多个测试用例,但仅使用一个断言?

时间:2020-04-26 07:34:18

标签: python python-3.x pytest

我有一个类方法,该方法接受一个列表并根据函数确定该列表是否有效。

我想在存储为静态变量的三个列表上对其进行测试,因为它们稍后将在代码中的其他单元测试中使用。

def test__validate(self):
    decoder = Validator()
    slow_valid = Validator.validate(TestValidator.list_slow)
    med_valid = Validator.validate(TestValidator.list_med)
    fast_valid = Validator.validate(TestValidator.list_fast)


    assert slow_valid == True
    assert med_valid == False
    assert fast_valid == False

删除多个assert语句的正确方法是什么?

我是否定义了test__validate的多个版本,或者从最佳实践位置是否可以使用多个assert语句?

1 个答案:

答案 0 :(得分:1)

如@IanShelvington所建议的那样,在输入改变(结果取决于您的情况)的情况下进行重复测试的最佳实践是测试参数化。使用pytest,您可以执行以下操作:

import pytest

@pytest.mark.parametrize("val_list, result",
                         [(TestValidator.list_slow, True),
                          (TestValidator.list_med, False),
                          (TestValidator.list_fast, False)])

def test_validate(val_list, result):
    assert Validator().validate(val_list) == result

这将为您提供输出:

============================= test session starts =============================
...
collecting ... collected 3 items

param_result.py::test_validate[val_list0-True] PASSED                    [ 33%]
param_result.py::test_validate[val_list1-False] PASSED                   [ 66%]
param_result.py::test_validate[val_list2-False] PASSED                   [100%]

============================== 3 passed in 0.04s ==============================

如您所见,这将创建3个单独的测试,名称中带有参数,因此可以轻松识别失败的测试。

如果您想要显示的测试的自定义名称,则可以使用ids提供它们:

@pytest.mark.parametrize("val_list, result",
                         [(TestValidator.list_slow, True),
                          (TestValidator.list_med, False),
                          (TestValidator.list_fast, False)],
                         ids=('slow', 'med', 'fast'))
...

这将输出:

============================= test session starts =============================
...
param_result.py::test_validate[slow] PASSED                              [ 33%]
param_result.py::test_validate[med] PASSED                               [ 66%]
param_result.py::test_validate[fast] PASSED                              [100%]

============================== 3 passed in 0.06s ==============================