如何在Python单元测试中避免导入父包且不修改生产代码？

这个问题我之前也碰到过，确实挺闹心的——不想动生产代码，但测试总被父包的初始化逻辑卡壳。给你几个不用修改生产代码的解决方案，按需选择就行：

方法1：直接加载模块文件，完全绕过包初始化 #

你可以用importlib.util模块直接读取目标模块的文件内容并加载，彻底跳过父包__init__.py的执行逻辑。这样就不会触发其中的环境变量读取操作了，示例代码如下：

import unittest
import importlib.util
import os

# 根据你的项目结构，调整目标模块的路径
module_path = os.path.join(os.path.dirname(__file__), 'parent', 'child', 'grandchildren.py')

# 加载指定路径的模块
spec = importlib.util.spec_from_file_location('parent.child.grandchildren', module_path)
some_code_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(some_code_module)
some_code = some_code_module.some_code

class SomeCodeTest(unittest.TestCase):
    def test_that_some_code_get_stuff_done(self):
        self.assertEqual(some_code("stuff"), "stuff done")

if __name__ == "__main__":
    unittest.main()

这种方式相当于直接执行目标模块的代码，完全隔离了父包的初始化逻辑，是最彻底的隔离方案。

方法2：Mock环境变量，让父包初始化顺利通过 #

如果不想绕开包导入逻辑，可以在测试启动前，用unittest.mock临时mock环境变量，给父包的__init__.py一个假的有效值，示例代码：

import unittest
from unittest.mock import patch
import os

# 在模块导入前就mock环境变量
@patch.dict(os.environ, {'AN_ENV_VARIABLE': 'dummy_test_value'})
def setUpModule():
    global some_code
    # 此时环境变量已经被mock，导入模块不会报错
    from parent.child.grandchildren import some_code

class SomeCodeTest(unittest.TestCase):
    def test_that_some_code_get_stuff_done(self):
        self.assertEqual(some_code("stuff"), "stuff done")

if __name__ == "__main__":
    unittest.main()

这里利用setUpModule在测试模块加载前执行mock操作，确保导入父包时环境变量已经存在，完美规避了报错问题。

方法3：调整Python路径，直接导入子模块 #

通过把目标子模块所在的目录添加到sys.path，可以直接导入子模块，而不会触发父包的__init__.py执行，示例代码：

import unittest
import sys
import os

# 将child目录添加到Python搜索路径的最前面
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'parent', 'child'))

# 直接导入grandchildren模块，无需经过parent包的初始化
from grandchildren import some_code

class SomeCodeTest(unittest.TestCase):
    def test_that_some_code_get_stuff_done(self):
        self.assertEqual(some_code("stuff"), "stuff done")

if __name__ == "__main__":
    unittest.main()

这个方法要注意路径的正确性，避免和其他同名模块冲突，适合项目结构比较简单的场景。

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