Python单元测试用unittest框架，核心是继承unittest.TestCase类，测试方法名必须以test_开头，通过unittest.main()运行；常用断言包括assertEquals、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;">def test_add_negative_numbers(self):
self.assertEqual(add(-1, -1), -2)
def test_add_zero(self):
self.assertEqual(add(0, 5), 5)
立即学习“Python免费学习笔记（深入）”；
注意：测试方法名必须以test_开头，否则不会被自动识别为测试用例。
常用断言方法
unittest提供多种assert*方法验证预期结果：
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
下载
让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限
-
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)
设置与清理（setUp / tearDown）
如果多个测试需要共用资源（如临时文件、数据库连接、实例对象），可用setUp和tearDown方法：
-
setUp()：每个测试方法运行前自动执行（适合初始化） -
tearDown()：每个测试方法运行后自动执行（适合清理）
例如：
class TestCalculator(unittest.TestCase):
def setUp(self):
self.calc = Calculator() # 每个测试前创建新实例
<pre class="brush:php;toolbar:false;">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 可留空，或关闭文件、清空缓存等
立即学习“Python免费学习笔记（深入）”；
运行测试的几种方式
有三种常用启动方式：
- 在脚本末尾加
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
