Question

B/moduleB.py定义为：

def text_function():
    raise KeyError

text_function()

ModuleA.py定义为：

from B.moduleB import text_function

class a():
    def __init__(self):
        self.text = 'abc'

    def mul(self, a, b):
        print(text_function() + str(a*b))

if __name__ == "__main__":
    rom = a()
    rom.mul(2, 3)

这是单元测试test_module.py：

from unittest.mock import MagicMock, patch
from moduleA import a

class TestmoduleA(unittest.TestCase):
    def setUp(self):
        pass

    def test_mul(self):
        print(a.mul)
        self.assertTrue(True)

现在，有没有一种方法可以在从moduleA导入时模拟B / moduleB.py中的text_function，使我无法得到KeyError？

基本上，在测试模块A时，我不希望模块B运行任何东西

预先感谢：）

Answer 1

您的主要问题是在导入模块时已经调用了有问题的函数，因此您无法修补该函数（因为修补会导入原始模块，然后调用该函数）。在这种情况下，您必须在导入测试模块之前对导入本身进行模拟：

import unittest
from unittest import mock
import sys

sys.modules['B.moduleB'] = mock.Mock()

from moduleA import a

class TestmoduleA(unittest.TestCase):
    def test_mul(self):
        print(a.mul) 
        self.assertTrue(True)

另请参见how to mock an import。

Python单元测试：从另一个模块模拟有问题的模块/功能

1 个答案: