如何使用Python pytest测试具有递归逻辑的算法函数?

浅墨君_5406

浅墨君_5406

2026-07-05

677人浏览

原创

递归函数测试时遇 recursionerror 应临时调高栈限制,如用 sys.setrecursionlimit(3000) 并及时恢复;边界用例需覆盖 base case、单层递归、临界输入及异常输入;mock 须正确指定 target 路径;参数化测试应使用可控结构数据而非随机数据。

如何使用python pytest测试具有递归逻辑的算法函数?

递归函数测试时 pytest 报 RecursionError 怎么办

pytest 默认的调用栈深度和普通 Python 解释器一致,而递归算法(比如深度优先遍历、阶乘、快速排序的分区递归)在输入稍大时就容易触发 RecursionError: maximum recursion depth exceeded。这不是测试写错了,而是 pytest 运行环境没做栈深度适配。

解决方法不是改算法,而是临时调高限制——但必须仅限测试作用域,避免污染全局或影响其他测试:

  • 在具体测试函数开头用 sys.setrecursionlimit(3000),并在结尾用 sys.setrecursionlimit(1000) 恢复(Python 默认值通常为 1000)
  • 更稳妥的做法是用 pytest.fixture + autouse=True 在 setup/teardown 中控制,但仅对含递归的测试类启用
  • 避免在 conftest.py 全局设置,否则可能掩盖真实栈溢出问题,或导致其他测试意外失败

如何为递归函数设计边界覆盖用例

递归函数的正确性高度依赖 base case 和 recursive case 的衔接,只测几个“看起来能跑通”的输入远远不够。重点不是穷举,而是卡住三个关键点:base case 触发、单层递归、以及刚好越过栈安全阈值的临界输入。

例如测试一个计算斐波那契的 fib(n):

  • fib(0) 和 fib(1) 必须返回 0/1,验证 base case 正确性
  • fib(2) 和 fib(3) 强制走一次和两次递归调用,确认递推逻辑不崩
  • fib(50)(非暴力,用记忆化或迭代验证结果)检查大输入下是否仍收敛,而非无限递归或数值溢出
  • 传入负数如 fib(-1),看是否抛出预期异常(比如 ValueError),而不是静默返回错误值或崩溃

mock 递归调用链中的某一层是否可行

直接 mock fib 自身来打断递归链,在 pytest 中是可行的,但要小心 patch 的 target 路径和作用域。常见错误是 mock 了错误位置,导致被测函数实际仍执行原递归逻辑。

Secure Python Code Sandbox
Secure Python Code Sandbox

安全 Python 代码沙箱:在隔离的沙箱中执行任意 Python 代码,预装 requests、NumPy、pandas 库。返回 stdout、stderr。

下载

假设 fib 定义在 utils.py 中,你在 test_utils.py 里测试:

  • 正确 patch 写法是 @patch('utils.fib'),不是 @patch('test_utils.fib') 或 @patch('__main__.fib')
  • 若想模拟“第二层递归失败”,可让 mock 在第一次调用返回正常值,第二次调用抛异常:mock_fib.side_effect = [1, RuntimeError("simulated deep error")]
  • 注意:mock 后函数不再真正递归,所以无法验证栈行为或性能,仅适合逻辑分支隔离测试

pytest 参数化测试递归深度与输入组合

用 @pytest.mark.parametrize 能高效覆盖多组输入,但对递归函数要注意参数间隐含的深度关系。比如测试树遍历函数 count_nodes(node),不能只给不同 node,还要确保这些 node 构造出已知深度的子树。

推荐做法:

  • 用辅助函数生成可控结构的数据,例如 make_binary_tree(height=3) 返回一棵确定深度的树
  • 参数化列表里显式带上 expected_depth 和 expected_count,让断言可读且可追溯
  • 避免用随机生成数据做 parametrize,否则失败时无法复现,也不利于判断是算法缺陷还是测试数据异常

递归测试最难的从来不是写断言,而是构造可验证、可复现、带深度语义的输入;一旦输入不可控,再多的 pytest 高级特性也救不回模糊的失败原因。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1811

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4764

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1769

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

28337

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3407

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3407

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1203

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2643

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习