11：pytest 框架 assert 验证测试

一、断言是什么？为什么 pytest 推荐用原生 assert？

断言是一种调试辅助工具，用于检查程序的状态是否符合预期：

当 assert 后的条件为 True 时，程序正常执行，无任何输出；
当条件为 False 时，Python 解释器会抛出 AssertionError 异常，终止当前用例执行，并打印错误信息。

在传统的 unittest 框架中，你需要用 self.assertEqual(a, b)、self.assertTrue(x) 这类专用方法来断言，不仅语法冗余，还需要额外记忆不同场景的方法名。

而 pytest 直接支持 Python 原生的 assert 语句，带来了两大核心优势：

语法简洁自然：和你平时写 Python 代码的逻辑完全一致，不用额外学习新语法；
错误信息超详细：pytest 会自动解析断言的表达式，生成包含预期值、实际值、差异对比的错误信息，不用手动拼接调试日志。

二、pytest 断言的基本语法

pytest 断言的核心语法非常简单：

python 复制代码

assert 布尔表达式, "断言失败时显示的错误信息（可选）"

布尔表达式：必须是一个结果为 True 或 False 的表达式，比如 a == b、len(list) > 0、"success" in response_text 等；
错误信息：可选参数，当断言失败时，会打印这段自定义信息，方便快速定位问题。

举个最简单的例子：

python 复制代码

def test_basic_assert():
    a = 1
    b = 1
    # 断言两个整数相等，失败时打印自定义信息
    assert a == b, f"预期值 {a} 与实际值 {b} 不相等"

当 a != b 时，pytest 会直接输出错误信息和详细的对比日志。

三、不同数据类型的断言实战

pytest 的 assert 支持所有 Python 数据类型的比较，下面是自动化测试中最常用的场景示例：

1. 基本数据类型断言

适用于整数、浮点数、字符串、布尔值等基础类型的验证：

python 复制代码

def test_basic_type_assert():
    # 断言整数相等
    a = 1
    b = 1
    assert a == b, "整数断言失败"

    # 断言字符串匹配（支持包含、不包含、大小写判断）
    result_str = "登录成功"
    assert "成功" in result_str, "返回结果中不包含'成功'字样"
    assert result_str.startswith("登录"), "返回结果不是以'登录'开头"

    # 断言浮点数（注意：浮点数直接用 == 可能有精度问题，建议用近似比较）
    pi = 3.14159
    assert abs(pi - 3.14) < 0.01, "圆周率精度不符合要求"

    # 断言布尔值
    is_login = True
    assert is_login, "登录状态为 False"

2. 复合数据结构断言

列表、元组、字典、集合这类数据结构，pytest 会自动递归对比每个元素，错误时还会高亮显示差异位置，非常适合接口返回数据、GUI 列表数据的验证：

python 复制代码

def test_data_structure_assert():
    # 1. 断言列表
    expect_list = [1, 'apple', 3.14]
    actual_list = [1, 'apple', 3.14]
    assert expect_list == actual_list, "列表数据不一致"

    # 2. 断言元组
    expect_tuple = (1, 'apple', 3.14)
    actual_tuple = (1, 'apple', 3.14)
    assert expect_tuple == actual_tuple, "元组数据不一致"

    # 3. 断言字典（最常用，比如接口返回的JSON数据）
    expect_user = {'name': 'Alice', 'age': 25, 'is_vip': True}
    actual_user = {'name': 'Alice', 'age': 25, 'is_vip': True}
    assert expect_user == actual_user, "用户信息不一致"

    # 进阶：只断言字典中的部分字段，避免无关字段影响用例
    assert actual_user['name'] == 'Alice', "用户名错误"
    assert actual_user['age'] == 25, "用户年龄错误"

    # 4. 断言集合（适合验证无重复元素的场景）
    expect_set = {1, 2, 3, 'apple'}
    actual_set = {1, 2, 3, 'apple}
    assert expect_set == actual_set, "集合数据不一致"

3. 函数与异常断言

除了直接比较值，assert 还可以用来验证函数的返回值，甚至配合 pytest 验证异常是否正确抛出：

python 复制代码

# 示例：除法函数，除数为0时抛出断言错误
def divide(a, b):
    assert b != 0, "除数不能为0"
    return a / b

def test_function_assert():
    # 正常情况：断言函数返回值
    result = divide(10, 2)
    assert result == 5.0, "除法计算结果错误"

    # 异常情况：验证函数是否按预期抛出异常
    import pytest
    with pytest.raises(AssertionError, match="除数不能为0"):
        divide(10, 0)

四、pytest 断言的错误信息优化技巧

pytest 原生的错误信息已经很详细，但在复杂场景下，我们可以通过自定义错误信息，让调试效率更高：

1. 用 f-string 拼接动态错误信息

把预期值、实际值、关键变量拼接到错误信息中，失败时一眼就能看到差异：

python 复制代码

def test_dynamic_error_msg():
    actual_code = 404
    expect_code = 200
    assert actual_code == expect_code, f"接口状态码错误，预期：{expect_code}，实际：{actual_code}"

2. 断言失败时打印额外日志

如果需要打印更多调试信息（比如 GUI 操作的截图路径、接口响应的完整日志），可以用逗号拼接多个值，pytest 会自动打印所有内容：

python 复制代码

def test_gui_assert():
    welcome_text = "欢迎回来，测试用户"
    expect_text = "欢迎回来，管理员"
    screenshot_path = "./screenshots/login_fail.png"
    assert welcome_text == expect_text, f"欢迎语不匹配，截图已保存至：{screenshot_path}"

3. 利用 pytest 的断言重写功能

pytest 会自动重写 assert 语句，生成更详细的对比信息，比如列表、字典的差异高亮，不用手动实现复杂的对比逻辑。

五、GUI 自动化测试中常用的断言场景

在 GUI 自动化测试中，断言的核心是验证界面状态是否符合预期，以下是高频场景示例：

1. 验证元素是否存在 / 可见

python 复制代码

from selenium import webdriver
from selenium.webdriver.common.by import By

def test_element_display():
    driver = webdriver.Chrome()
    driver.get("https://example.com/login")

    # 验证用户名输入框是否存在
    username_input = driver.find_element(By.ID, "username")
    assert username_input is not None, "用户名输入框未找到"

    # 验证登录按钮是否可见
    login_btn = driver.find_element(By.ID, "login-btn")
    assert login_btn.is_displayed(), "登录按钮不可见"
    assert login_btn.is_enabled(), "登录按钮处于禁用状态"

2. 验证文本内容与属性值

python 复制代码

def test_text_and_attribute():
    driver = webdriver.Chrome()
    driver.get("https://example.com/home")

    # 验证欢迎文本
    welcome_text = driver.find_element(By.ID, "welcome-msg").text
    assert "欢迎" in welcome_text, "欢迎文本未显示"
    assert welcome_text == "欢迎回来，测试用户", "欢迎文本内容错误"

    # 验证元素属性（比如链接的 href、输入框的 value）
    forgot_pwd_link = driver.find_element(By.ID, "forgot-pwd")
    assert forgot_pwd_link.get_attribute("href") == "https://example.com/forgot", "忘记密码链接地址错误"

    username_input = driver.find_element(By.ID, "username")
    username_input.send_keys("test_user")
    assert username_input.get_attribute("value") == "test_user", "输入框内容未正确填充"

3. 验证页面跳转与 URL

python 复制代码

def test_url_redirect():
    driver = webdriver.Chrome()
    driver.get("https://example.com/login")

    # 执行登录操作
    driver.find_element(By.ID, "username").send_keys("test_user")
    driver.find_element(By.ID, "password").send_keys("test_pwd")
    driver.find_element(By.ID, "login-btn").click()

    # 验证跳转后的URL
    assert "home" in driver.current_url, "登录后未跳转到首页"
    assert driver.current_url == "https://example.com/home", "跳转URL错误"

六、断言使用的最佳实践与避坑指南

1：一个用例中不要写太多断言:

避免前面的断言失败后，后面的每个用例只断言一个核心结果断言无法执行，无法判断所有问题。

比如登录用例，优先断言 "是否跳转到首页"，再断言 "欢迎文本是否正确"，最后断言 "用户头像是否显示"，每个断言对应一个独立的验证点。

2：不要用断言来控制业务逻辑:

断言的作用是 "验证状态"，不是 "控制流程"。

比如不要用 assert user is not None 来替代 if user is not None: 这种判断，因为断言在生产环境中可以被全局禁用，会导致业务逻辑出错。

3：浮点数断言要注意精度问题：

直接用 == 比较浮点数可能会因为精度误差导致断言失败，建议用 abs(actual - expect) < 误差范围的方式

python 复制代码

assert abs(3.1415 - 3.14) < 0.01。

4：断言失败信息要清晰可维护:

错误信息要包含 "预期值、实际值、错误场景"，不要写模糊的 "断言失败"，否则调试时还要重新运行才能看到具体差异。

5：GUI 自动化中，断言前要加等待:

元素加载、页面跳转需要时间，断言前一定要加显式等待，避免因为元素还没加载完成导致断言失败

python 复制代码

WebDriverWait(driver, 10).until(lambda d: d.find_element(By.ID, "welcome-msg").is_displayed())。

七、总结

pytest 让断言这件事变得前所未有的简单：用原生的 assert 语句，就能搞定从基础数据类型到复杂数据结构、从普通函数到 GUI 界面的所有验证场景，再加上详细的错误信息和灵活的自定义方式，完全能满足自动化测试中的所有断言需求。

掌握断言的正确用法，不仅能让你的测试用例更简洁易读，还能大幅提升调试效率，快速定位问题所在。后续我们可以结合 pytest-html 报告，把断言失败时的截图、日志嵌入报告，让测试结果更直观。