您的位置:

10. Python assert语句

Python assert语句

更新:

在 Python 中,如果给定条件评估为真,则使用assert语句继续执行。 如果断言条件评估为假,那么它会引发带有指定错误消息的AssertionError异常。

句法

assert condition [, Error Message] 

下面的示例演示了一个简单的 assert 语句。

Example: assert

x = 10
assert x > 0
print('x is a positive number.') 

Output

x is a positive number. 

在上面的例子中,断言条件x > 0评估为真,因此它将继续执行下一条语句,没有任何错误。

assert 语句可以选择性地包含一个错误消息字符串,该字符串与AssertionError一起显示。 考虑以下带有错误消息的assert语句。

Example: Assert Statement with Error Message

x = 0
assert x > 0, 'Only positive numbers are allowed'
print('x is a positive number.') 

Output

Traceback (most recent call last):
    assert x > 0, 'Only positive numbers are allowed'
AssertionError: Only positive numbers are allowed 

以上,x=0,所以断言条件x > 0变为假,所以它会用指定的消息“只允许正数”来引发AssertionError。 不执行print('x is a positive number.')语句。

下面的示例在函数中使用了 assert 语句。

Example: assert

def square(x):
    assert x>=0, 'Only positive numbers are allowed'
    return x*x

n = square(2) # returns 4
n = square(-2) # raise an AssertionError 

Output

Traceback (most recent call last):
    assert x > 0, 'Only positive numbers are allowed'
AssertionError: Only positive numbers are allowed 

上图中,square(2)将返回 4,而square(-2)将加注一个AssertionError,因为我们通过了-2。

AssertionError也是一个内置的异常,可以使用 try 来处理-除了如下所示的构造:

Example: AssertionError

def square(x):
    assert x>=0, 'Only positive numbers are allowed'
    return x*x

try:
    square(-2)
except AssertionError as msg:
    print(msg) 

Output

Only positive numbers are allowed 

上图,调用square(-2)会引发AssertionError,由除块处理。 assert 语句中的错误消息将作为参数传递给异常参数msg,使用as关键字。

因此,assert 语句通常应该用于防止可能的错误,并指定适当的错误消息。****