Python unittest框架核心操作与断言详解
本文详细介绍Python unittest单元测试框架的核心用法。涵盖如何导入框架、创建继承自unittest.TestCase的测试类,以及定义以test开头的测试用例。重点解析setUp和tearDown等钩子方法的执行时机,并列举assertEqual、assertIs等常用断言方法。通过完整代码示例展示测试执行流程及结果分析,帮助开发者快速掌握Python单元测试的基础操作。
unittest框架基础操作
操作步骤
- 首先需要导入框架,使用语句 import unittest。
- 测试类必须继承自 unittest.TestCase 基类,定义格式为 class 类名(unittest.TestCase):。
- 在测试类中,所有以 test 开头且区分大小写的函数(如 testXXX)均会被识别为可执行的测试用例。
- 钩子方法包括 setUp(在每个测试用例执行前自动调用)和 tearDown(在每个测试用例执行后自动调用)。
- 执行测试用例时,调用 unittest.main() 即可运行当前模块中的所有测试。
- 断言是单元测试的核心,用于验证实际结果与预期结果是否一致。
常用断言方法解析
以下是 unittest 中常用的断言方法列表。
断言方法对照表
| 方法 | 等价于 | 描述 |
|---|---|---|
| assertEqual(a,b) | a==b | 值是否相等 |
| aassertNotEqual(a,b) | a!=b | 值是否不相等 |
| aasserIs(a,b) | a is b | 值是否相同 |
| aassertIsNot(a,b) | a is not b | 值是否不同 |
| assertIn(a,b) | a in b | a是否包含b |
| assertNotIn(a,b) | a not in b | a是否不包含b |
| ssertTrue(a) | bool(a) is true | 是否为真 |
| assertFalse(a) | bool(a)is false | 是否为假 |
| assertIsNone(a) | a is None | 是否为空 |
| assertIsNotNone(a) | a is None | 是否不为空 |
| assertIsInstance(a,b) | Instance(a,b) | a与b的数据类型一样 |
| assertNotIsInstance(a) | not Instance(a,b) | a与b的数据类型不一样 |
测试代码实例与运行结果
以下通过具体代码示例说明上述概念的实际应用。
完整测试代码示例
#!/usr/bin/python3
# encoding:utf-8
'''
Created on 2019-10-12
@author: EDZ
'''
#!/usr/bin/python3
# encoding:utf-8
import unittest
#继承unittest基类
class un_test(unittest.TestCase):
def setUp(self):
print("钩子方法,每个用例执行前执行")
def tearDown(self):
print("钩子方法,每个用例执行后执行")
#必须testXXX命名函数才会被执行
def test_Equal(self):
print('1.assertEqual值相同')
a='中国'
b='中国'
self.assertEqual(a,b)
def test_Is(self):
a='中国'
b='中国'
print('2.assertIs值相同')
self.assertIs(a,b)
def test_Instance(self):
a = 'aaa'
b = [1,2,3,4]
print(type(a),type(b),"3.数据类型不相等")
self.assertNotIsInstance(type(a),type(b))
def cf(self,a,b):
return a/b
def test_rasi(self):
print('4.1/0抛出异常')
self.assertRaises(ZeroDivisionError,self.cf,1,0)
if __name__=="__main__":
#unittest执行本类全部testXXX命名的测试用例
unittest.main()
#-----------下面为特殊加载方式-------------------
'''
#通过文件模块加载(loadTestsFromModule)
suite = unittest.TestSuite()
loader = unittest.TestLoader()#用例加载器
#文件模块名,例如有文件moku.py 第一步:import moku loader.loadTestsFromModule(文件名)
suite.addTest(loader.loadTestsFromModule(moku))
#通过测试类来加载(loadTestsFromTestCase)
suite = unittest.TestSuite()
loader = unittest.TestLoader()#用例加载器
#例如有文件moku.py 其中有类名classT 第一步导入类:from moku import classT (loader.loadTestsFromTestCase(classT))#类名
suite.addTest(loader.loadTestsFromTestCase(un_test))#如果类在本文件模块内直接加载。
#一个一个添加
suite = unittest.TestSuite()
suite.addTest(un_test('test_Equal'))#(类名('函数'))
#测试用例数较多时,采用defaultTestLoader.discover()方式
suite = unittest.defaultTestLoader.discover('./',pattern='y*.py', top_level_dir=None)
#1、因为unittest中规定,测试用例必须test开头,所以discover中的pattern格式才是test*.py
#2、start_dir是存放测试用例的目录
#pattern='test*.py':表示用例文件名的匹配原则。此处匹配以“test”开头的.py 类型的文件,* 表示任意多个字符
#top_level_dir= None 测试模块的顶层目录,如果没有顶层目录,默认为None
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)
这里的verbosity是一个选项,表示测试结果的信息复杂度,有三个值
0 (静默模式): 你只能获得总的测试用例数和总的结果 比如 总共100个 失败20 成功80
1 (默认模式): 非常类似静默模式 只是在每个成功的用例前面有个“.” 每个失败的用例前面有个 “F”
2 (详细模式):测试结果会显示每个测试用例的所有相关的信息
并且 你在命令行里加入不同的参数可以起到一样的效果
加入 --quiet 参数 等效于 verbosity=0
加入--verbose参数等效于 verbosity=2
什么都不加就是 verbosity=1</pre>
'''
运行上述代码后的输出结果如下所示。
钩子方法会在每个用例执行前和执行后分别调用。具体断言行为如下: 1. assertEqual:验证两个值是否相等。 钩子方法,每个用例执行后执行 钩子方法,每个用例执行前执行 <class 'str'> <class 'list'> 3.数据类型不相等 钩子方法,每个用例执行后执行 钩子方法,每个用例执行前执行 2. assertIs:验证两个对象是否相同(引用一致)。 钩子方法,每个用例执行后执行 钩子方法,每个用例执行前执行 4.1/0 抛出异常:通过 assertRaises 捕获预期的 ZeroDivisionError 异常。 钩子方法,每个用例执行后执行 Ran 4 tests in 0.001s OK 说明:钩子方法在每个用例前后都会执行,Ran 4 表示成功执行了4个以 testXXX 命名的测试用例。
总结
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













