Pytest学习笔记8-参数化

2021年7月2日 8点热度 0条评论 来源: RockChe

前言

我们在实际自动化测试中,某些测试用例是无法通过一组测试数据来达到验证效果的,所以需要通过参数化来传递多组数据

在unittest中,我们可以使用第三方库parameterized来对数据进行参数化,从而实现数据驱动测试

而在pytest中,也提供了功能强大的@pytest.mark.parametrize装饰器来实现数据参数化

Pytest参数化的方式

pytest有三种传参方式

  • @pytest.mark.parametrize() 通过装饰器方式进行参数化(最常使用)
  • pytest.fixture()方式进行参数化,fixture装饰的函数可以作为参数传入其他函数
  • conftest.py文件中存放参数化函数,可作用于模块内的所有测试用例

@pytest.mark.parametrize实现参数化

装饰测试类

当装饰器 @pytest.mark.parametrize 装饰测试类时,会将数据集合传递给类的所有测试用例方法

举个

import pytest

# 定义测试数据
data1 = [
    (1, 2, 3),
    (4, 5, 9)
]


# 定义add方法
def add(a, b):
    return a + b


# 添加parametrize装饰器
@pytest.mark.parametrize('a, b, expect', data1)
class TestParametrize(object):

    def test_parametrize_1(self, a, b, expect):
        print(f'\n测试用例1数据为{a}-{b},结果为{expect}')
        assert add(a, b) == expect

    def test_parametrize_2(self, a, b, expect):
        print(f'\n测试用例2数据为{a}-{b},结果为{expect}')
        assert add(a, b) == expect

执行结果如下

装饰测试函数

单个数据

当测试用例只需要一个参数时,我们使用列表存放测试数据,例如定义一个列表

data = [1,2]

使用@pytest.mark.parametrize装饰器时,第一个参数使用变量a接收列表中的每个元素,第二个参数传递存储数据的列表

在测试用例中使用同名的变量a接收测试数据,列表有多少个元素就会生成并执行多少个测试用例

上代码

import pytest

data = [1, 2 , 3]


@pytest.mark.parametrize('a', data)
def test_parametrize(a):
    print(f'\n被加载测试数据为{a}')

执行结果如下

一组数据

当测试用例需要多个数据时,我们可以使用嵌套序列(嵌套元组&嵌套列表)的列表来存放测试数据

装饰器@pytest.mark.parametrize()可以使用单个变量接收数据,也可以使用多个变量接收,同样,测试用例函数也需要与其保持一致

当使用单个变量接收时,测试数据传递到测试函数内部时为列表中的每一个元素或者小列表,需要使用索引的方式取得每个数据

当使用多个变量接收数据时,那么每个变量分别接收小列表或元组中的每个元素

列表嵌套多少个列表或元组,测生成多少条测试用例

上代码

import pytest

data = [
    [1, 2, 3],
    [4, 5, 9]
]


@pytest.mark.parametrize('a, b, expect', data)
def test_parametrize_1(a, b, expect):  
# 当使用多个变量接收数据时,那么每个变量分别接收小列表或元组中的每个元素
    print(f'\n测试数据为{a},{b},{expect}')
    actual = a + b
    assert actual == expect


@pytest.mark.parametrize('value', data)
def test_parametrize_2(value):
当使用单个变量接收时,测试数据传递到测试函数内部时为列表中的每一个元素或者小列表,需要使用索引的方式取得每个数据
    print(f'\n测试数据为{value}')
    actual = value[0] + value[1]
    assert actual == value[2]

执行结果如下

组合数据

一个测试函数还可以同时被多个参数化装饰器装饰,多个装饰器中的数据会进行交叉组合的方式传递给测试函数,进而生成n*n个测试用例,这也为我们的测试设计时提供了方便

上代码

import pytest

data_1 = [1, 2]
data_2 = [3, 4, 5]


@pytest.mark.parametrize('a', data_1)
@pytest.mark.parametrize('b', data_2)
def test_parametrize_1(a, b):
    print(f'\n测试数据为{a},{b}')

执行结果如下

标记用例

当我们不想执行某组测试数据时,我们可以标记skip或skipif

当我们预期某组数据会执行失败时,我们可以标记为xfail

上代码

import pytest

data1 = [
    [1, 2, 3],
    pytest.param(3, 4, 8, marks=pytest.mark.xfail),
    pytest.param(3, 4, 7, marks=pytest.mark.skip)
]


def add(a, b):
    return a + b


@pytest.mark.parametrize("a,b,expected", data1)
def test_mark(a, b, expected):
    print(f'测试数据为{a},{b},结果为{expected}')
    assert add(a, b) == expected

执行结果如下

嵌套字典

数据列表中也可以使用字典类型的数据

上代码

import pytest

data_1 = (
    {
        'user': 1,
        'pwd': 2
     },
    {
        'user': 3,
        'pwd': 4
    }
)


@pytest.mark.parametrize('dic', data_1)
def test_parametrize_1(dic):
    print(f'测试数据为{dic}')

执行结果如下

增加可读性

使用ids参数

参数化装饰器有一个额外的参数ids,可以标识每一个测试用例,自定义测试数据结果的显示,用来增加测试用例的可读性

上代码

import pytest

data = [1, 2, 3]

ids = [f'TestData-{a}' for a in data]

@pytest.mark.parametrize('a', data ,ids= ids)
def test_parametrize(a):
    print(f'\n被加载测试数据为  {a}')

执行结果为

自定义id做标识

除了使用ids参数增加输出可读性外,我们还可以在参数列表的参数旁边定义一个id值来做标识

上代码

import pytest

data = [
pytest.param(1, id="this is test1"),
pytest.param(2, id="this is test2")
]



@pytest.mark.parametrize('a', data)
def test_parametrize(a):
    print(f'\n被加载测试数据为  {a}')

执行结果如下

pytest.fixture()方式进行参数化

待更新

整理参考

linxu超

    原文作者:RockChe
    原文地址: https://www.cnblogs.com/crdym/p/14961500.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系管理员进行删除。