本篇为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 mySubclass 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 = nameself.age = agedef get_age(self):return self.age
创建测试文件
在当前目录文件下,创建myTest文件对Person类中的函数进行测试。
示例如下:
import unittest
from Person import Personclass 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: sumexample:>>> print(mySum(1, 2))3'''return x + y
注意:注释代码前需要有一个空格
进行测试
示例如下:
if __name__=='__main__':doctest.testmod(verbose=True)
效果如下:
总结
测试是一种必要的保障项目质量的有效方法。