1. 程式人生 > >Python 3 小知識 assert用法

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使用起來還是很方便的,可以避免不必要的未知錯誤。