20 分钟

工程化

unittest 单元测试

使用 unittest 编写和运行测试

理解测试驱动开发
编写 unittest 测试用例
使用 setUp/tearDown

第一个测试

示例代码（可运行）

Python 0%

1import unittest

3def add(a, b):

4 return a + b

6def divide(a, b):

7 if b == 0:

8 raise ValueError("不能除以0")

9 return a / b

11class TestMath(unittest.TestCase):

12 def test_add(self):

13 self.assertEqual(add(2, 3), 5)

14 self.assertEqual(add(-1, 1), 0)

16 def test_divide(self):

17 self.assertEqual(divide(10, 2), 5)

18 with self.assertRaises(ValueError):

19 divide(1, 0)

21# 运行测试

22suite = unittest.TestLoader().loadTestsFromTestCase(TestMath)

23runner = unittest.TextTestRunner(verbosity=2)

24runner.run(suite)

常用断言

示例代码（可运行）

Python 0%

1import unittest

3class TestAssertions(unittest.TestCase):

4 def test_examples(self):

5 self.assertEqual(1 + 1, 2) # 相等

6 self.assertTrue(3 > 2) # 为True

7 self.assertFalse(1 > 2) # 为False

8 self.assertIn(1, [1, 2, 3]) # 包含

9 self.assertIsNone(None) # 是None

10 self.assertAlmostEqual(0.1+0.2, 0.3, places=1)

12unittest.main(argv=[""], exit=False, verbosity=0)

setUp 和 tearDown

示例代码（可运行）

Python 0%

1import unittest

3class Stack:

4 def __init__(self):

5 self.items = []

6 def push(self, item): self.items.append(item)

7 def pop(self): return self.items.pop()

8 def size(self): return len(self.items)

10class TestStack(unittest.TestCase):

11 def setUp(self):

12 """每个测试前执行"""

13 self.stack = Stack()

14 self.stack.push(1)

15 self.stack.push(2)

17 def test_pop(self):

18 self.assertEqual(self.stack.pop(), 2)

19 self.assertEqual(self.stack.size(), 1)

21 def test_push(self):

22 self.stack.push(3)

23 self.assertEqual(self.stack.size(), 3)

25unittest.main(argv=[""], exit=False, verbosity=2)

💡测试原则

一个测试只测一个行为。测试名称要描述场景：test_divide_by_zero_raises_error。先写测试再写代码（TDD）能写出更好的设计。

预测输出

print(3 > 2 and 1 > 2)

找 Bug用assert而非self.assertXxx，测试失败时unittest无法正确捕获和报告。

class T(unittest.TestCase): def test_add(self): assert add(2,2) == 5

填空题填写空白处的代码

self.Equal(a, b) # 断言相等

选择题

每个测试前执行的方法？

资深工程师加餐

底层原理 · 大厂视角 · 工程经验，点卡片展开

底层是大量、快速、稳定的单元测试（优先覆盖纯函数与边界/异常路径），中间是少量集成测试（验证模块协作），顶端是少量端到端测试（慢且脆）。测试最大的价值是当「安全网」：有它你才敢放心重构。没测试兜底的重构，本质是在赌博。写用例先想正常、边界、异常三类。

挑战任务

unittest挑战

简单+50 XP

给字符串工具函数写完整测试。

unittest挑战

Python 0%

2 个测试用例

课后作业

计算器测试

中等+20 XP

给加减乘除计算器写完整测试。

计算器测试

Python 0%

1 个测试用例