Pytest之参数化

熟悉unittest单元测试框架的小伙伴知道,使用ddt进行数据驱动测试,那么身为功能更加强大且更加灵活的Pytest框架怎么可能没有数据驱动的概念呢?Pytest使用@pytest.mark.parametrize装饰器来实现数据驱动测试的,也就是常说的参数化。一

parametrize语法

 

parametrize(self,argnames, argvalues, indirect=False, ids=None, scope=None)

 

argnames:参数名。

 

argvalues:参数对应值,类型必须为list。如果只有一个参数,里面则是值的列表:

如:@pytest.mark.parametrize("username", ["yy", "yy2", "yy3"])。如果有多个参数,则需要用元组来存放值,一个元组对应一组参数的值,如:@pytest.mark.parametrize("name,pwd", [("yy1", "123"), ("yy2", "123"), ("yy3", "123")])。

 

indirect:如果设置成True,则把传进来的参数当函数执行,而不是一个参数。

 

ids:用例的ID,传一个字符串列表,用来标识每一个测试用例,自定义测试数据结果,增加可读性。

 

 

参数讲解

 

1单个数据

 

import pytest
data = ["小红", "小明"]
@pytest.mark.parametrize("name",data)
def test_demo(name):
    print("测试数据为{}".format(name))

 

运行结果为:

Pytest之参数化

 

 

2一组数据

 

01 列表嵌套字典

import pytest
data_1 = [
    {"username": "admin1", "password": "123456"},
    {"username": "admin2", "password": "12345678"},
]
@pytest.mark.parametrize("data", data_1)
def test_login(data):
    print("账号:{},密码:{}".format(data["username"],data["password"]))

 

运行结果为:

Pytest之参数化

 

02 列表嵌套列表

import pytest
data_1 = [
    ["admin1", "123456"],
    ["admin2", "12345678"],
]
@pytest.mark.parametrize("username,password", data_1)
def test_login(username,password):
    print("账号:{},密码:{}".format(username, password))

 

运行结果为:

Pytest之参数化



03 列表嵌套元组

import pytest
data_1 = [
    ("admin1", "123456"),
    ("admin2", "12345678"),
]
@pytest.mark.parametrize("username,password", data_1)
def test_login(username,password):
    print("账号:{},密码:{}".format(username, password))

 

运行结果为:

Pytest之参数化

 

 

 

场景运用

 

 

1装饰函数

 

import pytest
@pytest.mark.parametrize("username,password",[("admin01","123456"),("admin02","12345678")])
def test_demo(username,password):
    print("用户名:{},密码:{}".format(username,password))

 

运行结果为:

Pytest之参数化

 

 

 

2装饰类
import pytest
@pytest.mark.parametrize("username,password",[("admin01","123456"),("admin02","12345678")])
class TestDemo:
    def test_demo(self,username,password):
        print("用户名:{},密码:{}".format(username,password))

 

运行结果为:

Pytest之参数化

 

注意:装饰测试类时,类内所有的方法必须接收测试数据,否则会报错;装饰测试函数时比较灵活,如果函数不使用数据就可以不装饰。

 

 

3多个参数化装饰器

 

使用多个参数化装饰器,数据会进行交叉组合的方式传递给测试函数,进而生成n*n个测试用例(笛卡尔积)。

test_demo.py

import pytest
username = ["admin1", "admin2", "admin3"]
password = ["123456", "1234567", "12345678"]
@pytest.mark.parametrize("uname",username)
@pytest.mark.parametrize("pwd",password)
def test_demo(uname,pwd):
    print("用户名:{},密码:{}".format(uname, pwd))

 

运行结果如下:

Pytest之参数化

 

 

4增加可读性

 

parametrize参数中的ids,可以标识每一个测试用例,自定义测试数据结果的显示,增加可读性。

test_demo.py

import pytest
data_1 = [
    (1, 2, 3),
    (4, 5, 9)
]
ids = ["a:{} + b:{} = expect:{}".format(a, b, expect) for a, b, expect in data_1]
def add(a, b):
    return a + b
@pytest.mark.parametrize('a, b, expect', data_1, ids=ids)
def test_parametrize_1(a, b, expect):
    print('\n测试函数1测试数据为\n{}-{}'.format(a, b))
    assert add(a, b) == expect

 

命令行输入:pytest -v

Pytest之参数化

上一篇:C. Nice Garland---暴力枚举--Codeforces Round #535 (Div. 3)


下一篇:6. 两个数组的交集