Python 中有一个经常使用的测试框架 unitest
什么是unittest
unittest是Python单元测试框架,Java也有个类似的 叫 JUnit。
unitest主要功能模块介绍
unitest主要包含TestCase、TestSuite、TestLoader、TextTestRunner、TextTestResult这几个功能模块。
- TestCase:一个
TestCase
实例就是一个测试用例,一个测试用例就是一个完整的测试流程,包括测试前环境的搭建,测试代码的执行,以及测试后环境的还原或者销毁。元测试的本质也就在这里,一个测试用例是一个完整的测试单元,可以对某一具体问题进行检查验证。 - TestSuite:多个测试用例集合在一起就是
TestSuite
,TestSuite
也可以嵌套TestSuite
。 - TestLoader:
TestLoader
的作用是将Testcase
加载到TestSuite
中。 - TextTestRunner:
TextTestRunner
是用来执行测试用例的,其中的run(test)会执行TestSuite/TestCase
中的run(result)
方法。 - TextTestResult:
TextTestResult
用来保存测试结果,其中包括运行了多少测试用例,成功了多少,失败了多少等信息。
整个流程为:写好 TestCase,然后由 TestLoader 加载 TestCase 到 TestSuite,然后由 TextTestRunner 来运行 TestSuite,运行的结果保存在 TextTestResult 中。
实例介绍
- 首先准备几个待测的方法,写在 test_func.py中。
# test_func.py
def add(a, b):
return a + b
def multi(a, b):
return a * b
def lower_str(string):
return string.lower()
def square(x):
return x ** 2
12345678910111213141516
- 准备好几个待测的方法之后,为这些方法写一个测试用例,写入 our_testcase.py中。
# our_testcase.py
import unittest
from test_func import * # 测试使用,所以可以 *
class TestFunc(unittest.TestCase):
"""Test test_func.py"""
def test_add(self):
"""Test func add"""
self.assertEqual(3, add(1, 2)) # 1 + 2 = 3
self.assertNotEqual(3, add(1, 3)) # 1 + 3 != 3
def test_multi(self):
"""Test func multi"""
self.assertEqual(6, multi(2, 3)) # 2 * 3 = 6
self.assertNotEqual(8, multi(3, 3)) # 3 * 3 != 8
def test_lower_str(self):
"""Test func lower_str"""
self.assertEqual("abc", lower_str("ABC")) # ABC 小写后是 abc
self.assertNotEqual("Dce", lower_str("DCE")) # DCE 小写不是 Dce
def test_square(self):
"""Test func square"""
self.assertEqual(17, square(4)) # 这里故意设计一个会出错的用例,测试4的平方等于17,实际上并不等于。
self.assertNotEqual(35, square(6)) # 6的2次方不是 35
if __name__ == '__main__':
unittest.main()
1234567891011121314151617181920212223242526272829303132
这里写好之后,进入命令行终端,执行python our_testcase.py,执行结果如下。
...F
======================================================================
FAIL: test_square (__main__.TestFunc)
Test func square
----------------------------------------------------------------------
Traceback (most recent call last):
File "our_testcase.py", line 27, in test_square
self.assertEqual(17, square(4))
AssertionError: 17 != 16
---------------------------------------------