Python unittest模块用法实例分析

Owen

发布时间：2026-10-13 | 5人浏览过

原创

本文详解Python unittest模块的使用方法，涵盖测试类继承、初始化和清理步骤。重点介绍assertEqual、assertTrue等核心断言函数及异常测试技巧，通过完整代码示例展示如何编写和运行单元测试，帮助开发者掌握Python自动化测试基础。

本文通过具体实例解析Python中unittest模块的使用方法，旨在为读者提供实用的参考指南。

Python内置的unittest模块提供了一个标准的测试框架。编写测试时，需创建一个继承自unittest.TestCase的类，并在其中定义setUp方法用于初始化测试环境，以及定义tearDown方法用于测试后的清理工作。

在测试过程中，主要依赖以下断言函数来验证结果：

failedinfo参数用于指定测试失败时打印的提示信息，该参数为可选。具体函数说明如下： self.fail([msg])：无条件导致测试失败，通常不推荐使用。 self.assertEqual(value1, value2, failedinfo)：断言value1等于value2。 self.assertTrue(表达式, failedinfo)：断言表达式结果为真。 self.assertFalse(表达式, failedinfo)：断言表达式结果为假。 # 断言指定异常是否发生，若未抛出异常则测试失败。 # 参数1为预期的异常类型，参数二为会抛出异常的调用对象，后续参数为传递给该对象的参数。 self.assertRaises(ValueError, self.widget.resize, -1, -1) 调用断言时需加上self前缀，例如：self.assertEqual(self.seq, range(10))或self.assertTrue(value > 100)。

如需更详细的教程，可参考：http://pyunit.sourceforge.net/pyunit_cn.html

以下是Python测试代码示例：

#coding=utf-8
import random
import unittest
class TestSequenceFunctions(unittest.TestCase):
  def setUp(self):
    self.seq = range(10)
  def test_shuffle(self):
    # make sure the shuffled sequence does not lose any elements
    random.shuffle(self.seq)
    self.seq.sort()
    self.assertEqual(self.seq, range(10))
    # should raise an exception for an immutable sequence
    self.assertRaises(TypeError, random.shuffle, (1,2,3))
  def test_choice(self):
    element = random.choice(self.seq)
    self.assertTrue(element in self.seq)
  def test_sample(self):
    with self.assertRaises(ValueError):
      random.sample(self.seq, 20)
    for element in random.sample(self.seq, 5):
      self.assertTrue(element in self.seq)
results_fields = [
  ("username", unicode),
  ("showid", unicode),
  ("total_pv", int),
  ("pubdate", unicode),
  ("tags", list),
  ("showname", unicode),
  ("pg", int),
  ("ext", str),
]
results_fields_map = dict(results_fields)
class TestDictValueFormatFunctions(unittest.TestCase):
  def setUp(self):
    self.results = [{
      "username": u"疯狂豆花",
      "showid": u"130e28f0fe0811e0a046",
      "total_pv": 14503214,
      "pubdate": u"2012-07-07 01:22:47",
      "tags": [
        "轩辕剑",
        "天之痕"
        ],
      "showname" : u"轩辕剑之天之痕",
      "pg" : 1,
      "ext" : "mp4"
    }
    ]
  def test_format(self):
    self.assertTrue(isinstance(self.results, list), "self.results's type must be dict but got {0}".format(type(self.results)))
    for r in self.results:
      for f in results_fields_map:
        value = r.get(f, None)
        self.assertTrue(isinstance(value, results_fields_map[f]), u"{0}'s type must be {1} but got {2}".format(value, results_fields_map[f], type(value)))
        #self.assertTrue(isinstance(value, results_fields_map[f]))
  def test_value(self):
    for r in self.results:
      self.assertEqual(r["pg"], 1)
      self.assertEqual(r["ext"], u"mp4")
if __name__ == '__main__':
  # unittest.main() # 用这个是最简单的，下面的用法可以同时测试多个类
  # unittest.TextTestRunner(verbosity=2).run(suite1) # 这个等价于上述但可设置verbosity=2，省去了运行时加-v
  suite1 = unittest.TestLoader().loadTestsFromTestCase(TestSequenceFunctions)
  suite2 = unittest.TestLoader().loadTestsFromTestCase(TestDictValueFormatFunctions)
  suite = unittest.TestSuite([suite1, suite2])
  unittest.TextTestRunner(verbosity=2).run(suite)

执行上述代码后的运行结果如下：

test_choice (__main__.TestSequenceFunctions) ... ok test_sample (__main__.TestSequenceFunctions) ... ok test_shuffle (__main__.TestSequenceFunctions) ... ok test_format (__main__.TestDictValueFormatFunctions) ... ok test_value (__main__.TestDictValueFormatFunctions) ... ok

---------------------------------------------------------------------- Ran 5 tests in 0.013s

OK

如需了解更多Python相关知识，建议查阅本站专题：《Python入门与进阶经典教程》、《Python字符串操作技巧汇总》、《Python列表(list)操作技巧总结》、《Python编码操作技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》及《Python文件与目录操作技巧汇总》。

希望本文的内容能对您的Python程序设计学习有所帮助。