一、openpyxl介绍安装
1.为什么要学Excel
存测试数据
有时候有大批量的数据，存到TXT文件里面显然不是最佳的方式，我们可以存到Excel里面去，第一方便我们存数据和做数据，另一方面方便我们读取数据，比较明朗。测试的时候就从数据库中读取出来，这点是非常重要的。
存测试结果
可以批量把结果存入到Excel中，也是比较好整理数据点，比我们的TXT要好。
2.安装openpyxl
python中与excel操作相关的模块：
- xlrd库：从excel中读取数据，支持xls、xlsx
- xlwt库：对excel进行修改操作，不支持对xlsx格式的修改
- xlutils库：在xlw和xlrd中，对一个已存在的文件进行修改。
- openpyxl：主要针对xlsx格式的excel进行读取和编辑。目前用的比较多 ；
安装方式：pip install openpyxl
3.Excel中的三大对象
- WorkBook：工作簿对象
- Sheet：表单对象
- Cell：表格对象
二、openpyxl对Excel的操作
- 创建一个工作薄：wb = openpyxl.Workbook()
- 新增一个sheet表单：wb.create_sheet('test_case')
- 保存case.xlsx文件：wb.save('cases.xlsx')
- 打开工作簿：wb = openpyxl.load_workbook('cases.xlsx')
- 选取表单：sh = wb['Sheet1'
- 读取第一行、第一列的数据：ce = sh.cell(row = 1,column = 1)
- 按行读取数据：row_data = list(sh.rows)
- 关闭工作薄：wb.close()
- 按列读取数据：columns_data = list(sh.columns)
- 写入数据之前，该文件一定要处于关闭状态
- 写入第一行、第四列的数据 value = 'result'：sh.cell(row = 1,column = 4,value = 'result')
- 获取最大行总数、最大列总数：sh.max_row、sh.max_column
- del 删除表单的用法：del wb['sheet_name']
- remove 删除表单的用法：sh = wb['sheet_name'] wb.remove(sh)
import openpyxl
# 创建一个工作簿
wb = openpyxl.Workbook()
# 创建一个test_case的sheet表单
wb.create_sheet('test_case')
# 保存为一个xlsx格式的文件
wb.save('cases.xlsx')
# 读取excel中的数据
# 第一步：打开工作簿
wb = openpyxl.load_workbook('cases.xlsx')
# 第二步：选取表单
sh = wb['Sheet1']
# 第三步：读取数据
# 参数 row:行 column：列
ce = sh.cell(row = 1,column = 1) # 读取第一行，第一列的数据
print(ce.value)
# 按行读取数据 list(sh.rows)
print(list(sh.rows)[1:]) # 按行读取数据，去掉第一行的表头信息数据
for cases in list(sh.rows)[1:]:
case_id = cases[0].value
case_excepted = cases[1].value
case_data = cases[2].value
print(case_excepted,case_data)
# 关闭工作薄
wb.close()
三.封装一个读取用例的excel类：用来实现读取数据和写入数据的功能
cases.xlsx的测试数据：
1.按行读取数据，存储在列表中
import openpyxl
class Case: #这个类用来存储用例的
__slots__ = [] #特殊的类属性，可以用来限制这个类创建的实例属性添加 可写可不写
pass
class ReadExcel(object): #读取excel数据的类
def __init__(self,file_name,sheet_name):
"""
这个是用来初始化读取对象的
:param file_name: 文件名 ---> str类型
:param sheet_name: 表单名 ———> str类型
"""
# 打开文件
self.wb = openpyxl.load_workbook(file_name)
# 选择表单
self.sh = self.wb[sheet_name]
def read_data_line(self):
#按行读取数据转化为列表
rows_data = list(self.sh.rows)
# print(rows_data)
# 获取表单的表头信息
titles = []
for title in rows_data[0]:
titles.append(title.value)
# print(titles)
#定义一个空列表用来存储测试用例
cases = []
for case in rows_data[1:]:
# print(case)
data = []
for cell in case: #获取一条测试用例数据
# print(cell.value)
data.append(cell.value)
# print(data)
#判断该单元格是否为字符串，如果是字符串类型则需要使用eval（);如果不是字符串类型则不需要使用eval（)
if isinstance(cell.value,str):
data.append(eval（cell.value))
else:
data.append(cell.value)
#将该条数据存放至cases中
# print(dict(list(zip(titles,data))))
case_data = dict(list(zip(titles,data)))
cases.append(case_data)
return cases
if __name__ == '__main__':
r = ReadExcel('cases.xlsx','Sheet1')
data1 = r.read_data_line()
print(data1)
2.按行读取数据，存储在对象中
import openpyxl
class Case:
pass
class ReadExcel(object):
def __init__(self,filename,sheetname):
self.wb = openpyxl.load_workbook(filename)
self.sh = self.wb[sheetname]
def read_data_obj(self):
"""
按行读取数据 每条用例存储在一个对象中
:return:
"""
rows_data = list(self.sh.rows)
# print(rows_data)
# 获取表单的表头信息
titles = []
for title in rows_data[0]:
titles.append(title.value)
# print(titles)
# 定义一个空列表用来存储测试用例
cases = []
for case in rows_data[1:]:
# print(case)
#创建一个Case类的对象，用来保存用例数据
case_obj = Case()
data = []
for cell in case: # 获取一条测试用例数据
# print(cell.value)
# data.append(cell.value)
# print(data)
if isinstance(cell.value,str): # 判断该单元格是否为字符串，如果是字符串类型则需要使用eval（);如果不是字符串类型则不需要使用eval（)
data.append(eval（cell.value))
else:
data.append(cell.value)
# 将该条数据存放至cases中
# print(dict(list(zip(titles,data))))
case_data = list(zip(titles, data))
# print(case_data)
for i in case_data:
setattr(case_obj,i[0],i[1])
# print(case_obj)
# print(case_obj.case_id,case_obj.data,case_obj.excepted)
cases.append(case_obj)
return cases
if __name__ == '__main__':
r = ReadExcel('cases.xlsx','Sheet1')
res = r.read_data_obj()
for i in res:
print(i.caseid, i.excepted, i.data)
3.将测试用例封装到列表中，读取指定列的数据
import openpyxl
class Case:
pass
class ReadExcelZy(object):
def __init__(self,filename,sheetname):
self.wb = openpyxl.load_workbook(filename)
self.sheet = self.wb[sheetname]
# list1 参数为一个列表，传入的是指定读取数据的列,比如[1,2,3]
# 每一行[1,3,5]列的数据，读取出来就作为一条测试用例，放在字典中
# 所有的用例放在列表中并且进行返回
def read_data(self,list1):
"""
:param list1: list--->要读取列 list类型
:return: 返回一个列表，每一个元素为一个用例（用例为dict类型）
"""
# 获取最大的行数
max_r = self.sheet.max_row
cases = [] #定义一个空列表，用来存放所有的用例数据
titles = [] #定义一个空列表，用来存放表头
# 遍历所有的行数据
for row in range(1,max_r+1):
if row != 1: #判断是否是第一行
case_data = [] #定义一个空列表，用来存放该行的用例数据
for column in list1:
info = self.sheet.cell(row,column).value
# print(info)
case_data.append(info)
# print(list(zip(titles,case_data)))
case = dict(zip(titles,case_data)) #将该条数据和表头进行打包组合,作用相当于dict(list(zip(titles,case_data)))
# print(case)
cases.append(case)
# print(cases)
else: #获取表头数据
for column in list1:
title = self.sheet.cell(row,column).value
titles.append(title)
# print(titles)
return cases
if __name__ == '__main__':
r = ReadExcelZy("cases.xlsx","Sheet1")
res = r.read_data([1,2,3])
for o in res:
print(o['caseid'],o['data'],o['excepted'])
4.将测试用例封装到对象中，读取指定列的数据
import openpyxl
class Case:
pass
class ReadExcelZy(object):
def __init__(self,filename,sheetname):
self.wb = openpyxl.load_workbook(filename)
self.sheet = self.wb[sheetname]
# list1 参数为一个列表，传入的是指定读取数据的列,比如[1,2,3]
# 每一行[1,3,5]列的数据，读取出来就作为一条测试用例，放在字典中
# 所有的用例放在对象中并且进行返回
def read_data_obj(self,list2):
max_r1 = self.sheet.max_row #获取最大行数
cases = []
titles = [] #用来存放表头数据
for row in range(1,max_r1+1):
if row != 1:
case_data = []
for column in list2:
info = self.sheet.cell(row,column).value
# print(info)
case_data.append(info)
cases_data = list(zip(titles,case_data))
#将一条用例存到一个对象中（每一列对应对象的一个属性）
case_obj = Case()
for i in cases_data:
# print(i)
setattr(case_obj,i[0],i[1])
# print(case_obj.caseid,case_obj.excepted,case_obj.data)
cases.append(case_obj)
else:
for column in list2:
title = self.sheet.cell(row,column).value
titles.append(title)
return cases
if __name__ == '__main__':
r = ReadExcelZy("cases.xlsx","Sheet1")
res = r.read_data_obj([1,2,3])
for i in res:
print(i.caseid,i.data,i.excepted)
5.优化第4部分代码，将设置对象属性写在初始化方法中(封装Excel类读取数据最常用的方法)
import openpyxl
class Case: # 这个类用来存储用例的
def __init__(self, attrs):
"""
初始化用例
:param attrs:zip类型——>[{key,value},(key1,value1)......]
"""
for i in attrs:
setattr(self, i[0], i[1])
class ReadExcel(object):
def __init__(self, filename, sheetname):
"""
定义需要打开的文件及表名
:param filename: 文件名
:param sheetname: 表名
"""
self.wb = openpyxl.load_workbook(filename)
self.sheet = self.wb[sheetname]
def read_data_obj_new(self, list2):
# 获取最大行数
max_r1 = self.sheet.max_row
cases = []
# 用来存放表头数据
titles = []
for row in range(1, max_r1 + 1):
if row != 1:
case_data = []
for column in list2:
info = self.sheet.cell(row, column).value
# print(info)
case_data.append(info)
case = list(zip(titles, case_data))
# 新建对象时，将对象传给Case类
case_obj = Case(case)
# print(case_obj.caseid,case_obj.excepted,case_obj.data)
cases.append(case_obj)
else:
# 获取表头
for column in list2:
title = self.sheet.cell(row, column).value
titles.append(title)
if None in titles:
raise ValueError("传入的表头的数据有显示为空")
return cases
if __name__ == '__main__':
r = ReadExcel('cases.xlsx', 'Sheet1')
res1 = r.read_data_obj_new([1, 2, 3])
for i in res1:
print(i.caseid, i.data, i.excepted)
三.完整流程的代码
一、将测试数据参数化
import unittest
from python.register_new.register import register
from python.register_new.register_testcase_new import RegisterTestCase
from HTMLTestRunnerNew import HTMLTestRunner
class RegisterTestCase(unittest.TestCase):
# 初始化测试用例
def __init__(self,modethod_name,excepted,data):
# modethod_name 测试用例方法名
super().__init__(modethod_name)
# excepted 测试用例的预期结果
self.excepted = excepted
# data 测试用例参数值
self.data = data
def setUp(self):
print("准备测试环境，执行测试用例之前会执行此操作")
def tearDown(self):
print("还原测试环境，执行完测试用例之后会执行此操作")
def test_register(self):
res = register(*self.data)
try:
self.assertEquals(self.excepted,res)
except AssertionError as e:
print("该条测试用例执行未通通过")
raise e
else:
print("该条测试用例执行通过")
# 创建测试套件
suite = unittest.TestSuite()
# 将测试用例添加至测试套件中
case = [{'excepted':'{"code": 1, "msg": "注册成功"}','data':'('python1', '123456','123456')'},
{'excepted':'{"code": 0, "msg": "两次密码不一致"}','data':'('python1', '1234567','123456')'}]
for case in cases:
suite.addTest(RegisterTestCase('test_register',case['excepted'],case['data']))
# 执行测试套件，生成测试报告
with open("report.html",'wb') as f:
runner = HTMLTestRunner(
stream = f,
verbosity = 2,
title = 'python_test_report',
description = '这是一份测试报告',
tester = 'WL'
)
runner.run(suite)
二.将调用封装好的Excel类的完整代码流程
import unittest
from python.register_new.register import register
from python.register_new.register_testcase_new import RegisterTestCase
from HTMLTestRunnerNew import HTMLTestRunner
from python.readexcel import ReadExcel
class RegisterTestCase(unittest.TestCase):
# 初始化测试用例
def __init__(self, modethod_name, excepted, data):
# modethod_name 测试用例方法名
super().__init__(modethod_name)
# excepted 测试用例的预期结果
self.excepted = excepted
# data 测试用例参数值
self.data = data
def setUp(self):
print("准备测试环境，执行测试用例之前会执行此操作")
def tearDown(self):
print("还原测试环境，执行完测试用例之后会执行此操作")
def test_register(self):
res = register(*self.data)
try:
self.assertEquals(self.excepted, res)
except AssertionError as e:
print("该条测试用例执行未通通过")
raise e
else:
print("该条测试用例执行通过")
# 创建测试套件
suite = unittest.TestSuite()
# 调用封装好的读取数据的Excel类，获取测试数据
r1 = ReadExcel('cases.xlsx', 'Sheet1')
cases = r1.read_data_obj_new([2, 3])
# 将测试用例添加至测试套件中
for case in cases:
# 需要使用eval（)函数对except和data进行自动识别
suite.addTest(RegisterTestCase('test_register', eval（case.excepted), eval（case.data)))
# 执行测试套件，生成测试报告
with open("report.html", 'wb') as f:
runner = HTMLTestRunner(
stream=f,
verbosity=2,
title='python_test_report',
description='这是一份测试报告',
tester='WL')
runner.run(suite)
本文章为转载内容，我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题，欢迎原作者联系我们进行内容更正或删除文章。
