Python By Natalie

PyTest 測試框架入門:讓你的腳本不再充滿 Bug

從 print 除錯進化到單元測試

很多工程師寫腳本習慣用 print() 除錯,但隨著邏輯變複雜,這種方式效率極低且難以維護。PyTest 是目前最流行的 Python 測試框架,語法簡單且功能強大。

核心概念

  • Assert:直接使用 Python 原生的 assert 語句 (如 assert add(1, 2) == 3),失敗時 PyTest 會顯示詳細的變數值差異。
  • Fixtures:用於設定測試環境 (Setup/Teardown),例如建立 DB 連線或準備測試資料。使用 @pytest.fixture 裝飾器定義。
  • Parametrize:使用 @pytest.mark.parametrize 可以用多組輸入數據測試同一個函數,輕鬆覆蓋邊界條件。

為你的核心邏輯 (如股價計算公式) 加上測試,是邁向資深工程師的必經之路。