Python学习之assert()函数的使用

assert()函数的作用

断言函数是对表达式布尔值的判断，要求表达式计算值必须为真。可用于自动调试。

如果表达式为假，触发异常；如果表达式为真，不执行任何操作。

1、简单举例

assert(1 == 1)
assert(5 > 1)
a = "hello"
b = "hello"
assert(a == b)

assert()中表达式都为真，故程序正常执行，不会报错，也不会有输出 

2、使用assert判断数组是否相等

当我们的数组是list类型的时候，可直接进行比较。如下我们使用assert（a==b）判断a，b是否相等，可以顺利通过：、

a = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
b = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
assert(a == b)

但当a,b为numpy.array类型时，需要采用下列方式：

import numpy as np
a = np.arange(10)
b = np.arange(10)
assert(a.all() == b.all())