博客导语

手动运行测试效率低、易遗漏、无法回归。单元测试是工程开发标准流程，用于检测函数、接口、模块是否符合预期，修改代码后一键回归测试，避免改崩原有功能。本文使用Python内置unittest模块，零依赖快速上手。

一、单元测试核心规则

• 测试类必须继承 unittest.TestCase

• 测试方法必须以 test_ 开头

• 支持断言判断结果是否正确

二、完整实战案例

1. 待测试函数 calc.py

def add(a,b):
    return a + b

def sub(a,b):
    return a - b

2. 编写测试用例 test_calc.py

import unittest
from calc import add, sub

class TestCalc(unittest.TestCase):

    def test_add(self):
        self.assertEqual(add(1,2), 3)
        self.assertEqual(add(-1,1), 0)

    def test_sub(self):
        self.assertEqual(sub(5,2), 3)

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

三、常用断言方法

self.assertEqual(a,b)   # 相等
self.assertNotEqual(a,b) # 不等
self.assertTrue(cond)   # 为真
self.assertIn(x,list)   # 包含

四、前置后置方法

每个用例执行前后自动运行，用于初始化、销毁资源。

def setUp(self):
    print("用例前置：初始化数据")

def tearDown(self):
    print("用例后置：清理数据")

五、适用场景

工具函数、核心算法、接口逻辑、公共方法必须编写单元测试，保证迭代稳定。