从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" ]
相关推荐
m0_570466415 分钟前
代码随想录算法训练营第二十八天 | 买卖股票的最佳实际、跳跃游戏、K次取反后最大化的数组和
java·开发语言·算法
程序喵大人6 分钟前
分享个C++线程池的实现源码
开发语言·c++·线程池
和鲸社区11 分钟前
《斯坦福CS336》作业1开源,从0手搓大模型|代码复现+免环境配置
人工智能·python·深度学习·计算机视觉·语言模型·自然语言处理·nlp
不会吃萝卜的兔子30 分钟前
go webrtc - 1 go基本概念
开发语言·golang·webrtc
豌豆花下猫43 分钟前
Python 潮流周刊#118:Python 异步为何不够流行?(摘要)
后端·python·ai
THMAIL1 小时前
深度学习从入门到精通 - LSTM与GRU深度剖析:破解长序列记忆遗忘困境
人工智能·python·深度学习·算法·机器学习·逻辑回归·lstm
要做朋鱼燕1 小时前
【C++】 priority_queue 容器模拟实现解析
开发语言·c++·笔记·职场和发展
jiaway1 小时前
【C语言】第四课 指针与内存管理
c语言·开发语言·算法
励志不掉头发的内向程序员1 小时前
C++进阶——继承 (1)
开发语言·c++·学习
悠哉悠哉愿意1 小时前
【数学建模学习笔记】机器学习分类:随机森林分类
学习·机器学习·数学建模