如何在python中模拟numpy.linalg？

Question

我想在numpy包中模拟线性代数模块，我一直在尝试以下方法：

from mock import MagicMock
sys.modules["numpy.linalg"] = MagicMock()
import numpy.linalg

我收到以下错误：

File "/usr/lib/python3/dist-packages/numpy/__init__.py", line 167, in <module>
from . import linalg
ImportError: cannot import name 'linalg'

似乎当mock使用以下形式的导入时：

import numpy.linalg

它无法模拟表单的导入：

from numpy import linalg

这有什么解决方法吗？非常感谢你的帮助。