Skip to main content
 首页 » 编程设计

python之如何使用 python-mock 库

2025年05月04日7exmyth

我在理解 python 中的模拟库时遇到问题。 这是例子: A有两个文件test.py和another.py

测试.py

#! 
from mock import patch 
from another import C 
 
 
class A(object): 
    def method(self): 
        return 2 
 
 
@patch('another.C') 
@patch('test.A') 
class TestB(object): 
 
    def test1(self, mA, mC): 
        print mA, A 
        print mC, C 

另一个.py

class C(object): 
 
    def a(self): 
        return 3 

所以,问题是“为什么 A 有输出:

<MagicMock name='A' id='13985040'> <MagicMock name='A' id='13985040'> 
<MagicMock name='C' id='13993936'> <class 'another.C'> 

为什么不能修补 another.py 中的类? 在我的例子中,我有很重的功能要测试,它从其他文件导入类。 所以我不明白如何正确处理它们。

谢谢

请您参考如下方法:

使用 Python 模拟库,您始终可以在实际使用它的位置修补(模拟)函数或类。

由于您将“C”导入“测试”并在那里使用它,您可以像这样修补它:

@patch('test.C') 
@patch('test.A') 
class TestB(object): 
 
    def test1(self, mA, mC): 
        print mA, A 
        print mC, C 

这会给你这样的输出:

<MagicMock name='A' id='3070076204'> <MagicMock name='A' id='3070076204'> 
<MagicMock name='C' id='3070084940'> <MagicMock name='C' id='3070084940'> 

您可以根据需要模拟任何类、方法或函数。但是您需要在实际使用对象的位置模拟该对象。