【Python】 斷言 assert

| | |

在 Python 中 assert 陳述句長這樣：

1
assert <test>, <message>

我會跟你說我記錄這個的原因是因為我老是拼錯嗎？

斷言（Assertion）

所謂的斷言，指的就是斷定程式進行到特定時間、地點時其變數的狀態，像是變數大小、變數狀態、是否為空值…等。我個人還滿喜歡用斷言，與其讓程式在執行時 crash，再一行一行去追 log，倒不我先檢查變數，錯誤時直接中斷運作，…至少 log 也好看一點 XD


下面這段程式碼用來檢查選擇的預訓練模型是否支援。因為 selected 不在 SUPPORT_MODELS 中，所以會中斷運作，並跳出我寫好的錯誤訊息。

1
2
3
4
5
6
7
8
SUPPORT_MODELS = ['ResNet50', 'Xception', 'VGG16', "DenseNet121",
                  "InceptionV3", "InceptionResNetV2", "MobileNetV2"]

selected = 'VGG19'

assert selected in SUPPORT_MODELS, \
 "{} isn't supported Please select a model of the following: {}".format(self.apps, ",".join(SUPPORT_MODELS))

參考資料

  1. 林信良。使用 assert 。檢自 語言技術：Python Gossip (2020-02-27)。