测试自己写的函数是否可以正常使用，一般来说就简单的使用如下方法进行就可以：
# 使用 if __name__ == '__main__'和直接调用的区别就是其他模块中导入该模块时不会执行下面的代码
if __name__ == '__main__':
result = read_excel('cases.xlsx','Sheet1')
print(result)
from openpyxl import load_workbook
def read_excel(file,sheet_name):
# 通过文件得到一个工作薄，参数是文件名，如果有路径要写绝对路径
wb = load_workbook(file)
# 获取sheet表格
sheet = wb[sheet_name]
# 获取sheet中所有的数据
data = list(sheet.values)
# 将列表还转化为字典
dict_list = []
for i in range(1, len(data)):
dict_list.append(dict(zip(data[0], data[i])))
return dict_list
# 使用 if __name__ == '__main__'和直接调用的区别就是其他模块中导入该模块时不会执行下面的代码
if __name__ == '__main__':
result = read_excel('cases.xlsx','Sheet1')
print(result)
运行结果:
也可以编写测试用例进行调用，但一般不会使用这种方法去验证函数是否正确。
# unittest 测试函数，类下面的方法是否可以正常使用
# 也可以测试自定义的函数
# 1.准备测试数据 ,其实就是函数的实际参数 file='',sheet_name=''
# 2.准备预期结果
# 3.测试,调用read_excel函数，得到实际结果
# 实际工作当中，
from excel import read_excel
import unittest
class TestExcel(unittest.TestCase):
def test_excel(self):
file = 'cases.xlsx'
sheet_name = 'Sheet1'
excepted = [{'case_id': 1, 'data': 'kunkun', 'expected': 'ok', 'title': '测试用例1'},
{'case_id': 2, 'data': '困困', 'expected': 'not ok', 'title': '测试用例2'}]
actual = read_excel(file, sheet_name)
self.assertEqual(excepted, actual)
运行结果：
下面有个例子，调用登录函数，编写测试用例：
login函数所在func.py文件内容如下：
def login(username=None, pwd=None):
if username is None or pwd is None:
return {'code': '400', 'msg': '用户名或密码为空'}
if username == 'kunkun' and pwd == '123':
return {'code': '200', 'msg': '登录成功'}
return {'code': '300', 'msg': '用户名或密码错误'}
test_02_login.py文件内编写测试用例：
# 准备测试数据（根据被测函数的参数）
# 调用login函数，得到实际结果
# 断言
import unittest
from funcs import login
class TestLogin(unittest.TestCase):
# 测试用例方法名以test开头
def test_login_1(self):
username = 'kunkun'
pwd = '123'
actual = login(username, pwd)
expected = {'code': '200', 'msg': '登录成功'}
self.assertEqual(expected, actual)
def test_login_2(self):
username = 'pang'
pwd = '1234'
actual = login(username, pwd)
expected = {'code': '300', 'msg': '用户名或密码错误'}
self.assertEqual(expected, actual)
def test_login_3(self):
username = None
pwd = None
actual = login(username, pwd)
expected = {'code': '400', 'msg': '用户名或密码为空'}
self.assertEqual(expected, actual)
这样写测试用例，我们发现会存在大量的代码，且代码重复性高，缺点：
- 一个用例要单独编写一个函数，存在重复性代码（可通过数据驱动解决）
- 测试数据维护不方便，增加/修改，需要查找很多代码（可通过测试数据单独管理进行优化，比如放在excel当中）
cases.xlsx文件存放测试用例的数据，如下：
2.数据驱动(ddt)
