3.4.5 断言

断言(assertions)的使用方式类似于if语句,只是在不满足条件时,会直接抛出异常。类似于下面的if语句(伪代码)

if not condition:       #如果不满足条件,会直接抛出异常,程序会中断
    crash program

那么究竟为什么需要这样的代码呢?主要原因为需要检测程序在某个地方是否满足条件,如果不满足条件,应该及时通知开发人员,而不是将这些bug隐藏起来,知道关键的时候在崩溃。

其实在TDD中经常使用断言,TDD会在程序发现异常时执行断言,并抛出异常。

在Python语言中,断言需要使用assert语句,在assert关键字的后面指定断言的条件表达式。如果条件表达式的值是False,那么就会抛出异常。而且断言后面的语句都不会执行,相当于程序的一个断点。

>>> value = 20
>>> assert value < 10 or value > 30     #条件不满足,会抛出异常
>> assert value < 30        #条件满足,会正常执行后面的语句

可以看到,value变量的值是20,而assert后面的条件是“value < 10 or value > 30”,很明显,条件不满足,所以在断言出会抛出异常。而后面的断言,条件是“value < 30”,这个条件是满足的,所以在断言后面的语句都会正常执行。

当断言条件不满足时抛出异常,在默认情况下,只显示了抛出异常的位置,并没有显示抛出异常的原因,所以为了异常信息更明确,可以为assert语句指定异常描述。

>>> value = 20
>>> assert value < 10 or value > 30,'value值必须在10~20'    #为断言指定异常描述信息

[例 3.6] 本例演示了断言的用法。

name = "Bill"       #定义变量name
assert name == "Bill"       #断言条件表达式的值是True,继续执行下面的语句

age = 20        #定义变量age
#断言条件表达式的值是False,抛出异常,后面的代码不会被执行
assert 0 < age < 10, '年龄必须小于10岁'

print("hello world")        #这行代码不会被执行