软件测试/测试开发全日制|Pytest结合yaml实现数据驱动

发布时间:2024年01月09日

在这里插入图片描述

前言

我们之前介绍过pytest的参数化,我们使用了@pytest.mark.parametrize(argnames, argvalues) 来实现批量传送参数的目的,但是我们不进行数据驱动的话,我们一旦对数据进行变更,就需要在代码里对我们的参数进行逐一的修改,这对我们来说,是额外增加的工作量,而通过专门存储数据的文件来实现数据驱动,就可以节省我们的工作量,改变参数时,我们只需要对数据文件进行修改,在数据驱动测试中,yaml文件是我们最常使用的数据驱动文件之一,本文就给大家介绍一下pytest结合yaml实现数据驱动。

环境安装

我们在使用yaml之前,需要先安装好yaml的第三方库,命令如下:

pip install PyYAML

创建测试文件和数据文件

假设我们要测试一个简单的函数,比如计算两个数的和:

# code.py

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

接下来,创建一个yaml文件,用于存储测试数据:

# test_data.yaml
test_case_1:
  a: 2
  b: 3
  expected_result: 5

test_case_2:
  a: -1
  b: 10
  expected_result: 9

test_case_3:
  a: 0
  b: 0
  expected_result: 0

编写测试用例

现在,创建一个pytest测试文件,编写测试用例并读取yaml文件中的数据:

# test_code.py

import yaml
import pytest
from code import add

def load_test_data():
    with open('test_data.yaml', 'r') as file:
        test_data = yaml.safe_load(file)
    return test_data

@pytest.mark.parametrize("input_data", load_test_data().values())
def test_add(input_data):
    a = input_data['a']
    b = input_data['b']
    expected_result = input_data['expected_result']

    result = add(a, b)
    assert result == expected_result, f"计算错误:{a} + {b} 应该得到 {expected_result},实际得到 {result}"

运行测试

我们接下来可以运行测试,直接在pycharm中点击绿色三角形运行即可,我们也可以使用命令行运行,命令如下:

pytest test_code.py

pytest将会读取test_code.py文件中的测试用例,并根据test_data.yaml中提供的数据对add函数进行测试。每个测试用例都会使用yaml文件中的数据来进行数据驱动测试。

总结

结合pytestyaml,我们实现了数据驱动测试,有效地测试了同一个函数在不同输入下的行为。这种方法使得测试用例易于维护,并且可以轻松地添加更多的测试数据,确保代码的稳定性和正确性。

获取更多技术资料,请点击!

推荐

Python全栈开发与自动化测试开发班
由浅入深实战进阶,从小白到高手

以Python全栈开发为基础,深入教授自动化测试技能,为学员打造全面的技术能力。通过系统学习和实际项目实战,学员将具备在职场中脱颖而出的竞争力。不仅能够灵活运用Python进行开发,还能够保障项目质量通过自动化测试手段。这是一个全面提升职业竞争力的机会。

课程详情
Python开发必备基础技能与项目实战
Pvthon 编程语言/算法和数据结构/面向对象编程Web后端开发/前端开发/测试管理平台项目实战

人工智能ChatGPT实战
人工智能辅助学习各种开发和测试技能/Pytorch深度学框架/平台开发实战

数据分析与自动化办公
数据采集/Pandas与数据处理技术/ECharts与数据可视化技术/爬虫实战/自动化办公/批量文件处理

UI自动化测试与高级项目实战
Web自动化测试/App自动化测试/ PageObject设计模式

接口自动化测试
接口协议分析/Mock实战/服务端接口测试

性能测试
性能测试流程与方法/JMeter 脚本参数化/Grafana监控系统搭建

简历指导与模拟面试
1V1简历指导/模拟真实面试/测试开发岗面试全攻略名企私教服务
名企专家1v1辅导/行业专家技术指导/针对性解决工作难题/绩效提升辅导与晋升复盘

课程亮点
名企私教服务 先学习后付费 高额奖学金
专属社群+晚自习在线答疑
5V1全方位辅导作业+考试强化学习效果
简历修改 模拟面试 就业内推 面试复盘

文章来源:https://blog.csdn.net/Hogwartstester/article/details/135460508
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。