首页 > > 网络编程 > 其它 >

python assert的用处

2019-04-11 10:21:55来源：博客园阅读 ()

python assert 句语格式及用法很简单。通常程序在运行完之后抛出异常，使用assert可以在出现有异常的代码处直接终止运行。而不用等到程序执行完毕之后抛出异常。

python assert的作用

python 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__())

输出：

1 AssertionError
2 traceback 打印异常
3 Traceback (most recent call last):
4   File "main.py", line 7, in <module>
5     assert a < 0
6 AssertionError
7 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))  # 抛出异常

输出：

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