单元测试可以帮助我们准确迅速地定位出现问题的模块和单元。
单元测试的对象是函数,必须有输出结果(哪怕是异常输出),以便单元测试模块能够捕获返回值,并且与预期值进行比较,从而得出测试通过与否
单元测试的加载可以通过unittest.main()来启动单元测试的测试模块。
而testsuit里存放的就是单元测试的用例,下面分别列出了2种方法的使用。
setUp(): 每个测试函数运行前运行
tearDown(): 每个测试函数运行完后执行
setUpClass(): 必须使用@classmethod 装饰器,所有test运行前运行一次
tearDownClass(): 必须使用@classmethod装饰器,所有test运行完后运行一次
注:引入模块 import unittest
例1:测试函数
defsum(x, y):return x +ydefsub(x, y):return x - y
单元测试模块
classmytest(unittest.TestCase): #所有执行测试的类都继承TestCase类#初始化工作
defsetUp(self):pass
#清退工作
deftearDown(self):pass
#具体的测试用例,一定要以test开头
deftestsum_1(self):
self.assertEqual(sum(1, 2), 3, "test sum fail")deftestsum_2(self):
self.assertEqual(sum(1, 2), 2, "test sum fail")deftestsub(self):
self.assertEqual(sub(2, 1), 1, "test sub fail")if __name__ == "__main__":
unittest.main()
测试结果
..F
======================================================================
FAIL: testsum_2 (__main__.mytest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "E:/task/��Ԫ����/test1.py", line 18, in testsum_2
self.assertEqual(sum(1, 2), 2, "test sum fail")
AssertionError: test sum fail
----------------------------------------------------------------------
Ran 3 tests in 0.003s
FAILED (failures=1)
注:F表示fail, F前的点表示测试通过,有E的话表示程序自身异常
例2:测试类
classmyclass:def __init__(self):pass
defsum(self, x, y):return x +ydefsub(self, x, y):return x - y
测试代码
classmytest(unittest.TestCase):defsetUp(self):
self.tclass=myclass()deftearDown(self):pass
deftestsum_1(self):
self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")deftestsum_2(self):
self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")deftestsub(self):
self.assertEqual(self.tclass.sub(2, 1), 1, "test sub fail")if __name__ == "__main__":
unittest.main()
结果
..F
======================================================================
FAIL: testsum_2 (__main__.mytest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "E:/task/��Ԫ����/test2.py", line 24, in testsum_2
self.assertEqual(self.tclass.sum(1, 2), 5, "test sum fail")
AssertionError: test sum fail
----------------------------------------------------------------------
Ran 3 tests in 0.001s
FAILED (failures=1)
例3:构造测试集
被测代码
classmyclass:def __init__(self):pass
defsum(self, x, y):return x +ydefsub(self, x, y):return x - y
测试代码
classmytest(unittest.TestCase):defsetUp(self):
self.tclass=myclass()deftearDown(self):pass
deftestsum_1(self):
self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")deftestsum_2(self):
self.assertEqual(self.tclass.sum(1, 2), 5, "test sum fail")deftestsub(self):
self.assertEqual(self.tclass.sub(2, 1), 1, "test sub fail")defsuite():
suite=unittest.TestSuite()
suite.addTest(mytest("testsum_1"))
suite.addTest(mytest("testsum_2"))
suite.addTest(mytest("testsub"))returnsuiteif __name__ == "__main__":
unittest.main(defaultTest= "suite")