python pytest 参数化的几种方式

一、使用@pytest.mark.parametrize装饰器：

可以使用pytest提供的@pytest.mark.parametrize装饰器来指定参数化测试的参数。下面是一个示例：

import pytest


# @pytest.mark.parametrize装饰器
# 其中num expected，分别对应(1, 1),(2, 4),(3, 9)，即num=1时，期望值expected=1；num=2时，期望值expected=4；num=3时，期望值expected=9；
@pytest.mark.parametrize("num,expected", [(1, 1),
                                          (2, 4),
                                          (3, 9)])
# 定义一个函数，求num的平方，看结果是否和期望值expected相等
def test_square(num, expected):
    assert num ** 2 == expected
    print(num ** 2)


# 在main函数中调用函数test_square()
if __name__ == '__main__':
    test_square()

输出结果：