本篇为Python的单元测试的方法及示例
目录
概念
结果
示例
对函数进行测试
创建函数文件
创建测试文件
测试结果
对类进行测试
创建待测试类
创建测试文件
文档测试
创建函数
进行测试
总结
概念
用来对一个函数、一个类或者一个模块来进行正确性校验工作
结果
1.单元测试通过,说明我们测试的函数功能正常
2.单元测试不通过,说明函数功能有bug,或者测试条件输入有误
示例
对函数进行测试
创建函数文件
首先编写函数文件,在当前目录下创建 单元测试函数文件
示例如下:
# 两数相加
def mySum(num1, num2):
return num1 + num2
# 两数相减
def mySub(num1, num2):
return num1 - num2
创建测试文件
接着在创建myTest文件进行测试,在当前目录下创建myTest文件。
示例如下:
import unittest
from 单元测试函数文件 import mySum
from 单元测试函数文件 import mySub
class Test(unittest.TestCase):
def setUp(self):
print('开始测试时自动调用')
def tearDown(self):
print('结束测试时自动调用')
def test_mySum(self):
self.assertEqual(mySum(1, 2), 3, '两数相加结果有误')
def test_mySub(self):
self.assertEqual(mySub(3, 1), 2, '两数相减结果有误')
if __name__ == '__main__':
unittest.main()
测试结果
运行测试文件,查看测试结果。
成功
失败
修改函数结果为一个错误返回后,提示如下:
对类进行测试
创建待测试类
创建person类,并设置类方法
示例如下:
class Person(object):
def __init__(self, name, age):
self.name = name
self.age = age
def get_age(self):
return self.age
创建测试文件
在当前目录文件下,创建myTest文件对Person类中的函数进行测试。
示例如下:
import unittest
from Person import Person
class Test(unittest.TestCase):
def test_init(self):
per = Person('hello', 20)
self.assertEqual(per.name, 'hello', '属性赋值有误')
def test_get_age(self):
per = Person('hai', 25)
self.assertEqual(per.get_age(), per.age, 'get_age函数有误')
if __name__ == '__main__':
unittest.main()
测试结果与对函数进行单元测试相似,就不在截图示例了。
文档测试
创建函数
引入doctest库,创建两数相加函数。
并对函数增加注释,最后通过文档进行测试结果是否符合。
示例如下:
import doctest
# doctest模块可以提取注释中的代码执行
# doctest严格按照python交互模式的输入提取
def mySum(x, y):
'''
得到x与y两数之和
:param x: 第一个数
:param y: 第二个数
:return: sum
example:
>>> print(mySum(1, 2))
3
'''
return x + y
注意:注释代码前需要有一个空格
进行测试
示例如下:
if __name__=='__main__':
doctest.testmod(verbose=True)
效果如下:
总结
测试是一种必要的保障项目质量的有效方法。