跳到主要内容
Python pytest 框架使用指南:自动化测试入门 | 极客日志
Python
Python pytest 框架使用指南:自动化测试入门 pytest 是 Python 生态中流行的高效测试框架,支持函数式与面向对象风格。通过 pip 安装后,遵循特定命名规则自动发现用例。提供丰富的命令行参数控制执行流程,支持配置文件集中管理选项。内置前后置钩子(fixture/setup)处理环境初始化与清理,原生 assert 语句进行断言验证。支持 @pytest.mark.parametrize 实现数据驱动测试,覆盖多场景。配合 requests 等库可实现接口自动化,插件生态完善,适合复杂项目扩展。
人间失格 发布于 2026/3/16 更新于 2026/5/7 8 浏览一、pytest 介绍
pytest 是一个非常流行且高效的 Python 测试框架,提供丰富功能和灵活用法,让测试用例的编写与运行变得简单高效。
官方文档地址:pytest
1.1 pytest 的优点
简单易用:语法简洁清晰,对编写测试用例非常友好,上手速度快;
强大的断言库:内置丰富断言库,可轻松判断测试结果;
支持参数化测试:允许用不同参数多次运行同一个测试函数,大幅提升测试效率;
丰富的插件生态系统:可通过插件扩展多种功能(如覆盖率测试、测试报告生成、失败用例重复执行等),还支持与 selenium、requests、appium 等结合,实现 Web 自动化、接口自动化、App 自动化测试;
灵活的测试控制:允许跳过指定用例,或标记预期失败的用例,支持重复执行失败用例。
1.2 主流 Python 接口自动化框架对比
维度 unittest(Python 内置) pytest Robot Framework 安装方式 无需安装(Python 标准库) pip install pytest pip install robotframework 语法风格 基于类(需继承 TestCase) 函数式或面向对象(无需样板代码) 关键字驱动(表格化用例) 断言方法 self.assertEqual() 等 原生 assert 表达式 关键字断言(如 Should Be Equal) 参数化支持 需 subTest 或第三方库 内置(@pytest.mark.parametrize) 数据驱动(Test Template) 插件生态 少(依赖扩展库如 HTMLTestRunner) 丰富(如 pytest-html、pytest-xdist、allure-pytest 等) 一般(需安装额外库如 RequestsLibrary) 测试报告 需插件生成报告 支持多格式报告(HTML、Allure 等) 自带详细日志和报告 学习曲线 中等(需熟悉 xUnit 模式) 低(语法简洁) 高(需掌握关键字和语法) BDD 支持 不支持 支持(通过 pytest-bdd 插件) 支持(通过 robotframework-bdd) 适用场景 简单项目或遗留系统维护 复杂项目、高扩展性需求 团队协作、非技术人员参与
二、安装
安装命令:
pip install pytest
[图片]
安装成功验证:
安装完成后,可通过 pip list 命令查看当前项目下 pytest 包是否安装成功。同时需确认 PyCharm 中 Python 解释器已更新,安装 pytest 前后的代码运行差异如下:
未安装 pytest:需编写 main 函数手动调用测试用例;
安装 pytest:测试方法名前会出现直接运行标志,无需手动编写调用代码。
[图片]
三、用例运行规则 要让 pytest 自动发现并运行测试用例,需遵循以下命名规则:
文件名必须以 test_ 开头或者 _test 结尾;
测试类必须以 Test 开头,并且不能有 __init__ 方法;
测试方法必须以 test 开头。
满足规则后,可通过命令行执行 pytest 命令直接运行符合条件的用例:
注意事项:
Python 测试类中不可添加 __init__ 方法,否则 pytest 会抛出 pytest_collectionWarning 警告,无法收集该测试类。原因是 pytest 采用自动发现机制收集测试用例,会自动实例化测试类并调用其 test 开头的方法,若存在 __init__ 方法,可能掩盖测试逻辑并引入额外副作用,影响测试结果准确性。
若测试类需初始化操作,可使用 setUp() 和 tearDown() 方法、类属性或 fixture 函数替代。
四、pytest 命令参数
4.1 常见参数 pytest 提供丰富的命令行选项控制测试执行,常用参数及说明如下:
命令 描述 备注 pytest 在当前目录及其子目录中搜索并运行测试 - pytest -v 增加输出的详细程度 - pytest -s 显示测试中的 print 语句 - pytest test_module.py 运行指定的测试模块 - pytest test_dir/ 运行指定目录下的所有测试 - pytest -k 只运行测试名包含指定关键字的测试 - pytest -m 只运行标记为指定标记的测试 - pytest -q 减少输出的详细程度 - pytest --html=report.html 生成 HTML 格式的测试报告 需要安装 pytest-html 插件 pytest --cov 测量测试覆盖率 需要安装 pytest-cov 插件
4.2 命令使用示例 详细打印并显示 print 内容(-s 和 -v 可连写为 -sv):
五、pytest 配置文件 当需要频繁使用复杂命令参数时,可将配置统一写入 pytest.ini 文件(项目根目录下创建),避免重复输入。
5.1 常见配置选项 参数 解释 addopts 指定在命令行中默认包含的选项 testpaths 指定搜索测试的目录 python_files 指定发现测试模块时使用的文件匹配模式 python_classes 指定发现测试类时使用的类名前缀或模式 python_functions 指定发现测试函数和方法时使用的函数名前缀或模式 norecursedirs 指定在搜索测试时应该避免递归进入的目录模式 markers 定义测试标记,用于标记测试用例
5.2 配置示例 配置 pytest.ini 文件,实现详细输出 cases 包下文件名以 test_ 开头、类名以 Test 开头的所有用例:
[pytest]
addopts = -vs
testpaths = cases
python_files = test*.py
配置完成后,命令行直接执行 pytest 命令即可,无需额外指定参数,运行结果如下:
六、前后置操作 前后置操作用于在测试用例执行前后完成环境设置、数据准备、资源清理等工作,pytest 提供三种实现方式:
6.1 setup_method 和 teardown_method 用于类中每个测试方法的前置和后置操作,每个测试方法执行前都会触发 setup_method,执行后触发 teardown_method。
import pytest
class TestExample :
def setup_method (self ):
print ("Setup: Before each test" )
def teardown_method (self ):
print ("Teardown: After each test" )
def test_example1 (self ):
print ("Running test_example1" )
def test_example2 (self ):
print ("Running test_example2" )
6.2 setup_class 和 teardown_class 用于整个测试类的前置和后置操作,测试类中所有方法执行前触发一次 setup_class,所有方法执行后触发一次 teardown_class。
class TestExample :
@classmethod
def setup_class (cls ):
print ("Setup: Before all test" )
@classmethod
def teardown_class (cls ):
print ("Teardown: After all test" )
def test_example1 (self ):
print ("Running test_example1" )
def test_example2 (self ):
print ("Running test_example2" )
七、断言 断言(assert)是一种调试辅助工具,用于检查程序状态是否符合预期,若断言失败(条件为假),Python 解释器会抛出 AssertionError 异常。pytest 支持使用标准 Python assert 语句验证预期值与实际值。
条件:必须是布尔表达式;
错误信息:条件为假时显示的提示信息(可选)。
7.1 基本数据类型断言:
a = 1
b = 2
assert a == b, "a 和 b 不相等"
str_val = "hello"
assert "hello" == str_val
7.2 数据结构断言: def test ():
expect_list = [1 , 'apple' , 3.14 ]
actual_list = [1 , 'apple' , 3.14 ]
expect_tuple = (1 , 'apple' , 3.14 )
actual_tuple = (1 , 'apple' , 3.14 )
expect_dict = {'name' : 'Alice' , 'age' : 25 }
actual_dict = {'name' : 'Alice' , 'age' : 25 }
expect_set = {1 , 2 , 3 , 'apple' }
actual_set = {1 , 2 , 3 , 'apple' }
assert expect_list == actual_list
assert expect_tuple == actual_tuple
assert expect_dict == actual_dict
assert expect_set == actual_set
7.3 函数断言: def divide (a, b ):
assert b != 0 , "除数不能为 0"
return a / b
print (divide(10 , 2 ))
print (divide(10 , 0 ))
7.4 接口返回值断言: import requests
from jsonschema.validators import validate
def test1 ():
url = "http://jsonplaceholder.typicode.com/posts/1"
r = requests.get(url=url)
expect_data = {
"userId" : 1 ,
"id" : 1 ,
"title" : "sunt aut facere repellat provident occaecati excepturi optio reprehenderit" ,
"body" : "quia et suscipit\nsuscipit recusandae consequuntur expedita et cum\nreprehenderit molestiae ut ut quas totam\nnostrum rerum est autem sunt rem eveniet architecto"
}
print (r.json())
assert r.json() == expect_data
assert r.json()['userId' ] == 1
def test2 ():
url = "http://jsonplaceholder.typicode.com/comments?postId=1"
r = requests.get(url=url)
print (r.json())
assert r.json()[1 ]['id' ] == 1
def test3 ():
url = "http://jsonplaceholder.typicode.com/"
r = requests.get(url=url)
assert "Use your own data" in r.text
八、参数化 参数化设计可让测试用例通过不同参数多次运行,提高测试效率和覆盖度,pytest 通过 @pytest.mark.parametrize 装饰器实现参数化,支持在测试函数、类、模块级别使用。
8.1 测试函数参数化 import pytest
@pytest.mark.parametrize("test_input, expected" , [
("3+5" , 8 ),
("2+4" , 6 ),
("6*9" , 42 )
] )
def test_eval (test_input, expected ):
assert eval (test_input) == expected
装饰器定义了三组 (test_input, expected) 元组,test_eval 函数会依次使用每组参数运行三次。
8.2 测试类参数化 import pytest
@pytest.mark.parametrize("n, expected" , [(1 , 2 ), (3 , 4 )] )
class TestClass :
def test_simple_case (self, n, expected ):
assert n + 1 == expected
def test_weird_simple_case (self, n, expected ):
assert (n * 1 ) + 1 == expected
参数集将作用于类中所有测试方法,每个方法都会使用每组参数运行。
8.3 模块级别参数化 通过给 pytestmark 全局变量赋值,实现模块内所有测试的参数化:
import pytest
pytestmark = pytest.mark.parametrize("n, expected" , [(1 , 2 ), (3 , 4 )])
class TestClass :
def test_simple_case (self, n, expected ):
assert n + 1 == expected
def test_weird_simple_case (self, n, expected ):
assert (n * 1 ) + 1 == expected
8.4 自定义参数化数据源 def data_provider ():
return ["a" , "b" ]
@pytest.mark.parametrize("data" , data_provider( ) )
def test_data (data ):
assert data is not None
print (f"Testing with data provider: {data} " )
除了使用 @parametrize 添加参数外,pytest.fixture() 允许对 fixture 函数进行参数化,详情参考下一篇文章。
相关免费在线工具 curl 转代码 解析常见 curl 参数并生成 fetch、axios、PHP curl 或 Python requests 示例代码。 在线工具,curl 转代码在线工具,online
Base64 字符串编码/解码 将字符串编码和解码为其 Base64 格式表示形式即可。 在线工具,Base64 字符串编码/解码在线工具,online
Base64 文件转换器 将字符串、文件或图像转换为其 Base64 表示形式。 在线工具,Base64 文件转换器在线工具,online
Markdown转HTML 将 Markdown(GFM)转为 HTML 片段,浏览器内 marked 解析;与 HTML转Markdown 互为补充。 在线工具,Markdown转HTML在线工具,online
HTML转Markdown 将 HTML 片段转为 GitHub Flavored Markdown,支持标题、列表、链接、代码块与表格等;浏览器内处理,可链接预填。 在线工具,HTML转Markdown在线工具,online
JSON 压缩 通过删除不必要的空白来缩小和压缩JSON。 在线工具,JSON 压缩在线工具,online