python中的assert用法
时间: 2024-04-19 21:21:49 浏览: 9
在Python中,`assert`是一种用于断言的关键字。它用于在代码中检查某个条件是否为真,如果条件为假,则会引发`AssertionError`异常。`assert`的语法如下:
```python
assert condition, message
```
其中,`condition`是需要检查的条件,如果条件为假,则会触发异常;`message`是可选的错误信息,用于在触发异常时显示。
`assert`的作用是在开发过程中进行调试和测试,用于确保程序的正确性。它可以用于验证输入参数、检查函数返回值、确认代码逻辑等。
下面是一些使用`assert`的示例:
```python
# 验证输入参数
def divide(a, b):
assert b != 0, "除数不能为0"
return a / b
# 检查函数返回值
def get_positive_number(numbers):
for num in numbers:
assert num > 0, "列表中存在非正数"
return numbers
# 确认代码逻辑
def calculate_discount(price, discount):
assert 0 <= discount <= 1, "折扣应在0到1之间"
return price * (1 - discount)
```