一、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进行参数化测试，我们可以快速、准确地测试各种输入情况下的代码逻辑，提高测试效率。
原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/287073.html
微信扫一扫 
支付宝扫一扫 