自动化测试项目实战 | 使用Excel作为数据源（附限免视频）

Excel是常用的电子表格，以．xlsx或．xls作为文件后缀。在本节中使用著名的第三方函数库openpyxl和xlrd来读取Excel中的测试数据。执行命令来安装openpyxl：

openpyxl只能读取．xlsx格式的Excel文件。

执行命令安装xlrd，命令如下：

xlrd只能读取.xls格式的Excel文件。

接着，在data_source_util模块中新增读取Excel文件的read_excel()函数，代码如下：

从以上代码可以看出，由于需要兼容．xlsx和．xls两种后缀的Excel文件，read_excel()函数对文件后缀作了判断，如果是以．xlsx作为后缀的Excel文件，就使用openpyxl，否则使用xlrd。对比openpyxl和xlrd可以看出，前者API更为复杂，获取一行数据需要使用行号和列号；而后者只需要行号即可。默认读取的Sheet是Sheet1。

在使用以上代码之前，还需要添加导入语句，代码如下：

新增Excel文件test-data．xlsx，其内容如图7-6所示。

■ 图7-6Excel文件内容

将test_ims模块的read_csv()函数改成read_excel()函数，代码如下：

最后添加导入语句，代码如下：

重新执行test_login()测试函数，测试通过，说明读取Excel文件成功。

如果使用后缀为．xls的Excel文件，结果是一样的，读者可自行试验。