基本形式

assert condition

当condition为False时，抛出AssertionError

带错误消息形式

assert condition, message

当condition为False时，抛出AssertionError并显示message

1. 验证函数参数

def calculate_discount(price, discount):
    # 确保价格和折扣都是正数
    assert price > 0, "价格必须大于0"
    assert 0 <= discount <= 1, "折扣必须在0-1之间"
    
    return price * (1 - discount)

# 测试
print(calculate_discount(100, 0.2))  # 正常情况
print(calculate_discount(-50, 0.1)) # 触发断言：价格必须大于0

2. 检查函数返回值

def get_user_role(user_id):
    # 模拟从数据库获取用户角色
    roles = {1: "admin", 2: "editor", 3: "viewer"}
    role = roles.get(user_id, None)
    
    # 确保返回值有效
    assert role is not None, f"无效的用户ID: {user_id}"
    return role

# 测试
print(get_user_role(2))  # 返回 'editor'
print(get_user_role(5))  # 触发断言：无效的用户ID: 5

3. 验证数据结构

def process_student_scores(scores):
    # 确保输入是字典且所有值都是数字
    assert isinstance(scores, dict), "输入必须是字典"
    for name, score in scores.items():
        assert isinstance(score, (int, float)), f"{name}的分数必须是数字"
        assert 0 <= score <= 100, f"{name}的分数必须在0-100之间"
    
    # 处理逻辑...
    return {name: "及格" if score >= 60 else "不及格" for name, score in scores.items()}

# 测试
data = {"张三": 85, "李四": 92, "王五": 58}
print(process_student_scores(data))

# 错误数据会触发断言
# invalid_data = {"张三": "优秀", "李四": 75}  # 触发断言：张三的分数必须是数字

使用原则：

• 使用断言检查程序内部状态
• 使用异常处理外部输入和不可控因素
• 不要用断言代替输入验证 - 用户输入错误是预期可能发生的
• 断言不是错误处理机制，而是调试辅助工具

• 可被禁用：使用python -O运行程序时，所有assert语句会被忽略
• 不适合生产环境：生产环境中应使用异常处理机制
• 错误信息有限：基本形式只提供AssertionError，没有具体错误信息
• 性能影响：大量复杂的断言可能影响程序性能

1. 使用描述性的错误消息：帮助快速定位问题
2. 避免副作用：断言条件中不要包含改变程序状态的操作
3. 用于开发阶段：在测试和开发期间使用断言捕捉逻辑错误
4. 不要用于数据验证：用户输入验证应使用if语句和异常处理
5. 保持简单：断言条件应简单明了，避免复杂逻辑