Pytest结合数据驱动-Excel

最新推荐文章于 2024-04-16 14:11:03 发布

代码小怡

最新推荐文章于 2024-04-16 14:11:03 发布

阅读量58

点赞数

文章标签： pytest excel 软件测试程序员接口测试自动化测试测试工程师

本文链接：https://blog.csdn.net/weixin_57805858/article/details/133681523

版权

本文介绍了如何使用Pytest结合openpyxl库在Python中读取Excel文件，进行数据驱动的测试用例编写。重点展示了如何从Excel中读取数据，并在测试函数中参数化执行。同时提及了工程目录结构和测试准备步骤，以及配套的软件测试资源。

摘要由CSDN通过智能技术生成

Pytest 结合数据驱动 Excel

读取 Excel 文件

第三方库
- xlrd
- xlwings
- pandas

openpyxl
- 官方文档： openpyxl - A Python library to read/write Excel 2010 xlsx/xlsm files — openpyxl 3.1.2 documentation

openpyxl 库的安装

安装：pip install openpyxl
导入：import openpyxl

openpyxl 库的操作

读取工作簿
读取工作表
读取单元格

import openpyxl

# 获取工作簿
book = openpyxl.load_workbook('../data/params.xlsx')

# 读取工作表
sheet = book.active

# 读取单个单元格
cell_a1 = sheet['A1']
cell_a3 = sheet.cell(column=1, row=3)  # A3

# 读取多个连续单元格
cells = sheet["A1":"C3"]

# 获取单元格的值
cell_a1.value

工程目录结构

data 目录：存放 excel 数据文件
func 目录：存放被测函数文件
testcase 目录：存放测试用例文件

# 工程目录结构
.
├── data
│   └── params.excel
├── func
│   ├── __init__.py
│   └── operation.py
└── testcase
    ├── __init__.py
    └── test_add.py

测试准备

被测对象：operation.py
测试用例：test_add.py

# operation.py 文件内容
def my_add(x, y):
    result = x + y
    return result

# test_add.py 文件内容
class TestWithEXCEL:
    @pytest.mark.parametrize('x,y,expected', get_excel())
    def test_add(self, x, y, expected):
        assert my_add(int(x), int(y)) == int(expected)

测试准备

测试数据：params.xlsx

Pytest 数据驱动结合 Excel 文件

# 读取Excel文件
import openpyxl
import pytest

def get_excel():
    # 获取工作簿
    book = openpyxl.load_workbook('../data/params.xlsx')

    # 获取活动行（非空白的）
    sheet = book.active

    # 提取数据，格式：[[1, 2, 3], [3, 6, 9], [100, 200, 300]]
    values = []
    for row in sheet:
        line = []
        for cell in row:
            line.append(cell.value)
        values.append(line)
    return values