快速上手环境搭建
安装与版本要求
Python 3.x 是进行单元测试的理想版本，unittest 作为 Python 的标准测试框架，通常无需额外安装。确保你的开发环境中已经安装了合适的 Python 版本，并且路径已正确配置，以便在命令行直接调用 python 命令。一致的版本 能减少在不同环境下的测试差异。
在开展单元测试之前，推荐使用虚拟环境来隔离测试依赖，避免全局包冲突。通过虚拟环境，可以确保测试用例在干净的环境中执行，获得可重复的测试结果。以下示例展示创建并激活虚拟环境的常用命令：
python -m venv venv
# macOS/Linux
source venv/bin/activate
# Windows
venv\\Scripts\\activate
创建测试目录结构
一个清晰的目录结构能显著提升单元测试的可维护性。将实现代码放在 src/，测试代码放在 tests/，并在 tests/ 下按模块或功能划分测试文件，可以帮助团队快速定位问题。测试分层 有助于后续的持续集成和测试覆盖率分析。
示例结构有助于在实际项目中开展单元测试工作流：
# 项目结构示例
src/math_utils.py
tests/test_math_utils.py
unittest 基础用法
编写第一个测试用例
使用 unittest.TestCase 作为测试基类，是学习单元测试的起点。测试方法需要以 test_ 开头，框架会自动发现这些方法并执行。通过断言来校验期望结果，是单元测试的核心。
下面给出一个最简单的测试用例，验证 1+1 的结果是否等于 2。通过运行该脚本，可以直观地观察测试执行过程。
import unittestclass TestMath(unittest.TestCase):def test_addition(self):self.assertEqual(1 + 1, 2)if __name__ == '__main__':unittest.main()
断言方法
unittest 提供了丰富的断言集合，用于覆盖不同类型的断言场景。常用断言包含：assertEqual、assertTrue、assertFalse、assertIn、assertRaises 等。选用合适的断言方法，是提高测试鲁棒性的关键。
通过组合多种断言，可以在一个测试用例中覆盖多种边界情况与异常场景。以下示例演示了对除法的异常处理与正常结果的断言。
# 断言示例
import unittestdef divide(a, b):return a / bclass TestDivide(unittest.TestCase):def test_divide_by_zero(self):with self.assertRaises(ZeroDivisionError):divide(1, 0)def test_basic_division(self):self.assertEqual(divide(6, 3), 2)if __name__ == '__main__':unittest.main()
测试发现与执行
测试用例组织与发现
为了让测试能够自动化运行，测试发现（Discovery） 机制会扫描符合命名规范的测试模块与测试用例。通常，测试文件名以 test_ 开头，并放置在测试目录中，框架会自动加载这些用例。
组织好测试用例后，可以利用 unittest 的发现机制实现一次性加载与执行，减少手动指定测试集合的工作量。下面代码展示如何在包含测试的目录中进行发现与执行。
# tests/test_math_utils.py
import unittest
from src.math_utils import addclass TestMathUtils(unittest.TestCase):def test_add(self):self.assertEqual(add(2, 3), 5)if __name__ == '__main__':unittest.main()
命令行运行测试
最常用的执行方式是通过命令行直接运行 python -m unittest，它会根据当前工作目录自动发现测试用例并执行。若希望在执行时获得更详细的输出，可以使用 -v 选项。
在实际项目中，命令行执行通常结合虚拟环境进行，确保测试在与生产环境一致的条件下运行。
# 直接在项目根目录执行
$ python -m unittest# 带详细输出
$ python -m unittest -v
实战案例：对简单应用的单元测试
案例1：数学运算模块
在真实项目中，数学运算模块常包含加、减、乘、除等基本函数。通过 unittest，可以为这些函数建立覆盖性强的测试用例，确保各个运算在边界条件下的正确性。覆盖常见输入与极值情况，有助于提早发现实现中的潜在问题。
以下示例展示一个简单的算术模块及其测试用例的完整实现过程，帮助你快速理解如何把逻辑实现与单元测试绑定起来。
# src/arith.py
def add(a, b): return a + b
def sub(a, b): return a - b
def mul(a, b): return a * b
def div(a, b): return a / b
# tests/test_arith.py
import unittest
from src.arith import add, sub, mul, divclass TestArith(unittest.TestCase):def test_all(self):self.assertEqual(add(2,3), 5)self.assertEqual(sub(5,2), 3)self.assertEqual(mul(3,4), 12)self.assertEqual(div(10,2), 5)def test_div_error(self):with self.assertRaises(ZeroDivisionError):div(1, 0)if __name__ == '__main__':unittest.main()
案例2：字符串处理模块
除了数值运算，字符串处理逻辑也是单元测试的重要对象，例如拼接、分片、大小写变换等操作。通过合理的测试用例集，可以确保字符串处理在不同输入下的鲁棒性，提升代码质量。
下面给出一个示例，用于验证字符串拼接和大小写转换功能是否符合预期。
# src/str_utils.py
def join_strings(a, b, sep=' '):return f"{a}{sep}{b}"def to_upper(s):return s.upper()
# tests/test_str_utils.py
import unittest
from src.str_utils import join_strings, to_upperclass TestStrUtils(unittest.TestCase):def test_join(self):self.assertEqual(join_strings('Hello', 'World'), 'Hello World')def test_upper(self):self.assertEqual(to_upper('abc'), 'ABC')if __name__ == '__main__':unittest.main()
