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

Posted 2021-05-26 永远不要矫情

读取excel文件：

1. 首先安装xlrd：使用命令pip install xlrd==1.2.0
2. 在使用时，导入xlrd
3. xlrd.open_workbook(filename)方法打开一个excel文件，获取到一个工作簿对象wb
4. wb.sheet_by_index()获取sheet对象，通过索引的方法，即索引0代表sheet1
5. sheet.nrows和sheet.ncols获取行和列
6. 使用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。再运行正常输出。