从0开始python学习-52.pytest之ddt数据封装

  1. 判断是否为ddt数据
python 复制代码
def read_testcase(yaml_path):
    with open(yaml_path,encoding="utf-8") as f:
        case_list = yaml.safe_load(f)
        if "parametrize" in dict(*case_list).keys(): # 读取出来的yaml用例带有parametrize就说明是ddt用例,格式为:[{}]
            new_caseinfo = ddt(*case_list)
            return new_caseinfo # 将数据解包后,返回格式为:[{},{}]
        else:
            return case_list
  1. 处理数据
python 复制代码
def ddt(caseinfo:dict):
    data_list = caseinfo["parametrize"] # 获取到caseinfo下的parametrize数据
    len_flag = True # 设置一个判断指标
    name_len = len(data_list[0])  # 获取参数的长度,即有几个参数
    for data in data_list:
        if len(data) != name_len: # 判断获取到的所有的参数长度是否与name_len一致,不一致直接结束
            len_flag = False
            print("parametrize数据长度不一致")
            break

    # 将caseinfo转为str类型方便后续操作
    str_caseinfo = yaml.dump(caseinfo)
    new_caseinfo = []
    if len_flag: # 当长度一致时,进行后续操作
        for x in range(1,len(data_list)): # 从下标为1开始取行数据,即排除数据名称行后的就是用例数据行
            raw_caseinfo = str_caseinfo
            for y in range(0,name_len): # 从下标为0开始取列数据,即长度有多少就去多少列
                # 处理数字类型的字符串数据
                if isinstance(data_list[x][y],str) and data_list[x][y].isdigit():
                    data_list[x][y] = "'" + data_list[x][y] + "'" # 转为字符串类型的数字
                raw_caseinfo = raw_caseinfo.replace("$ddt{"+data_list[0][y]+"}",str(data_list[x][y])) # 将获取到的数据写入
            case_list = yaml.safe_load(raw_caseinfo) # 再次进行数据转换回list
            case_list.pop("parametrize")
            new_caseinfo.append(case_list)
    return new_caseinfo
  1. yaml用例中设置和使用数据
python 复制代码
-
  feature: 测试模块4
  story: 测试接口4
  title: 验证测试接口4正常返回
  request:
    method: post
    url: http://192.168.0.1/phpwind/index.php?m=u&c=login&a=dorun
    data:
      username: $ddt{username}
      password:  $ddt{password}
  parametrize:
    - [ "username","password" ]
    - [ "ces1","123" ]
    - [ "ces2",123 ]
    - [ "ces","ces" ]
相关推荐
不爱学英文的码字机器5 分钟前
数字孪生的浪潮:从虚拟镜像到现实世界的 IT 变革
大数据·python
小白—人工智能9 分钟前
数据可视化 —— 直方图
python·信息可视化·数据可视化
四夕白告木贞12 分钟前
stm32week13
stm32·单片机·嵌入式硬件·学习
九亿AI算法优化工作室&14 分钟前
结合大语言模型的机械臂抓取操作学习
人工智能·学习·语言模型·自然语言处理
anqi2724 分钟前
如何在 IntelliJ IDEA 中编写 Speak 程序
java·大数据·开发语言·spark·intellij-idea
XuX0328 分钟前
MATLAB小试牛刀系列(1)
开发语言·matlab
大模型真好玩36 分钟前
初学者必看大模型微调指南:Unsloth官方微调技巧大公开!
人工智能·python
Suckerbin39 分钟前
第十四章-PHP与HTTP协议
开发语言·http·php
Best_Liu~44 分钟前
TransactionTemplate 与@Transactional 注解的使用
java·开发语言·spring boot·后端
谈不譚网安1 小时前
初识Python
开发语言·python