Python assert用法
Python assert用法
在Python中,assert
是一个用于调试目的的语句,它用于在代码中设置检查点。当assert
语句后的条件为True
时,程序会正常执行;但如果条件为False
,则会引发一个AssertionError
异常,并可以选择性地输出一条错误消息。
基本用法
assert
语句的基本语法如下:
assert condition, "optional error message"
其中,condition
是一个表达式,它应该计算为True
或False
。如果condition
为False
,则程序将中断,并引发一个AssertionError
异常。"optional error message"
是一个可选的字符串,用于在断言失败时提供额外的错误信息。
示例
下面是一个简单的示例,展示了如何使用assert
语句来检查一个列表是否包含特定的元素:
def find_element(lst, element):
for item in lst:
if item == element:
return True
return False
my_list = [1, 2, 3, 4, 5]
element_to_find = 3
# 使用assert来检查元素是否在列表中
assert find_element(my_list, element_to_find), f"Element