Python中的assert语句用于检查代码的条件是否为True。如果判断条件为False,assert会抛出AssertionError异常。assert语句的基本语法如下:
assert expression, message
其中,expression是需要判断的条件,如果不为True则抛出异常。message是可选的,用于自定义抛出异常时的错误信息。
以下是一个简单的代码示例,使用assert语句检查条件是否满足:
def divide(x, y):
assert y != 0, "除数不能为0"
return x / y
print(divide(10, 5)) # 输出 2.0
print(divide(10, 0)) # 抛出 AssertionError: 除数不能为0
在这个示例中,我们定义了一个名为divide的函数,它用于计算两个数的商。在函数内部,我们使用了assert语句来检查给定的除数是否为0。如果除数为0,则抛出AssertionError异常。
在Python 3中,assert语句还支持一个新的语法形式。新语法形式中,可以省略message参数,只写assert expression。这时,抛出异常时,会使用默认的错误信息:'AssertionError”。
def divide(x, y):
assert y != 0
return x / y
print(divide(10, 5)) # 输出 2.0
print(divide(10, 0)) # 抛出 AssertionError
需要注意的是,在代码中使用assert语句要谨慎,在生产环境的代码中应该避免使用assert,因为它可能会抛出AssertionError异常,导致程序的异常终止。