1. 本期主題：Python單元測試框架unittest詳解

unittest是Python內置的單元測試框架，遵循Java JUnit的"測試驅動開發"（TDD）理念，通過繼承TestCase類實現測試用例的模塊化組織。本文聚焦于獨立測試腳本的編寫，暫不涉及數據庫集成或參數化測試等高級場景（參數化測試建議使用parameterized包或pytest實現）。

涵蓋內容

• 基礎機制：測試類與測試方法的定義規范
• 執行流程：從腳本運行到結果輸出的完整鏈路
• 結果解讀：通過符號標記快速定位測試問題

不涵蓋內容

• DOM操作（前端測試建議使用Selenium）
• 數據庫集成測試（需結合unittest.mock或pytest-fixture）
• 參數化測試（后續文章將單獨講解）

---

2. unittest核心機制與執行流程

2.1 測試腳本結構解析

import unittestclass CalculatorTestCase(unittest.TestCase):"""加法器測試類"""def test_add_positive(self):"""正數加法測試"""self.assertEqual(10 + 5, 15)def test_add_negative(self):"""負數加法測試"""self.assertEqual(-3 + 7, 4)if __name__ == "__main__":unittest.main(verbosity=2)  # 增加輸出詳細度

關鍵點說明：

1. 命名規范：測試類以Test結尾，測試方法以test_開頭
2. 文檔字符串：類和方法建議添加說明性注釋
3. 執行參數：verbosity=2可顯示測試名稱而非僅點號

2.2 執行結果解讀

運行上述腳本將輸出：

test_add_negative (__main__.CalculatorTestCase) ... ok
test_add_positive (__main__.CalculatorTestCase) ... ok----------------------------------------------------------------------
Ran 2 tests in 0.001sOK

• OK標記：所有測試通過
• FAIL標記：斷言失敗時會顯示具體值（如self.assertEqual(10, 20)會輸出10 != 20）

---

3. 常用斷言方法詳解

3.1 基礎斷言

class StringTestCase(unittest.TestCase):def test_string_operations(self):# 字符串相等驗證self.assertEqual("tianxin".upper(), "TIANXIN")# 字符串包含驗證self.assertIn("xin", "tianxin")# 布爾值驗證self.assertTrue("tianxin".startswith("tian"))self.assertFalse("tianxin".endswith("xin"))  # 實際會失敗，此處僅為示例

失敗案例演示：

def test_false_positive(self):self.assertEqual(10, 20)  # 輸出: AssertionError: 10 != 20

3.2 異常驗證

class DivisionTestCase(unittest.TestCase):def test_zero_division(self):with self.assertRaises(ZeroDivisionError):5 / 0def test_invalid_type(self):with self.assertRaises(TypeError):"10" + 5  # 字符串與整數拼接會觸發TypeError

應用場景：驗證邊界條件（如除零、類型錯誤）

---

4. 生命周期鉤子：setUp與tearDown

4.1 層級說明與執行順序

鉤子方法	執行時機	適用場景
setUpModule	模塊首次導入時	初始化全局資源（如數據庫連接池）
setUpClass	測試類首次實例化時	類級別資源（如測試文件路徑）
setUp	每個測試方法執行前	測試方法獨占資源（如臨時文件）
tearDown	每個測試方法執行后	清理測試殘留（如刪除臨時文件）
tearDownClass	測試類所有方法執行完畢后	釋放類級別資源
tearDownModule	模塊所有測試執行完畢后	關閉全局資源

4.2 完整示例

import os
import unittestclass FileOperationTestCase(unittest.TestCase):temp_file = "temp_test.txt"@classmethoddef setUpClass(cls):print("? 準備測試文件...")with open(cls.temp_file, "w") as f:f.write("initial content")def setUp(self):print("  → 每個測試前重置文件內容")with open(self.temp_file, "w") as f:f.write("")  # 清空文件def test_write_content(self):with open(self.temp_file, "a") as f:f.write("line1\n")self.assertTrue(os.path.exists(self.temp_file))def test_append_content(self):with open(self.temp_file, "a") as f:f.write("line2\n")with open(self.temp_file) as f:self.assertEqual(f.read(), "line2\n")  # 驗證清空操作是否生效@classmethoddef tearDownClass(cls):print("? 刪除測試文件")os.remove(cls.temp_file) if os.path.exists(cls.temp_file) else Noneif __name__ == "__main__":unittest.main()

輸出順序：

? 準備測試文件...→ 每個測試前重置文件內容
test_write_content ... ok→ 每個測試前重置文件內容
test_append_content ... ok
? 刪除測試文件

---

5. 測試跳過機制

5.1 裝飾器應用

import unittest
import platformclass PlatformTestCase(unittest.TestCase):@unittest.skipIf(platform.system() == "Windows", "Windows系統暫不支持")def test_linux_feature(self):print("僅在Linux下運行的測試")@unittest.skipUnless(hasattr(os, "symlink"), "系統不支持符號鏈接")def test_symlink(self):print("符號鏈接測試")@unittest.skip("功能重構中，暫不測試")def test_deprecated_feature(self):print("已棄用功能測試")

執行結果：

s (skipped) ... skipped 'Windows系統暫不支持'
s (skipped) ... skipped '系統不支持符號鏈接'
s (skipped) ... skipped '功能重構中，暫不測試'

---

6. 總結與建議

1. 測試設計原則：

• 每個測試方法只驗證一個功能點
• 使用有意義的測試名稱（如test_add_positive而非test1）
• 優先使用setUp/tearDown而非重復代碼

2. 擴展方向：

• 數據庫測試：結合unittest.mock模擬數據庫連接
• 參數化測試：使用pytest.mark.parametrize或parameterized包
• 集成測試：通過subTest實現測試數據驅動

3. 工具鏈建議：

• 簡單項目：直接使用unittest
• 復雜項目：遷移至pytest（支持更簡潔的語法和插件生態）
• 持續集成：結合tox實現多Python版本測試

---

通過以上結構化講解，讀者可系統掌握unittest的核心用法，并逐步向更復雜的測試場景擴展。