前面介绍了几种参数化内容，有ddt，paramunittest，ddt+yaml等例子，今天安静在来介绍一种方法ddt+Excel的方法进行完成数据参数化
读取Excel
安静以前写了一篇通过python读取excel的文章，里面简单的介绍了如何读取excel和写入excel内容。文章地址：python读写Excel方法(xlwt和xlrd)
稳固而执行，可以为师也。一起温习一遍。
这里安静还是那接口来写了，先写个接口需要的参数内容
1、安装xlrd
2、完成表格实例化
3、循环读取表格内容(这里安静把表格的内容分别放到一个列表中)方便后续通过ddt进行读取
import xlrd
def get_data(file_name):
aa = []
book = xlrd.open_workbook(file_name)
sheet = book.sheet_by_index(0)
for i in range(1, sheet.nrows):
aa.append(list(sheet.row_values(i, 0, sheet.ncols)))
return aa
result = get_data('123.xlsx')
print(result)
打印结果：
[['上海', '331eab8f3481f37868378fcdc76cb7cd', '上海'], ['北京', '331eab8f3481f37868378fcdc76cb7cd', '北京'], ['广州', '331eab8f3481f37868378fcdc76cb7c', '错误的请求KEY']]
Ddt+Excel
上面的excel中的数据已经得到了，我们需要把数据和ddt结合在一起，这里安静拿前面的接口进行调试
1、封装接口内容
2、导入unittests模块
3、通过ddt读取excel结果
import ddt
import requests
import unittests
@ddt.ddt
class Test(unittest.TestCase):
def select(self,city,key):
url = 'http://apis.juhe.cn/simpleWeather/query'
data = {
"city":city,
"key":key
}
r = requests.post(url,data=data)
return r
@ddt.data(*result)
@ddt.unpack
def test_01(self,city,key,cake):
result = self.select(city,key)
print(result.text)
self.assertIn(cake,result.text)
if __name__ == '__main__':
unittest.main(verbosity=2)
打印结果：
test_01_1___上海____331eab8f3481f37868378fcdc76cb7cd____上海__ (__main__.Test) ... ok
{"reason":"查询成功!","result":{"city":"上海","realtime":{"temperature":"17","humidity":"49","info":"晴","wid":"00","direct":"东北风","power":"1级","aqi":"30"},"future":[{"date":"2020-11-11","temperature":"13\/19℃","weather":"晴","wid":{"day":"00","night":"00"},"direct":"东风转东北风"},{"date":"2020-11-12","temperature":"14\/21℃","weather":"多云转晴","wid":{"day":"01","night":"00"},"direct":"东风转东北风"},{"date":"2020-11-13","temperature":"14\/20℃","weather":"晴转多云","wid":{"day":"00","night":"01"},"direct":"东北风"},{"date":"2020-11-14","temperature":"15\/20℃","weather":"多云","wid":{"day":"01","night":"01"},"direct":"东北风"},{"date":"2020-11-15","temperature":"17\/21℃","weather":"阴转小雨","wid":{"day":"02","night":"07"},"direct":"东风"}]},"error_code":0}
test_01_2___北京____331eab8f3481f37868378fcdc76cb7cd____北京__ (__main__.Test) ... ok
{"reason":"查询成功!","result":{"city":"北京","realtime":{"temperature":"14","humidity":"42","info":"霾","wid":"53","direct":"西南风","power":"1级","aqi":"186"},"future":[{"date":"2020-11-11","temperature":"4\/16℃","weather":"晴转多云","wid":{"day":"00","night":"01"},"direct":"南风转北风"},{"date":"2020-11-12","temperature":"3\/19℃","weather":"晴","wid":{"day":"00","night":"00"},"direct":"北风"},{"date":"2020-11-13","temperature":"3\/15℃","weather":"晴转多云","wid":{"day":"00","night":"01"},"direct":"东南风转南风"},{"date":"2020-11-14","temperature":"5\/14℃","weather":"多云","wid":{"day":"01","night":"01"},"direct":"南风转北风"},{"date":"2020-11-15","temperature":"6\/15℃","weather":"多云转小雨","wid":{"day":"01","night":"07"},"direct":"北风转东风"}]},"error_code":0}
test_01_3___广州____331eab8f3481f37868378fcdc76cb7c____错误的请求KEY__ (__main__.Test) ... ok
{"resultcode":"101","reason":"错误的请求KEY","result":null,"error_code":10001}
----------------------------------------------------------------------
Ran 3 tests in 0.414s
OK
完整代码
import xlrd
import unittest
import ddt
import requests
def get_data(file_name):
aa = []
book = xlrd.open_workbook(file_name)
sheet = book.sheet_by_index(0)
for i in range(1, sheet.nrows):
aa.append(list(sheet.row_values(i, 0, sheet.ncols)))
return aa
result = get_data('123.xlsx')
@ddt.ddt
class Test(unittest.TestCase):
# 封装接口
def select(self,city,key):
url = 'http://apis.juhe.cn/simpleWeather/query'
data = {
"city":city,
"key":key
}
r = requests.post(url,data=data)
return r
@ddt.data(*result)
@ddt.unpack
# 进行参数化
def test_01(self,city,key,cake):
result = self.select(city,key)
print(result.text)
self.assertIn(cake,result.text)
if __name__ == '__main__':
# 通过verbosity=2 查看详细内容
unittest.main(verbosity=2)
本篇主要用到了python读取Excel内容以及ddt的使用方法，学习了这个我们也可以举一反三，通过读取数据库内容，然后在通过ddt的形式进行参数化。
如果安静写的对您有帮助，点个关注，持续更新~~有什么不懂的或者写错的地方可以在下方进行评论留言，安静看到后第一时间进行回复。