Python斷言
Python assert 語句,又稱斷言語句,可以看做是功能縮小版的 if 語句,它用於判斷某個表達式的值,如果值為真,則程序可以繼續往下執行;反之,Python 解釋器會報 AssertionError 錯誤。
assert 語句的語法結構為:
assert 表達式
assert 語句的執行流程可以用 if 判斷語句表示(等效),如下所示:
if 表達式==True: 程序繼續執行 else: 程序報 AssertionError 錯誤
明明 assert 會令程序崩潰,為什么還要使用它呢?這是因為,與其讓程序在晚些時候崩潰,不如在錯誤條件出現時,就直接讓程序崩潰,這有利於我們對程序排錯,提高程序的健壯性。
因此,assert 語句通常用於檢查用戶的輸入是否符合規定,還經常用作程序初期測試和調試過程中的輔助工具。
下面的程序演示了 assert 語句的用法:
mathmark = int(input()) #斷言數學考試分數是否位於正常范圍內 assert 0 <= mathmark <= 100 #只有當 mathmark 位於 [0,100]范圍內,程序才會繼續執行 print("數學考試分數為:",mathmark)
運行該程序,測試數據如下:
90
數學考試分數為: 90
再次執行該程序,測試數據為:
159
Traceback (most recent call last): File "C:\Users\mengma\Desktop\file.py", line 3, in <module> assert 0 <= mathmark <= 100 AssertionError
可以看到,當 assert 語句后的表達式值為真時,程序繼續執行;反之,程序停止執行,並報 AssertionError 錯誤。
常用斷言函數
-
這里介紹幾個常用斷言的使用方法,可以一定程度上幫助大家對預期結果進行判斷。-
-
assertEqual
-
assertNotEqual
-
assertTrue
-
assertFalse
-
assertIsNone
-
assertIsNotNone
-
assertEqual 和 assertNotEqual
-
assertEqual
:如兩個值相等,則pass -
assertNotEqual
:如兩個值不相等,則pass -
使用方法:
assertEqual(first,second,msg)
其中first
與second
進行比較,如果相等則通過;msg
為失敗時打印的信息,選填;斷言assertNotEqual反着用就可以了。 -
assertTrue和assertFalse
-
assertTrue
:判斷bool值為True,則pass -
assertFalse
:判斷bool值為False,則Pass -
使用方法:
assertTrue(expr,msg)
其中express
輸入相應表達式,如果表達式為真,則pass;msg
選填;斷言assertFalse
如果表達式為假,則pass -
assertIsNone和assertIsNotNone
-
assertIsNone:不存在,則pass
-
assertIsNotNone:存在,則pass
-
使用方法:
assertIsNone(obj,msg)
檢查某個元素是否存在 -
總結:
-
常用
assertEqual(a, b) a == b assertNotEqual(a, b) a != b assertTrue(x) bool(x) is True assertFalse(x) bool(x) is False assertIs(a, b) a is b 2.7 assertIsNot(a, b) a is not b 2.7 assertIsNone(x) x is None 2.7 assertIsNotNone(x) x is not None 2.7 assertIn(a, b) a in b 2.7 assertNotIn(a, b) a not in b 2.7 assertIsInstance(a, b) isinstance(a, b) 2.7 assertNotIsInstance(a, b) not isinstance(a, b) 2.7
- 其它
assertAlmostEqual(a, b) round(a-b, 7) == 0 assertNotAlmostEqual(a, b) round(a-b, 7) != 0 assertGreater(a, b) a > b 2.7 assertGreaterEqual(a, b) a >= b 2.7 assertLess(a, b) a < b 2.7 assertLessEqual(a, b) a <= b 2.7 assertRegexpMatches(s, re) regex.search(s) 2.7 assertNotRegexpMatches(s, re) not regex.search(s) 2.7 assertItemsEqual(a, b) sorted(a) == sorted(b) and works with unhashable objs 2.7 assertDictContainsSubset(a, b) all the key/value pairs in a exist in b 2.7 assertMultiLineEqual(a, b) strings 2.7 assertSequenceEqual(a, b) sequences 2.7 assertListEqual(a, b) lists 2.7 assertTupleEqual(a, b) tuples 2.7 assertSetEqual(a, b) sets or frozensets 2.7 assertDictEqual(a, b) dicts 2.7 assertMultiLineEqual(a, b) strings 2.7 assertSequenceEqual(a, b) sequences 2.7 assertListEqual(a, b) lists 2.7 assertTupleEqual(a, b) tuples 2.7 assertSetEqual(a, b) sets or frozensets 2.7assertDictEqual(a, b) dicts 2.7
-
使用時機:
那么我們什么時候應該使用斷言呢?如果沒有特別的目的,斷言應該用於如下情況:
- 防御性的編程
- 運行時對程序邏輯的檢測
- 合約性檢查(比如前置條件,后置條件)
- 程序中的常量
- 檢查文檔
- 例:防御性編程中添加斷言
在代碼開始添加注釋是個好的開端,但是人們都不太喜歡讀和更新這些注釋,這些注釋會很快變得過時。但對於斷言,我們可以同時對這塊代碼編寫文檔,如果這些斷言被違反了,會直接引起一個簡單而又直接的失敗。
assert target in(x, y, z) if target ==x: run_x_code() elif target ==y: run_y_code() else: assert target == z run_z_code()