使用unittest编写Python单元测试需创建继承自TestCase的类，测试方法以test_开头，通过断言方法验证逻辑。例如为calculator模块编写TestCalculator类，用assertEqual、assertRaises等方法测试加减乘除函数，确保正常与异常情况均被覆盖。命令行运行python test_calculator.py即可执行测试，-v参数显示详细结果。
写Python单元测试主要用内置的unittest框架，也可以选择第三方库如pytest。这里以unittest为例，介绍编写方法和实际例子，帮助你快速上手。
1. 理解单元测试的基本结构
unitte
st是Python标准库中的测试框架，基于类和方法组织测试。每个测试用例继承自unittest.TestCase，测试方法名必须以test_开头。
基本结构如下：
import unittestclass TestMathFunctions(unittest.TestCase): def test_addition(self): self.assertEqual(2 + 2, 4)
if name == 'main': unittest.main()
2. 编写被测函数和对应测试
假设我们有一个简单的计算器模块calculator.py：
# calculator.py
def add(a, b):
return a + b
def subtract(a, b):
return a - b
def multiply(a, b):
return a * b
def divide(a, b):
if b == 0:
raise ValueError("Cannot divide by zero.")
return a / b
接下来为它写测试文件test_calculator.py：
# test_calculator.py import unittest from calculator import add, subtract, multiply, divideclass TestCalculator(unittest.TestCase):
def test_add(self): self.assertEqual(add(2, 3), 5) self.assertEqual(add(-1, 1), 0) self.assertEqual(add(0, 0), 0) def test_subtract(self): self.assertEqual(subtract(5, 3), 2) self.assertEqual(subtract(0, 5), -5) def test_multiply(self): self.assertEqual(multiply(3, 4), 12) self.assertEqual(multiply(-2, 3), -6) def test_divide(self): self.assertEqual(divide(10, 2), 5) self.assertAlmostEqual(divide(7, 3), 2.333333, places=6) def test_divide_by_zero(self): with self.assertRaises(ValueError): divide(10, 0)if name == 'main': unittest.main()
3. 常用断言方法说明
TestCase类提供多种断言方法验证结果：
-
assertEqual(a, b)：检查a是否等于b -
assertNotEqual(a, b)：检查a不等于b -
assertTrue(x)：检查x为真 -
assertFalse(x)：检查x为假 -
assertIn(a, b)：检查a在b中 -
assertIsInstance(obj, cls)：检查对象是否为某类实例 -
assertRaises(exception)：检查是否抛出指定异常 -
assertAlmostEqual(a, b)：检查浮点数近似相等
4. 运行测试
在命令行执行测试文件：
python test_calculator.py
输出会显示运行了多少个测试，是否全部通过。如果某个测试失败，会指出具体哪一行出错。
加上参数-v可查看详细结果：
python test_calculator.py -v
基本上就这些。掌握结构、写好断言、覆盖正常和异常情况，就能写出有效的单元测试。
