pytest参数化实现DDT：读取excel文件

最新推荐文章于 2024-06-20 15:22:08 发布

永远不要矫情

最新推荐文章于 2024-06-20 15:22:08 发布

阅读量1.1k

点赞数

分类专栏： Python 文章标签： excel python pytest DDT

本文链接：https://blog.csdn.net/dingding_ting/article/details/117233296

版权

Python 专栏收录该内容

30 篇文章 2 订阅

订阅专栏

该博客介绍了如何使用Python的xlrd库读取.xls文件，并展示了相关代码示例。在xlrd 2.0.1版本之后，不再支持.xlsx格式，因此安装时需指定版本为1.2.0。代码成功读取并打印了Excel文件的内容，并使用pytest进行参数化测试，测试过程顺利，输出了所有数据。

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

读取excel文件：

首先安装xlrd：使用命令pip install xlrd==1.2.0
在使用时，导入xlrd
xlrd.open_workbook(filename)方法打开一个excel文件，获取到一个工作簿对象wb
wb.sheet_by_index()获取sheet对象，通过索引的方法，即索引0代表sheet1
sheet.nrows和sheet.ncols获取行和列
使用for循环获取数据

import pytest
import xlrd

def get_data():
    filename = 'test.xlsx'
    wb = xlrd.open_workbook(filename)
    sheet = wb.sheet_by_index(0)
    rows = sheet.nrows
    print(rows)
    cols = sheet.ncols
    print(cols)
    lst = []
    for row in range(rows):
        for col in range(cols):
            cell_data = sheet.cell_value(row, col)
            lst.append(cell_data)

    return lst
@pytest.mark.parametrize('name', get_data())
def test1(name):
    print(name)

if __name__ == '__main__':
    pytest.main(['-sv', 'test_excel.py'])

test.xlsx的数据如下：
在这里插入图片描述
代码结果输出：

plugins: allure-pytest-2.9.41, dependency-0.5.1
collecting ... 1
3
collected 3 items

test_excel.py::test1[lily] lily
PASSED
test_excel.py::test1[bob] bob
PASSED
test_excel.py::test1[rose] rose
PASSED

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

在安装xlrd的过程中需要注意的问题是，如果使用pip install xlrd,运行上述代码会报：xlrd.biffh.XLRDError: Excel xlsx file; not supported。

原因在于pip install xlrd指定的安装版本是2.0.1，而在xlrd官网： https://pypi.org/project/xlrd/#description 有提到：This library will no longer read anything other than .xls files. For alternatives that read newer file formats, please see http://www.python-excel.org/. ，xlrd从2.X版本开始只支持.xls格式的EXCEL文件，不支持其它的。
在这里插入图片描述
所以我们安装时需要指定版本：pip install xlrd==1.2.0。再运行正常输出。