是的,Python 的 assert
语句可以用于验证函数的返回值。assert
语句会在其条件为假时引发 AssertionError
异常。你可以使用 assert
语句来检查函数的返回值是否符合预期。
例如,假设你有一个函数 add
,它接受两个参数并返回它们的和:
def add(a, b): return a + b
你可以使用 assert
语句来验证 add
函数的返回值是否正确:
assert add(1, 2) == 3, "add(1, 2) should return 3"
在这个例子中,如果 add(1, 2)
的返回值不等于 3,assert
语句将引发一个 AssertionError
异常,并显示提供的错误消息。
请注意,assert
语句通常用于开发和测试阶段,以确保代码的正确性。在生产环境中,你可能希望使用更健壮的错误处理机制,例如抛出异常或使用日志记录。