Python单元测试基础示例：基于《Python编程从入门到实践》

Owen

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

原创

本文以《Python编程从入门到实践》为例，详解Python中如何使用unittest模块进行单元测试。通过定义被测函数、编写测试类及断言方法，展示如何验证函数行为。文中特别指出Python 2.7环境下主函数调用的兼容性问题及解决方案，帮助开发者掌握基础测试流程。

本文介绍一种Python单元测试的实用方法，供读者参考学习。

该示例源自Eric所著《Python编程从入门到实践》一书中的案例。

《Python编程从入门到实践》的随书源代码可在以下链接下载：https://www.jb51.net/books/582019.html。

首先，我们需要定义一个待测试的函数。

文件名为namefunction.py。

#-*- coding:cp936 -*-
def get_formmed_name(first, last):
 """该函数根据姓和名生成一个完整的姓名"""
 full_name = first + ' ' + last
 return full_name.title()

接下来，编写一个用于测试该函数的模块代码。

文件名为names.py。

#-*- coding:cp936 -*-
from name_function import get_formmed_name
"""该文件用来测试姓名生成函数"""
print "Enter 'q' to quit any time."
while True:
 first = raw_input("\nEnter first name:")
 if first == 'q':
  break
 last = raw_input("Enter last name:")
 if last == 'q':
  break
 formatted_name = get_formmed_name(first,last)
 print "Formmated name:",formatted_name

经过测试验证，name_function模块中的函数能够正确实现其预定功能。

最后，展示单元测试代码及测试用例的具体编写方式。

测试文件名为test_name_function.py。

# coding:utf-8
import unittest
from name_function import get_formmed_name
class NamesTestCase(unittest.TestCase):
 """测试name_function.py"""
 def test_first_last_name(self):
  """能够正确处理像Janis Joplin这样的姓名吗？"""
  formatted_name = get_formmed_name('janis','joplin')
  self.assertEqual(formatted_name,'Janis Joplin')

需要注意的是，在Python 2.7版本中直接调用书中的主函数unittest.main()会导致程序运行失败，若改用以下代码调用方式则可正常运行。

if __name__ == '__main__':
 unittest

通过对上述单元测试模块的分析，可以总结出以下关键点：

1. 导入单元测试所需的unittest类。

2. 导入需要测试的目标函数，本例中为name_function模块下的get_formatted_name()函数。

3. 创建一个继承自unittest.TestCase的测试类。

4. 在测试类中定义多个方法，从不同方面测试函数的行为。每个测试用例应仅聚焦于一个测试点，确保测试目的与内容明确。主要通过调用assertEqual、assertRaises等断言方法，判断程序实际执行结果是否与预期值一致。

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

希望本文的内容能对您的Python程序设计工作提供帮助。