Python assert 断言函数

assert 断言的执行逻辑是：
if 表达式的值为 True：
      程序继续运行；
else： # 表示式的值为 False
       程序引发 AssertionError 错误

assert(b.shape == (5,1)) #b的形状为(5,1)时，代码继续运行，否则报错