Python assert语句的用法

Python 中的 assert 语句可以帮助用户检测程序代码中的错误。

assert 语句的语法格式如下：

assert <测试码> [, 参数]

测试码是一段返回 True 或 False 的程序代码：

• 若测试码返回 True，则继续运行后面的程序代码；
• 若测试码返回 False，assert 语句则会输出一个 AssertionError 异常，并输出 assert 语句的[参数]作为错误信息字符串。

下面的示例是当变量 a 等于 0 时，输出一个 AssertionError 异常。

a = 100
assert (a != 0), "Error happened, a = 0"
a = 0
assert (a != 0), "Error happened, a = 0"

输出结果如下：
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AssertionError: Error happened, a = 0

下面的示例检测函数的参数类型是否是字符串，如果函数的参数类型不是字符串，就输出一个 AssertionError 异常。

import types
def checkType(arg):
    assert type(arg) ==str, "参数类型不是字符串"
　　
checkType(1)

输出结果如下：
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 2, in checkType
AssertionError: 参数类型不是字符串