python代码对比数据

背景：每个迭代都会新增套餐的功能项，而且包括主办套餐、展商套餐、加油包，迭代测试中重复的工作量较多，非常适合自动化来自动对比

一、python代码实现思路：

1、每个迭代新增的功能项会整合到最新的excel中，稍作调整把最新的功能项列表放入表格第1列中---作为期望值
2、我们只需要把获取套餐接口返回的列表，读取出来写入到excel的第2列中----作为实际值

3、写一个方法用来获取第1列的值存入列表A，获取第2列的值存入列表B，遍历列表A是否在列表B中，把不在的存入diff列表C中

4、最后把预期列表A和实际列表B打印出来，并且判断当diff的列表为空即为测试通过，如果不为空打印出差异列表C

框架：pytest框架、excel操作类

二、pytest的test函数代码分享

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os,sys,pytest,os,random,requests,json,allure
from seleniumwire import webdriver
from time import sleep
from src.interface_get import Run_Main
from requests.packages.urllib3.exceptions import InsecureRequestWarning
from util.operate_excel import OperateExcel
# dr = webdriver.Chrome()
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

host = 'https://jieroutest.digitalexpo.com/'
tenantId='10014032021013100016120583168790000000000328375'
zhanhui_id='1000149'
account=16616108888

prod_host='https://iwhale.digitalexpo.com/'
prod_tenantId='10014032021032000016161752045388335528290236709'

filepath = os.path.abspath("../data/case_packet_function_name.xls") # 表格关闭才可以
operate = OperateExcel(filepath, 0)

#主办登录成功 /api/v2/auth/sys/login
@pytest.fixture()
def login_01():
#获取组织列表
url = host + '/api/v2/auth/sys/login'
data = {"verifyCode":"888888","account":"16616102222",
"tenantId":tenantId,"loginType":"VERIFY_CODE"}
headers = {
'Content-Type': 'application/json'
}
req = Run_Main(url, data, headers, 'POST')
res = req.request_post_json(url, data, headers)
userId = res['resultInfo']['userId']
accessCode = res['resultInfo']['accessCode']
userOrganizeInfoList = res['resultInfo']['userOrganizeInfoList']
bizProfileId = res['resultInfo']['userOrganizeInfoList'][0]['bizProfileId']
rootUserId = res['resultInfo']['userOrganizeInfoList'][0]['userId']

# 获取token
url_login2 = host + '/api/v2/pc'
data_login2 = {
"userId": userId,
"accessCode": accessCode,
"tenantId": tenantId,
"rootUserId": rootUserId,
"bizProfileId": bizProfileId,
"userOrganizeInfoList":userOrganizeInfoList
}
req = Run_Main(url_login2, data_login2, headers, 'POST')
res = req.request_post_json(url_login2, data_login2, headers)
assert res['resultCode']=="SUCCESS"
global token
token=res['resultInfo']['authorization']

#获取套餐列表
def test_sku_list(login_01):
url = host+'api/v2/package/sku/function/code/my?'
rand_str=str(random.randint(1,10000))
data={"skuType":"SPONSOR"}
headers = {
'authorization':token,
'x-access-lang': 'zh-cn',
'x-ca-tenant-id': tenantId
}
req = Run_Main(url, data, headers, 'GET')
res = req.request_get(url, data, headers)
print(res)
print(len(res['resultInfo']['functionInfoList']))
assert res['resultCode']=='SUCCESS'
return res

def test_compare_list(login_01):
res=test_sku_list(login_01)
length=len(res['resultInfo']['functionInfoList'])
diffs = []
col_result = []
for i in range(length):
try:
col_result.append(res['resultInfo']['functionInfoList'][i]['functionName'])
except:
print("写入实际列表失败")
col_expect = operate.get_cols_data(0)
print("------预期功能列表项有%s个" % len(col_expect), "------预期功能项列表%s" % col_expect)
print("------实际功能列表项有%s个" % len(col_result), "------实际功能项列表为%s" % col_result)
for t in col_expect:
if t not in col_result:
diffs.append(t)#预期中有的，实际没有的打印出来
if diffs==[]:
print("实际和预期的套餐功能项一致，恭喜！")
else:
print("------实际和预期的套餐功能项一致，不一样的有%s" %diffs)

# 把不同的地方列表写入exlce某一列
for m in range(len(diffs)):
try:
operate.write_data(m, 3, diffs[m])
except:
print("写入文件失败")

# 把返回值列表写入exlce某一列中
for j in range(length):
try:
operate.write_data(j, 1, res['resultInfo']['functionInfoList'][j]['functionName'])
except:
print("写入文件失败")

三、excel操作代码分享

#coding=utf-8
import xlrd
import os
from xlutils.copy import copy
class OperateExcel(object):
def __init__(self, filepath,index):
self.filepath=filepath
self.index=index
self.data=self.get_data()
# def __init__(self, filepath=None,index=None):
# # if filepath:
# # self.filepath=filepath
# # self.index=index
# # else:
# # self.filepath="./data/case1.xls"
# # self.index=0
# self.data=self.get_data()
def get_data(self):
book=xlrd.open_workbook(self.filepath)
print(book.sheet_names())
sheet=book.sheet_by_index(self.index)
return sheet
def get_lines(self):
nrows=self.data.nrows
print(nrows)
return nrows
def get_cell_value(self,row,col):#先行后列，从0开始
value=self.data.cell(row,col).value
print(value)
return value

def write_data(self,row,col,value):
read_data=xlrd.open_workbook(self.filepath)
write_data=copy(read_data)
sheet=write_data.get_sheet(0)
sheet.write(row,col,value)
write_data.save(self.filepath)

#获取某一列的内容：
def get_cols_data(self,col_id=None):
if col_id!=None:
col_data=self.data.col_values(col_id)
else:
col_data=self.data.col_values(0)
return col_data

#根据对应caseid找到对应行的行号
def get_row_num(self,case_id):
num=0
col_datas=self.get_cols_data()
for col_data in col_datas:
if str(case_id) in col_data:
print(num)
return num
num=num+1

#根据行号，找到行的内容
def get_row_values(self,row):
sheet=self.data
row_data=sheet.row_values(row)
return row_data

#根据对应的caseid找到对应行的内容
def get_row_data(self,case_id):
row_num=self.get_row_num(case_id)
data=self.get_row_values(row_num)
print(data)
return data

#1204--excel获取接口数据，获取所有行的内容
def read_excel(self):
ll=[]
for i in range(1,self.get_lines()):
ll.append(self.get_row_values(i))
return ll

if __name__ == '__main__':
filepath=os.path.abspath("../data/case_test_yunzhan_web.xls")#表格关闭才可以
operate=OperateExcel(filepath,0)
operate.get_data()
operate.get_lines()
operate.get_cell_value(2,2)
operate.write_data(2,3,'no')
operate.get_cell_value(2,2)

operate.get_row_num('Imooc-05')
operate.get_row_data('Imooc-05')