assert这个关键字我们称之为“断言”,当这个关键字后边的条件为假的时候,程序自动崩溃并抛出AssertionError的异常。

什么情况下我们会需要这样的代码呢?当我们在测试程序的时候就很好用,因为与其让错误的条件导致程序今后莫名其妙地崩溃,不如在错误条件出现的那一瞬间我们实现“自爆”。

一般来说我们可以用Ta在程序中置入检查点,当需要确保程序中的某个条件一定为真才能让程序正常工作的话,assert关键字就非常有用了。

assert语句的语法格式 

assert expression [, arguments]
assert 表达式 [, 参数]

附加说明:assert也可以用于多个表达式的: assert expression1, expression2。

注意:表达式=false 时,则执行其后面的异常。

我们看几个示例

1、单个表达式:

a = 1
assert a < 0, '出错了,a大于0 啊'
print('这里不会输出')

输出:

Traceback (most recent call last):
  File "main.py", line 3, in <module>
    assert a < 0, '出错了,a大于0 啊'
AssertionError: 出错了,a大于0 啊

2、多个表达式:

a = 1
b = -1
assert a > 0, b < 0
print('正常输出,表达式返回真了') # 输出:正常输出,表达式返回真了

3、尝试捕获 assert 异常:

import traceback
try:
    assert a < 0
except AssertionError as aeeor:  # 明确抛出此异常
    # 抛出 AssertionError 不含任何信息,所以无法通过 aeeor.__str__()获取异常描述
    print('AssertionError', aeeor, aeeor.__str__())
    # 通过 traceback 打印详细异常信息
    print('traceback 打印异常')
    traceback.print_exc()
except:  # 不会命中其他异常
    print('assert except')
try:
    raise AssertionError('测试 raise AssertionError')
except AssertionError as aeeor:
    print('raise AssertionError 异常', aeeor.__str__())

输出:

AssertionError
traceback 打印异常
Traceback (most recent call last):
  File "main.py", line 7, in <module>
    assert a < 0
AssertionError
raise AssertionError 异常 测试 raise AssertionError

4、函数调用抛出异常:

# 除法运算
def foo(value, divide):
    assert divide != 0
    return value / divide
print('4除以2 =', foo(4, 2))  # 执行成功
print('4除以0 =', foo(4, 0))  # 抛出异常

输出:

4除以2 = 2.0
Traceback (most recent call last):
  File "main.py", line 8, in <module>
    print('4除以0 =', foo(4, 0))  # 抛出异常
  File "main.py", line 3, in foo
    assert divide != 0
AssertionError

云海天教程网,免费的在线学习python平台,欢迎关注!

来源:PY学习网:原文地址:https://www.py.cn/article.html

hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » python assert是什么