pytest测试框架使用基础07 fixture—parametrize获取参数的几种常用形式

news2025/1/7 7:23:01

【pytest】parametrize获取参数的几种常用形式:

a.数据结构
b.文件
c.数据库
d.conftest.py配置

一、直接在标签上传参

1.1 一个参数多个值

@ pytest.mark.parametrize("参数", (参数值1, 参数值2, 参数值3))

示例:


import pytest
# 单个参数的情况
@pytest.mark.parametrize("a", (1, 2, 3, 4))
def test_add(a):
    print("\na的值")

输出:

============================= test session starts =============================
collecting ... collected 4 items

parametrize01.py::test_add[1] PASSED                                     [ 25%]
a的值

parametrize01.py::test_add[2] PASSED                                     [ 50%]
a的值

parametrize01.py::test_add[3] PASSED                                     [ 75%]
a的值

parametrize01.py::test_add[4] PASSED                                     [100%]
a的值


============================== 4 passed in 0.04s ==============================

Process finished with exit code 0

1.2 多个参数多个值的情况

@ pytest.mark.parametrize("参数a, 参数b", ([a1, b1], [a2, b2], [a3, b3]))

示例:

@pytest.mark.parametrize("b,c,d", ([1, 2, 3], [4, 5, 6], [7, 8, 0]))
def test_add2(b, c, d):
    print("\nb,c,d的值分别为:", f'{b},{c},{d}')

输出:

============================= test session starts =============================
collecting ... collected 3 items

parametrize01.py::test_add2[1-2-3] PASSED                                [ 33%]
b,c,d的值分别为: 1,2,3

parametrize01.py::test_add2[4-5-6] PASSED                                [ 66%]
b,c,d的值分别为: 4,5,6

parametrize01.py::test_add2[7-8-0] PASSED                                [100%]
b,c,d的值分别为: 7,8,0


============================== 3 passed in 0.03s ==============================

Process finished with exit code 0

二、数据结构

2.1 列表(list)形式

示例:

import pytest

# 单个参数的情况
data1 = [1, 2, 3, 4]


@pytest.mark.parametrize("a", data1)
def test_add(a):
    print('\na的值:', a)


# 多个参数多个值得情况

data2 = [
    (1, 2, 3),
    (4, 5, 6),
    (0, 8, 9)
]


@pytest.mark.parametrize("a,b,c", data2)
def test_add2(a, b, c):
    print('\na,b,c的值分别为:', f'{a},{b},{c}')

输出:

============================= test session starts =============================
collecting ... collected 7 items
parametrize02_list.py::test_add[1] PASSED                                [ 14%]
a的值: 1
parametrize02_list.py::test_add[2] PASSED                                [ 28%]
a的值: 2
parametrize02_list.py::test_add[3] PASSED                                [ 42%]
a的值: 3
parametrize02_list.py::test_add[4] PASSED                                [ 57%]
a的值: 4
parametrize02_list.py::test_add2[1-2-3] PASSED                           [ 71%]
a,b,c的值分别为: 1,2,3
parametrize02_list.py::test_add2[4-5-6] PASSED                           [ 85%]
a,b,c的值分别为: 4,5,6
parametrize02_list.py::test_add2[0-8-9] PASSED                           [100%]
a,b,c的值分别为: 0,8,9
============================== 7 passed in 0.05s ==============================
Process finished with exit code 0

列表的基本操作:列表读取,列表添加

# 试验1:列表操作
# 列表定义

list1 = ['physics', 'chemistry', 'english', 'artist']
list2 = [1, 2, 3, 4, 5]

# 列表读取
# 方法1
for data in list1:
    print(data)
# 方法2
for i in range(0, len(list2)):
    print(list2[i])

# 列表元素追加
list1.append('history')
print(list1)

输出:

D:\ApiTest3\.venv\Scripts\python.exe D:\ApiTest3\Pytest_session\test_parametrize\list_test.py 
physics
chemistry
english
artist
1
2
3
4
5
['physics', 'chemistry', 'english', 'artist', 'history']

Process finished with exit code 0

2.2 字典(dictionary)形式

import pytest

userinfo = [
    {"username": "小李", "password": "123456"},
    {"username": "大白", "password": "894561"}
]


@pytest.mark.parametrize('info', userinfo)
def test_add(info):
    print(info)

输出:

============================= test session starts =============================
collecting ... collected 2 items

parametrize03_dic.py::test_add[info0] PASSED                             [ 50%]{'username': '小李', 'password': '123456'}

parametrize03_dic.py::test_add[info1] PASSED                             [100%]{'username': '大白', 'password': '894561'}


============================== 2 passed in 0.03s ==============================

Process finished with exit code 0

关于字典的一些基本操作如下:字典键值的获取,字典元素的追加

# 试验2:数据字典操作
# 数据字典定义

userinfo = {
    "username": "钟无艳",
    "password": "54674567"
}

# 字典读取
# 方法1:读取key
for key in userinfo.keys():
    print(key)

# 方法2:读取value
for val in userinfo.values():
    print(val)

# 方法3:读取所有内容
for key in userinfo:
    print(key,userinfo[key])
# 方法4:读取键值对
for data in userinfo.items():
    print(data)

# 字典元素追加
userinfo["phone"] = '13855455444'
print(userinfo)

输出:

D:\ApiTest3\.venv\Scripts\python.exe D:\ApiTest3\Pytest_session\test_parametrize\dic_test.py 
username
password
钟无艳
54674567
username 钟无艳
password 54674567
('username', '钟无艳')
('password', '54674567')
{'username': '钟无艳', 'password': '54674567', 'phone': '13855455444'}

Process finished with exit code 0

2.3 元组(tuple)形式

# 元组形式:
import pytest

data1 = [(1, 2, 3), (4, 5, 6)]


@pytest.mark.parametrize("a,b,c", data1)
def test_case1(a, b, c):
    print(a, b, c)

输出:

============================= test session starts =============================
collecting ... collected 2 items

parametrize04_tuple.py::test_case1[1-2-3] PASSED                         [ 50%]1 2 3

parametrize04_tuple.py::test_case1[4-5-6] PASSED                         [100%]4 5 6


============================== 2 passed in 0.03s ==============================

Process finished with exit code 0

三、json形式

有些数据是通过json文件来保存的,我们可以读取json文件的内容,然后将其转换为字典格式传参,下面讲一下如何从json文件读取参数。
大体分为以下四步

  1. 通过读取.json文件来达到数据分离
  2. 使用pytest中的parametrize参数化来读取json文件的内容
  3. 通过requests来指定请求方法,请求参数等
  4. 通过返回结果来判断json文件中的response是否包含(或者相等)/断言
  5. json数据
    整体代码:
    interface.json:
{
  "item": [
    {
      "request": {
        "url": "http://192.xxx.xxx18/Account/Login",
        "body": {
          "username": "lipan",
          "password": "E10ADC*********BE56E057F20F883E",
          "isRememberme": false,
          "validateCode": ""
        }
      },
      "response": {
        "status": 1,
        "msg": "",
        "data": null,
        "datas": null,
        "errCode": 0
      }
    }
  ]
}

parametrize05_json.py

# json形式:
import json
import pytest
import requests


# 1 读取json文件
# 由于需要的数据在item下面一层级,所以在读取时,需要指定位置
def rade_json():
    return json.load(open('interface.json', 'r'))['item']


# 通过pytest的parametrize参数化来使用read_json()里面的内容
@pytest.mark.parametrize('data', rade_json())
def test_json_login(data):
    # 通过requests来指定请求方法
    r = requests.post(
        # 通过指定read_json()里面对应的url
        url=data['request']['url'],
        # 通过指定read_json()里面对应的body
        json=data['request']['body'])

    # 通过判断返回的json数据判断read_json()里面response内容
    # 通过使用assert来断言判断是否满足预期结果
    assert r.json()['status'] == data['response']["status"]
    print(r.json())


if __name__ == '__main__':
    pytest.main('-s', '-v', 'parametrize05_json.py')

输出:

============================= test session starts =============================
collecting ... collected 1 item

parametrize05_json.py::test_json_login[data0] 

============================== 1 passed in 0.42s ==============================
PASSED                     [100%]{'status': 1, 'msg': '', 'data': None, 'datas': None, 'errCode': 0}

Process finished with exit code 0

四、文件

4.1 txt文件

file = open("文件名.txt", "r", encoding="utf-8")
content = file.read()
print(content)

示例,从下面txt文件中获取所有权限然后传参:

代码如下:

# txt文件传参:
import json
import pytest
import requests


# 获取权限列表方法
def get_txt():
    # 从文件中读取权限
    authoList = []
    # 打开txt文件
    txt_file = open('my_test.txt', 'r', encoding='utf-8')
    # 读取txt文件
    con = txt_file.read()
    print(con)

    result = con[1: -1].split(',')
    for autho in result:
        authoList.append(autho[1: -1])
        print(authoList)
    return authoList


@pytest.mark.parametrize('autho', get_txt())
def test_add_group(autho):
    print(autho)

4.2 csv文件

读取csv文件要导入csv库,基本格式如下:

# 读取csv文件
file1 = open("文件名.csv", "r")
content = csv.reader(file)
file1.close()
 
# 写入csv文件内容
file2 = open("文件名.csv""w")
write = csv.writer(file2)
file2.close()

示例,从以下csv表格中获取参数那列的值:
在这里插入图片描述

代码如下:

# csv文件传参:
import pytest
import csv


# 打开文件
def get_userdata():
    # 登入数据列表
    dataList = []
    file1 = open('login_params.csv', 'r')
    # 读文件
    readers = csv.reader(file1)
    # 跳过首行
    readers.__next__()
    for row in  readers:
        print(type(row[2]))
        # 将字符串类型转为字典类型
        userdata = eval(row[2])
        print(type(userdata),userdata)
        dataList.append(userdata)
    print(dataList)
    return dataList


@pytest.mark.parametrize('userinfo',get_userdata())
def test_add_group(userinfo):
    print(userinfo['username'])
    print(userinfo['password'])
    print(userinfo['state'])

输出:

============================= test session starts =============================
collecting ... collected 3 items

parametrize07_csv.py::test_add_group[userinfo0] PASSED                   [ 33%]test1
123456
1

parametrize07_csv.py::test_add_group[userinfo1] PASSED                   [ 66%]
123456
0

parametrize07_csv.py::test_add_group[userinfo2] PASSED                   [100%]test2

0


============================== 3 passed in 0.04s ==============================

Process finished with exit code 0

4.3 Excel 文件读取

读取Excel文件要导入pandas库,基本格式如下:

# 导入类库
import pandas as pd
 
# 读取Excel
df = pd.read_excel("文件名.xlsx")
 
# 提取对应行的内容
row = df.loc[[1, 3]].values  # 从0开始,0是首行,所以这里读取的是第1行和第4行的内容
 
# 提取对应列的内容
colum = df['列的标题'].values

示例,从以下Excel文件中获取参数那一列的内容:

# Excel文件传参:
import pandas
import pandas as pd
import pytest


def get_userdata():
    # 打开文件
    df = pd.read_excel("test_login.xlsx")
    # row = df.loc[[0, 2]].values
    # print(row)

    # 读文件
    cloum = df['输入参数'].values
    list_data = cloum.tolist()
    print(type(list_data), list_data)
    return list_data


@pytest.mark.parametrize("userinfo", get_userdata())
def test_add_group(userinfo):
    # 将其转换为字典类型
    userinfo_dic = eval(userinfo)
    print(userinfo_dic["username"])
    print(userinfo_dic["password"])

输出:

============================= test session starts =============================
collecting ... collected 4 items

parametrize08_Excel.py::test_add_group[{"username":"test1", "password":"123456"}] PASSED [ 25%]test1
123456

parametrize08_Excel.py::test_add_group[{"username":"", "password":"123456"}] PASSED [ 50%]
123456

parametrize08_Excel.py::test_add_group[{"username":"test3", "password":""}] PASSED [ 75%]test3


parametrize08_Excel.py::test_add_group[{"username":"test4", "password":""}] PASSED [100%]test4



============================== 4 passed in 2.08s ==============================

Process finished with exit code 0

五、数据库

读取数据库文件要导入pymysql库,基本格式如下:

# mysql数据库传参:

import pymysql
import pytest

"""
# 连接数据库
db = pymysql.connect(host="127.0.0.1", db="my_test", user="root", passwd="123456", charset="utf8")
# 访问数据库
cursor = db.cursor()
# 读取数据库对应内容
cursor.execute('select * from test_user')
data = cursor.fetchone()
print(data)
# 关闭数据库
db.close()
"""


def get_db_data():
    # 连接数据库
    db = pymysql.connect(host='127.0.0.1', db='my_test', user='root', password='123456', charset='utf8')
    # 访问数据库
    cursor = db.cursor()
    # 查询数据
    cursor.execute('select username, password from test_user')
    data = cursor.fetchall()
    print(data)  # 取出的数据类型是元组

    db.close()
    return data


@pytest.mark.parametrize("userinfo", get_db_data())
def test_add_group(userinfo):
    print(type(userinfo))  # 输入的是元组类型
    print(userinfo[0])
    print(userinfo[1])

输出:

============================= test session starts =============================
collecting ... collected 3 items

parametrize09_mysql.py::test_add_group[userinfo0] PASSED                 [ 33%]<class 'tuple'>
xiaoliu
123456

parametrize09_mysql.py::test_add_group[userinfo1] PASSED                 [ 66%]<class 'tuple'>
zhangshan
456187

parametrize09_mysql.py::test_add_group[userinfo2] PASSED                 [100%]<class 'tuple'>
chengyi
4536187


============================== 3 passed in 0.08s ==============================

Process finished with exit code 0

六、配置conftest.py文件

如果有多处文件都要使用同一个数据,可以将数据放在conftest.py文件下,与fixture固件结合,其他文件就可以直接使用,而不需要导入文件了。
注意点:

  • conftest.py文件名字是固定的,不可以做任何修改
  • conftest.py文件所在目录必须存在__init__.py文件
  • conftest.py文件不能被其他文件导入
  • 所有同目录测试文件运行前都会执行conftest.py文件
  • 用了fixture固件,方法上就不能再使用parametrize标签了
    示例:在conftest.py文件中
# conftest.py文件

import pytest


# 获取权限列表方法
@pytest.fixture(scope='session')
def get_autho():
    # 从文件中读取权限
    authoList = []
    # 打开txt文件
    f1 = open('my_test.txt', 'r', encoding='utf-8')
    # 读取txt文件
    con = f1.read()
    resule = con[1: -1].split(',')

    for autho in resule:
        authoList.append(autho[1: -1])
    return authoList

同目录下其他文件可直接引用(parametrize10_conftest.py)

import pytest


class TestDemo:
    def test_add_group(self, get_autho):
        for autho in get_autho:
            print(autho)


if __name__ == '__main__':
    pytest.main('parametrize10_conftest.py', '-s')

七、分析总结

  1. 私有数据放在对于的脚本中进行统一维护。
  2. 共有的数据:
    方案1:
    – 如果数据种类比较少,不超过5类以上
    –可以放在一个独立的脚本中,编写DDT的方法即可。
    方案2:
    – 需要的数据种类非常多,可以分类存放
    – 建议一个com数据层,创建多个.py文件
    方案3:
    –数据种类不多,复用性特别高
    –直接放在conftest.py中
  3. 一般来说,数据读取的方法和数据文件在相同路径下

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1500830.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

枚举 --java学习笔记

什么是枚举 枚举是一种特殊类 格式&#xff1a; 修饰符 enum 枚举类名{ 名称1&#xff0c;名称2&#xff0c;...&#xff1b; //枚举类的第一行必须罗列的是枚举对象的名字 其他成员... } 枚举类的第一行只能罗列一些名称&#xff0c;这些名称都是常量&#xff0c;…

[C++]类和对象,explicit,static,友元,构造函数——喵喵要吃C嘎嘎4

希望你开心&#xff0c;希望你健康&#xff0c;希望你幸福&#xff0c;希望你点赞&#xff01; 最后的最后&#xff0c;关注喵&#xff0c;关注喵&#xff0c;关注喵&#xff0c;大大会看到更多有趣的博客哦&#xff01;&#xff01;&#xff01; 喵喵喵&#xff0c;你对我真的…

Python数据处理实战(4)-上万行log数据提取并作图进阶版

系列文章&#xff1a; 0、基本常用功能及其操作 1&#xff0c;20G文件&#xff0c;分类&#xff0c;放入不同文件&#xff0c;每个单独处理 2&#xff0c;数据的归类并处理 3&#xff0c;txt文件指定的数据处理并可视化作图 4&#xff0c;上万行log数据提取并作图进阶版&a…

Vue组件中的scoped属性

Vue组件中的scoped属性的作用是&#xff1a;当前的单文件组件的css样式只用于当前组件的template模板&#xff0c;在Vue脚手架汇总组件间关系时避免样式命名重复的情况。 原理&#xff1a;使用data-*属性在template模板中使用样式的HTML元素上添加额外属性&#xff0c;再利用标…

(sub)三次握手四次挥手

TCP的三次握手与四次挥手理解及面试题 序列号seq&#xff1a;占4个字节&#xff0c;用来标记数据段的顺序&#xff0c;TCP把连接中发送的所有数据字节都编上一个序号&#xff0c;第一个字节的编号由本地随机产生&#xff1b;给字节编上序号后&#xff0c;就给每一个报文段指派一…

即插即用篇 | YOLOv8 引入 ParNetAttention 注意力机制 | 《NON-DEEP NETWORKS》

论文名称:《NON-DEEP NETWORKS》 论文地址:https://arxiv.org/pdf/2110.07641.pdf 代码地址:https://github.com/imankgoyal/NonDeepNetworks 文章目录 1 原理2 源代码3 添加方式4 模型 yaml 文件template-backbone.yamltemplate-small.yamltemplate-large.yaml

蓝桥杯2023年-买瓜(dfs,类型转换同样耗时)

题目描述 小蓝正在一个瓜摊上买瓜。瓜摊上共有 n 个瓜&#xff0c;每个瓜的重量为 Ai 。 小蓝刀功了得&#xff0c;他可以把任何瓜劈成完全等重的两份&#xff0c;不过每个瓜只能劈一刀。 小蓝希望买到的瓜的重量的和恰好为 m 。 请问小蓝至少要劈多少个瓜才能买到重量恰好…

Igraph入门指南 3

4、图转换到其他R数据结构 图是对实体关系的表达&#xff0c;在igraph中&#xff0c;图可以转换为三种数据结构。 4-1 图转邻接矩阵&#xff1a;as_adjacency_matrix | as_adj&#xff0c;结果是矩阵 邻接矩阵又分为有向图邻接矩阵和无向图邻接矩阵&#xff0c;但本函数使用…

MySQL-Linux安装

JDK安装&#xff08;linux版&#xff09; CentOS7环境&#xff1a; jdk下载地址huaweicloud.com 创建目录&#xff1a; mkdir /opt/jdk通过 ftp 客户端 上传 jdk压缩包&#xff08;linux版本&#xff09;到 1中目录进入目录&#xff1a;cd /opt/jdk解压&#xff1a;tar -zxv…

ArcGIS学习(十一)公服设施服务区划分与评价

ArcGIS学习(十一)公服设施服务区划分与评价 本任务带来的内容是公服设施服务区划分与公服设施服务区评价。本任务包括两个关卡: 公服设施服务区划分公服设施服务区空间价值评价1.公服设施服务区划分 首先,来看看这个案例的场景和基础数据。我们以上海市图书馆为例进行分析…

信息抽取技术在电商CRM中的应用与实践

一、引言 在当今快速发展的互联网电商领域&#xff0c;客户资源的有效管理和精准营销已成为商家提升竞争力的关键。为了实现这一目标&#xff0c;许多商家开始采用客户关系管理&#xff08;CRM&#xff09;系统来优化他们的客户互动策略。通过这些系统&#xff0c;商家能够收集…

【netty系列-02】深入理解socket本质和BIO底层实现

Netty系列整体栏目 内容链接地址【一】深入理解网络通信基本原理和tcp/ip协议https://zhenghuisheng.blog.csdn.net/article/details/136359640【二】深入理解Socket本质和BIOhttps://zhenghuisheng.blog.csdn.net/article/details/136549478 深入理解socket本质和bio底层实现 …

Spring Cloud Gateway自定义断言

问题&#xff1a;Spring Cloud Gateway自带的断言&#xff08;Predicate&#xff09;不满足业务怎么办&#xff1f;可以自定义断言&#xff01; 先看Spring Cloud Gateway是如何实现断言的 Gateway中断言的整体架构如下&#xff1a; public abstract class AbstractRoutePred…

flink重温笔记(十二): flink 高级特性和新特性(1)——End-to-End Exactly-Once(端到端精确一致性语义)

Flink学习笔记 前言&#xff1a;今天是学习 flink 的第 12 天啦&#xff01;学习了 flink 高级特性和新特性之 End-to-End Exactly-Once&#xff08;端到端精确一致性语义&#xff09;&#xff0c;主要是解决大数据领域数据从数据源到数据落点的一致性&#xff0c;不会容易造成…

MySQL面试题-锁(答案版)

锁 1、MySQL 有哪些锁&#xff1f; &#xff08;1&#xff09;全局锁 加了全局锁之后&#xff0c;整个数据库就处于只读状态了&#xff0c;这时其他线程执行以下操作&#xff0c;都会被阻塞&#xff1a; 对数据的增删改操作&#xff0c;比如 insert、delete、update等语句&…

【OJ比赛日历】快周末了,不来一场比赛吗? #03.09-03.15 #13场

CompHub[1] 实时聚合多平台的数据类(Kaggle、天池…)和OJ类(Leetcode、牛客…&#xff09;比赛。本账号会推送最新的比赛消息&#xff0c;欢迎关注&#xff01; 以下信息仅供参考&#xff0c;以比赛官网为准 目录 2024-03-09&#xff08;周六&#xff09; #6场比赛2024-03-10…

c# combox 行间距调整

初始化combox comboBox1.DropDownStyle ComboBoxStyle.DropDownList;comboBox1.ItemHeight 25; // 设置 combox 的行高comboBox1.DrawMode DrawMode.OwnerDrawVariable; 添加 DrawItem 事件 private void comboBox1_DrawItem(object sender, DrawItemEventArgs e){if (…

Web核心

JavaWeb技术栈 B/S架构&#xff1a;Browser/Server &#xff0c; 浏览器/服务器 架构模式&#xff0c;其特点为&#xff0c;客户端只需要浏览器&#xff0c;应用程序的逻辑和数据都存储在服务器端。浏览器只需要请求服务器&#xff0c;获取Web资源&#xff0c;服务器把Web资源…

RocketMQ快速入门_2. rocketmq 的应用场景、与其他mq的差异

0. 引言 之前我们讲解过rabbitMQ&#xff0c;本期我们将进入吞吐量更加强大的rocketMQ的学习。 1. 基础概念 如果你是刚接触MQ的同学&#xff0c;还不清楚消息队列的基础概念的&#xff0c;可以参考我之前这篇文章&#xff1a; https://wu55555.blog.csdn.net/article/deta…

凌鲨微应用开发流程

微应用开发流程 使用vite,nextjs等框架创建前端项目引入需要的api包通过调试界面进行调试 创建前端项目 vite yarn create vitenextjs yarn create next-app引入需要的api包 名称权限说明http跨域访问跨域http访问tauri提供的apilinksaas-minapp/api打开浏览器读本地文件…