Python 3.6 unittest 单元测试完整示例与最佳实践

Hazel

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

原创

本文通过Python 3.6环境下的Student类示例，详细演示如何使用unittest模块编写单元测试。内容涵盖测试类继承、断言方法assertEqual与assertRaises的使用，以及命令行运行测试的多种技巧，旨在帮助开发者掌握Python单元测试的核心流程与最佳实践。

Python 3.6 单元测试示例概述

本文详细介绍基于Python 3.6实现的单元测试实例，供读者参考学习。

通过Python 3.6编写一个单元测试示例，以Student类的简单测试为例进行说明。

Student类定义

第一步，定义Student类：


#!/usr/bin/env python3
# -*- coding: utf-8 -*-
class Student(object):
  def __init__(self,name,score):
    self.name = name
    self.score = score
  def get_grade(self):
    if self.score >= 80 and self.score <= 100:
      return 'A'
    elif self.score >= 60 and self.score <= 79:
      return 'B'
    elif self.score >= 0 and self.score <= 59:
      return 'C'
    else:
      raise ValueError('value is not between 0 and 100')

测试类编写

第二步，创建测试类TestStudent，该类需继承自：


#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import unittest
from student import Student
class TestStudent(unittest.TestCase):
  def test_80_to_100(self):
    s1 = Student('Bart',80)
    s2 = Student('Lisa',100)
    self.assertEqual(s1.get_grade(),'A')
    self.assertEqual(s2.get_grade(),'A')
  def test_60_to_80(self):
    s1 = Student('Bart',60)
    s2 = Student('Lisa',79)
    self.assertEqual(s1.get_grade(),'B')
    self.assertEqual(s2.get_grade(),'B')
  def test_0_to_60(self):
    s1 = Student('Bart',0)
    s2 = Student('Lisa',59)
    self.assertEqual(s1.get_grade(),'C')
    self.assertEqual(s2.get_grade(),'C')
  def test_invalid(self):
    s1 = Student('Bart',-1)
    s2 = Student('Lisa',101)
    with self.assertRaises(ValueError):
      s1.get_grade()
    with self.assertRaises(ValueError):
      s2.get_grade()
#运行单元测试
if __name__ == '__main__':
  unittest.main()

测试运行结果

第三步，执行测试后的运行结果如下：

命令行运行测试方法

第四步，另一种运行单元测试的方式是通过命令行使用-m unittest参数直接执行，例如命令：python -m unittest student_test

unittest模块使用总结

最后，对使用unittest模块进行总结：

编写单元测试时，必须创建一个测试类，并使其继承自unittest.TestCase
针对每个测试类，都需要定义一个test_xxx()方法
最常用的断言方法是assertEqual()
另一种重要的断言方式是预期抛出指定类型的Error，例如：with self.assertRaises(KeyError):
另一种执行方式是通过命令行使用-m unittest参数直接运行单元测试，例如：python -m unittest student_test
最简单的运行方式是在xx.py文件末尾添加以下两行代码：


if __name__ == '__main__':
  unittest.main()

更多Python相关内容感兴趣的读者可查看本站专题：《Python入门与进阶经典教程》、《Python字符串操作技巧汇总》、《Python列表(list)操作技巧总结》、《Python编码操作技巧总结》、《Python数据结构与算法教程》、《Python函数使用技巧总结》及《Python文件与目录操作技巧汇总》

希望本文所述对大家Python程序设计有所帮助。