Python 中的断言( assert )

Python 中的断言( assert )

Python 中的断言( assert )

由Python 文档https://docs.python.org/3/reference/simple_stmts.html#assert 可知，assert 语句的语法如下所示：

assert_stmt ::= “assert” expression1 [“,” expression2]

此乃编译原理的定义格式，通俗说法为：

assert expression1 [, expression2]

其中expression1是需要测试的条件， expression2是错误消息是可选的，测试的条件为假时显示之。在执行时，Python 解释器将每条断言语句大致转换为以下这些语句：

if __debug__:

if not expression1:

raise AssertionError(expression2)

需要说明两点：

*，代码在检查断言条件之前，还会检查__debug__全局变量。这是一个内置的布尔标记，，内置变量 __debug__ 在正常情况下为 True。

第二，当expression1为真时，程序继续往下执行，只是判断，不做任何处理；

当expression1为假时，中断程序，抛出AssertionError错误，并将expression2消息输出。

 

例1、

def foo(s):
n = int(s)
assert n != 0, ‘n is zero!’
return 10 / n

print(foo(‘2’)) #程序不报错，将该句改为print(foo(‘0’)) # 代码执行结果， AssertionError: n is zero!

例2、本例演示了用try语句捕获assert抛出异常，代码如下：

def testAssert():
for i in range(4):
try:
assert i<2
except AssertionError:
print(‘捕获assert抛出的异常!’)
print(i)
print(‘……end……’)

testAssert()

本例中定义了testAssert函数，当for循环变量i为2时，assert后的条件i<2为假，因此抛出AssertionError，因为这个异常被捕获处理，程序不会中断，直至for循环正常结束。

运行效果如下：

0
1
捕获assert抛出的异常!
2
捕获assert抛出的异常!
3
……end……

 

如何禁用Python中的断言？怎么做？

断言语句是可以被禁用的——需要满足__debug__ == flase，可以这样做：

若在命令行中使用-O 和-OO 标识，或修改Python 中的PYTHONOPTIMIZE环境变量，都会全局禁用断言。

如，，在命令行中使用-O标志（大写字母O）调用Python运行scriptName.py ：

python -O scriptName.py

 

关于__debug__可参见https://docs.python.org/zh-cn/3/library/constants.html?highlight=__debug__#__debug__