本篇为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)

效果如下：

总结

测试是一种必要的保障项目质量的有效方法。