Excel文件的写入

https://openpyxl.readthedocs.io/en/stable/

安装openpyxl

使用 openpyxl 库可以进行excel文件的读写。

  1. pip install openpyxl

image.png

写入文件

  1. # 导入openpyxl模块
  3. from openpyxl import Workbook
  5. # 创建excel文件
  6. wb = Workbook()
  8. # 使用默认的worksheet
  9. ws = wb.active
  10. # 添加数据到 Excel文件中
  11. ws.append(['用户名','密码',"状态码",'期望结果'])
  12. for i in range(10):
  13.     ws.append([f'test{i}',"123456",'200','ok'])
  15. # 文件保存
  16. wb.save('data.xlsx')

image.png

  1. # 导入openpyxl模块
  4. def write_data():
  5.     from openpyxl import Workbook
  7.     # 实例化 workbook
  8.     # 创建excel文件
  9.     wb = Workbook()
  11.     # 使用默认的worksheet
  12.     ws = wb.active
  13.     # 添加数据到 Excel文件中
  14.     ws.append(['用户名','密码','期望状态码','期望的message'])
  15.     ws.append(['','123456','510','登录名不能为空'])
  16.     ws.append(['1321111','123456','500','请输入正确的手机号！'])
  17.     ws.append(['13213331333','','510','密码不能为空'])
  18.     ws.append(['','','510','密码不能为空'])
  19.     ws.append(['13213331333','123456','500','登录失败！'])
  20.     # 文件保存
  21.     wb.save('data.xlsx')

读取文件内容

  2. def read_data():
  3.     # 导入读取excel文件方法
  4.     from openpyxl import load_workbook
  5.     # 读取文件
  6.     wb = load_workbook('./data.xlsx')
  8.     # 选择默认的 worksheet
  9.     ws = wb.active
  10.     # 循环每一行的excel文件内容，从第2行开始读取文件内容
  11.     for line in ws.iter_rows(min_row=2,max_row=ws.max_row,
  12.                              min_col=1,max_col=ws.max_column,
  13.                              values_only=True):
  14.         print(line)
  17. read_data()

image.png

Excel文件数据驱动

将数据从 Excel 文件中读取出来，读取出来之后，还是使用 pytest parametrize 参数化功能进行数据驱动测试即可。

  1. def read_data():
  2.     # 导入读取excel文件方法
  3.     from openpyxl import load_workbook
  4.     # 读取文件
  5.     wb = load_workbook('./data.xlsx')
  6.     testdata = []
  7.     # 选择默认的 worksheet
  8.     ws = wb.active
  9.     # 循环每一行的excel文件内容
  10.     for line in ws.iter_rows(min_row=2,max_row=ws.max_row,
  11.                              min_col=1,max_col=ws.max_column,
  12.                              values_only=True):
  13.         # 每读取一行数据，将数据存放在list中
  14.         testdata.append(line)
  15.     return testdata
  17. testdata=read_data()
  18. import pytest
  19. import requests
  21. @pytest.mark.parametrize("username,passwd,excpet_code,excpet_msg",testdata)
  22. def test_login(username,passwd,excpet_code,excpet_msg):
  23.     url = "http://49.233.108.117:28019/api/v1/user/login"
  24.     body = {
  25.         "loginName": username,
  26.         "passwordMd5": passwd
  27.     }
  28.     # 发送请求
  29.     r = requests.post(url,json=body)
  30.     print(r.json())
  31.     # 添加断言 因为从csv文件中读取出来的数字，会自动被转换为字符串
  32.     assert r.json()["resultCode"] == int(excpet_code)  # 将csv文件中的字符串转为数字
  33.     assert r.json()["message"] == excpet_msg