Python 3 小知識 assert用法

assert語句，如果沒記錯，這個東西在C或者C++里面也有的。屬於短小的斷言。下面的是來自python help document的說明：

Assert statements are a convenient way to insert debugging assertions into a program:

assert語句是一種插入調試斷點到程序的一種便捷的方式。

 

assert語句的使用格式

assert expression

這個語句是等價於下面的個句式:

if __debug__:
    if not expression: raise AssertionError

 

assert也可以用於多個表達式的斷言

assert expression1, expression2

 

我自己寫的一個關於質數判定的assert使用示例

def isPrime(n):
    """This function return a number is a prime or not"""
    assert n >= 2
    from math import sqrt
    for i in range(2, int(sqrt(n))+1):
        if n % i == 0:
            return False
    return True

 

assert使用起來還是很方便的，可以避免不必要的未知錯誤。