如何解决Python单元测试:从另一个模块模拟有问题的模块/功能
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运行任何东西
预先感谢:)
解决方法
您的主要问题是在导入模块时已经调用了有问题的函数,因此您无法修补该函数(因为修补会导入原始模块,然后调用该函数)。在这种情况下,您必须 在导入测试模块之前对导入本身进行模拟:
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)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。