如何让导入的旧应用函数使用新作用域且不修改原文件？

这种遗留代码的测试困境我太懂了——不想碰已经跑起来的旧代码，但又得给它套上单元测试的壳。好在Python的动态特性给了我们几个不用修改原文件的解决方案，我给你梳理一下：

1. 用exec加载原文件到自定义命名空间 #

这是最灵活的方案，相当于给原代码创建一个“沙盒”环境，你可以完全控制里面的变量。具体做法是：

• 创建一个字典作为自定义的命名空间，预先注入你需要模拟的变量
• 读取原文件的代码，用exec把它执行在这个自定义命名空间里
• 从命名空间里取出你要测试的函数，它会自动使用你注入的变量

示例代码：

# test_old_app.py
# 先定义好你需要的模拟变量，对应原文件里函数依赖的局部/模块变量
custom_env = {
    'config_value': 'test_config',
    'calculation_base': 100
}

# 读取原应用文件的内容，执行到自定义环境中
with open('old_app.py', 'r', encoding='utf-8') as f:
    exec(f.read(), custom_env)

# 从自定义环境中获取目标函数
target_func = custom_env['your_function_name']
# 调用函数，它会使用我们注入的变量
result = target_func()

这个方法不管原函数依赖的是模块级变量还是嵌套作用域的变量，都能覆盖到，完全不用动原文件。

2. 修改模块级函数的全局变量 #

如果你的目标函数是模块级的（不是嵌套在其他函数里），依赖的是原文件的模块全局变量，那这个方法更简单：

• 直接导入原模块
• 修改模块的全局变量，或者直接修改函数的__globals__字典

示例代码：

import old_app

# 方法一：直接修改模块的全局变量
old_app.config_value = 'test_config'

# 方法二：修改函数的globals字典（适合不想污染模块全局的情况）
old_app.your_function_name.__globals__['calculation_base'] = 100

# 调用函数，它会使用修改后的变量
result = old_app.your_function_name()

这个方法更轻量，但只适用于依赖模块全局变量的函数，嵌套函数的闭包变量没法用这个方法修改。

3. 手动修改闭包变量（hack但有效） #

如果你的函数是嵌套在其他函数里的闭包，依赖的是外层函数的局部变量，那可以通过修改函数的__closure__属性来替换变量值。这个方法有点hack，但确实能绕过修改原文件的限制：

示例代码：

import old_app
import inspect

# 获取目标函数的闭包cell对象（每个cell对应一个闭包变量）
closure_cells = old_app.your_nested_function.__closure__

# 找到你要修改的变量对应的cell（需要知道变量在闭包中的顺序，或者通过inspect辅助查找）
# 比如假设第一个cell是我们要修改的`x`变量
target_cell = closure_cells[0]
# 修改cell的内容
target_cell.cell_contents = 200

# 调用函数，它会使用新的变量值
old_app.your_nested_function()

注意：这个方法依赖Python的内部实现，不同版本可能有差异，而且需要你清楚闭包变量的顺序，适合实在没别的办法的情况。

最后提醒 #

这些方法都是临时的权宜之计，虽然能帮你快速写完测试，但长期来看，还是建议逐步重构旧代码——比如把依赖的变量改成函数参数，或者封装成类的属性，这样后续维护和测试都会更轻松。不过既然你现在不想动原代码，上面的方案应该能解决你的问题。

内容的提问来源于stack exchange，提问作者lekterable