Python 中的断言( assert )

深藏阁楼爱情的钟 2023-01-14 11:00 231阅读 0赞

### Python 中的断言( assert ) ###

我看到网上一些类似文章要么不准确，要么语焉不详，特此写本文，供新手参考。

由Python 文档[https://docs.python.org/3/reference/simple\_stmts.html\#assert][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\_\_][https_docs.python.org_zh-cn_3_library_constants.html_highlight_debug_debug]

[https_docs.python.org_3_reference_simple_stmts.html_assert]: https://docs.python.org/3/reference/simple_stmts.html#assert
[https_docs.python.org_zh-cn_3_library_constants.html_highlight_debug_debug]: https://docs.python.org/zh-cn/3/library/constants.html?highlight=__debug__#__debug__