前言
在进行软件测试或设计自动化测试框架时,一个比可避免的过程就是: 参数化,在利用python进行自动化测试开发时,通常会使用excel来做数据管理,利用xlrd、xlwt开源包来读写excel。
环境安装
首先在命令行下安装xlrd、xlwt
pip install xlrd
pip install xlwt
一个简单的读写示例
让我们先看一个简单的excel读写示例,示例代码功能,从表1中读取数据。
excel数据表如图所示
demo.xlsx
1、读取代码示例
#_*_ coding:utf-8 _*_
__author__ = '苦叶子'
import xlrd
if __name__ == '__main__':
# excel文件全路径
xlPath = "C:\\Users\\lyy\\Desktop\\demo.xlsx"
# 用于读取excel
xlBook = xlrd.open_workbook(xlPath)
# 获取excel工作簿数
count = len(xlBook.sheets())
print u"工作簿数为: ", count
# 获取 表 数据的行列数
table = xlBook.sheets()[0]
nrows = table.nrows
ncols = table.ncols
print u"表数据行列为(%d, %d)" % (nrows, ncols)
# 循环读取数据
for i in xrange(0, nrows):
rowValues = table.row_values(i) # 按行读取数据
# 输出读取的数据
for data in rowValues:
print data, " ",
print ""
运行效果如图
运行结果
2、写excel示例代码
#_*_ coding:utf-8 _*_
__author__ = '苦叶子'
import xlwt
import random
if __name__ == '__main__':
# 注意这里的excel文件的后缀是xls 如果是xlsx打开是会提示无效
newPath = unicode("C:\\Users\\lyy\\Desktop\\demo_new.xls", "utf8")
wtBook = xlwt.Workbook()
# 新增一个sheet
sheet = wtBook.add_sheet("sheet1", cell_overwrite_ok=True)
# 写入数据头
headList = [u'序号', u'数据1', u'数据2', u'数据3']
rowIndex = 0
col = 0
# 循环写
for head in headList:
sheet.write(rowIndex, col, head)
col = col + 1
# 写入10行 0-99的随机数据
for index in xrange(1, 11):
for col in xrange(1, 4):
data = random.randint(0,99)
sheet.write(index, 0, index) # 写序号
sheet.write(index, col, data) # 写数据
print u"写第[%d]行数据" % index
# 保存
wtBook.save(newPath)
运行结果如图
写excel结果
结束语
这里只是简单的对xlrd、xlwt模块的应用演示,对于实际做自动化测试过程中,需要封装一个通用的excel解析类,以便提高复用性和降低维护成本。
在实际应用中,我们通常需要对表格进行合并、样式设置等等系列动作,请参考官方文档,进行更深入的学习研究