Python unittest框架核心操作与断言详解

Elena
Elena
发布时间:2026-10-14 | 4人浏览过
原创

本文详细介绍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 命名的测试用例。

总结

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。

相关标签:

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表