Python单元测试教程：unittest实战详解

本文深入浅出地讲解了Python中使用unittest框架编写单元测试的完整实践流程，涵盖测试类定义（继承unittest.TestCase）、测试方法命名规范（必须以test_开头）、核心断言方法（如assertEqual、assertTrue、assertRaises等）的正确用法，以及如何通过setUp/tearDown高效管理测试前准备与后清理，并辅以加法、除零异常等实用示例，最后介绍了多种灵活运行测试的方式（脚本内执行、命令行调用、单测方法指定），帮助开发者快速上手、写出可靠可维护的单元测试代码。

Python单元测试用unittest写，核心是继承unittest.TestCase，把测试逻辑写在以test_开头的方法里，然后调用unittest.main()运行。

写一个基础测试类

每个测试文件通常对应一个被测模块。先导入要测的函数或类，再定义测试类，继承unittest.TestCase：

示例：测试一个加法函数

import unittest
<p>def add(a, b):
return a + b</p><p>class TestAdd(unittest.TestCase):
def test_add_positive_numbers(self):
self.assertEqual(add(2, 3), 5)</p><pre class="brush:php;toolbar:false"><code>def test_add_negative_numbers(self):
    self.assertEqual(add(-1, -1), -2)

def test_add_zero(self):
    self.assertEqual(add(0, 5), 5)</code>

注意：测试方法名必须以test_开头，否则不会被自动识别为测试用例。

常用断言方法

unittest提供多种assert*方法验证预期结果：

• self.assertEqual(a, b)：检查a是否等于b（推荐用于大多数值比较）
• self.assertTrue(x) / self.assertFalse(x)：检查布尔表达式
• self.assertIs(a, b)：检查是否为同一对象（如None、单例）
• self.assertIn(item, container)：检查元素是否在列表/字典/字符串中
• self.assertRaises(TypeError, func, arg)：检查调用是否抛出指定异常

例如测试异常：

def divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero")
    return a / b
<p>def test_divide_by_zero(self):
with self.assertRaises(ValueError):
divide(5, 0)
</p>

设置与清理（setUp / tearDown）

如果多个测试需要共用资源（如临时文件、数据库连接、实例对象），可用setUp和tearDown方法：

• setUp()：每个测试方法运行前自动执行（适合初始化）
• tearDown()：每个测试方法运行后自动执行（适合清理）

例如：

class TestCalculator(unittest.TestCase):
    def setUp(self):
        self.calc = Calculator()  # 每个测试前创建新实例
<pre class="brush:php;toolbar:false"><code>def test_add(self):
    self.assertEqual(self.calc.add(2, 3), 5)

def test_subtract(self):
    self.assertEqual(self.calc.subtract(5, 2), 3)
# tearDown 可留空，或关闭文件、清空缓存等</code>

运行测试的几种方式

有三种常用启动方式：

• 在脚本末尾加if __name__ == '__main__': unittest.main()，直接运行文件
• 命令行进入项目根目录，执行python -m unittest test_module（不带.py）
• 用-v参数获得详细输出：python -m unittest -v test_calc

还可以运行单个测试方法：python -m unittest test_calc.TestCalculator.test_add

以上就是《Python单元测试教程：unittest实战详解》的详细内容，更多关于的资料请关注golang学习网公众号！