单元测试 · 别等客户帮你测 | 小朵科技工作室 博客

很多人写完功能就跑，出了问题再改，改完再跑，反复循环。

测试写过一遍的项目，改起来心里有底。

最简单的入门：pytest。

1. 1
2. 2
3. 3
4. 4

def test_clean_text():
    result = clean_text("  Hello  World! ")
    assert result == "hello world"

跑测试：pytest

几个原则：

• 核心函数都写测试（数据预处理、特征计算）
• 改完代码先跑测试再 commit
• 测试覆盖率不必 100%，关键路径覆盖到就行

进阶可以用 mock：

from unittest.mock import patch
@patch('requests.get')
def test_fetch(mock_get):
    mock_get.return_value.json.return_value = {"status": "ok"}
    result = fetch_data()
    assert result == {"status": "ok"}

客户最讨厌的就是「我电脑上能跑啊」——测试覆盖率上去，这种话说不出口。

---

本文由 BBZ · 小朵科技工作室 出品。