如何高效使用pytestparametrize进行Python测试

一、Pytestparametrize简介

在编写Python测试时，经常会遇到需要编写多个测试用例的情况，这时我们可以使用pytestparametrize库来简化代码，提高测试效率。pytestparametrize是Python中的一个参数化库，可以通过参数化实现代码重用，减少代码量。该库可以以参数化方式运行测试多次，只需要在测试方法上添加装饰器“@pytest.mark.parametrize”，并提供测试数据即可，下面我们来看一下代码示例：

import pytest

@pytest.mark.parametrize("test_input,expected_output",[

    (3,9),

    (2,4),

    (5,25),

])

def test_calculate_square(test_input, expected_output):

    assert test_input*test_input == expected_output

在测试用例中添加了“@pytest.mark.parametrize”装饰器，后面的参数用列表的形式提供，在上面的代码中，我们测试了三组数据，分别是（3，9）、（2，4）和（5，25），每组数据都是一个元组，第一个元素表示传入测试方法的参数，第二个元素表示期望结果。pytestparametrize会将这几组数据分别传入测试方法test_calculate_square中，并验证它们是否符合期望结果，从而完成测试。

二、使用pytestparametrize进行测试数据组合

在Python中，pytestparametrize可以方便地进行测试数据组合，即将多个测试数据进行组合，生成新的测试数据来测试代码的各种逻辑。下面我们来看一个代码示例：

import pytest

@pytest.mark.parametrize("test_input1,test_input2,expected_output",[

    (1,2,3),

    (2,5,7),

    (7,9,16),

    (9,10,19),

    (15,20,35),

])

def test_calculate_addition(test_input1,test_input2, expected_output):

    assert test_input1+test_input2 == expected_output

在测试用例“test_calculate_addition”中，我们利用pytestparametrize实现了多个测试数据的组合，每组数据都需要输入两个参数test_input1和test_input2，并提供期望的输出结果expected_output。测试数据的组合，可以通过多个元素分别提供，以逗号区分，pytestparametrize会根据提供的测试数据集自动生成数据组合用于测试。在上面的代码中，我们测试了5组数据，其中第一组数据（1，2，3）表示test_input1=1，test_input2=2，expected_output=3，即测试1+2是否等于3，依次类推，从而完成测试。

三、使用pytestparametrize进行复杂逻辑测试

在实际工作中，我们经常需要测试一些比较复杂的逻辑，这时我们可以利用pytestparametrize进行参数化测试，从而对复杂的代码逻辑进行快速、准确的测试。下面我们来看一个实际代码的例子：

import pytest

def get_total_amount(discount_code, product_price, member_level):

    if discount_code=="ABC123":
        if member_level=="VIP":
            total_amount = product_price*0.7
        elif member_level=="SVIP":
            total_amount = product_price*0.6
        else:
            total_amount = product_price*0.8
    elif discount_code=="XYZ456":
        if member_level == "VIP":
            total_amount = product_price * 0.9
        elif member_level == "SVIP":
            total_amount = product_price * 0.8
        else:
            total_amount = product_price * 0.95
    else:
        total_amount = product_price

    return total_amount


@pytest.mark.parametrize("discount_code, product_price, member_level, expected_total_amount",[

    ("ABC123", 100, "VIP", 70),

    ("XYZ456", 100, "SVIP", 80),

    ("ABC123", 100, "NORMAL", 80),

    ("XYZ456", 100, "NORMAL", 95),

])

def test_total_amount(discount_code, product_price, member_level, expected_total_amount):

    assert get_total_amount(discount_code, product_price, member_level) == expected_total_amount

在上面的代码中，我们实现了一个get_total_amount函数，该函数用于根据传入的参数计算商品的总价并返回结果。我们通过pytestparametrize实现了多组参数化，其中包括折扣码（discount_code）、商品价格（product_price）、会员等级（member_level）以及期望的总价（expected_total_amount）。我们测试了4组数据，pytestparametrize会将这些数据逐一传入test_total_amount方法中，并验证get_total_amount函数的输出是否等于期望的total_amount，以验证该代码是否正确。通过利用pytestparametrize进行参数化测试，我们可以快速、准确地测试各种输入情况下的代码逻辑，提高测试效率。