如何使用Python unittest在另一个方法中模拟一个方法？

Question

我正在努力模仿另一个方法中的一个方法，也就是说，我有两个类a，B，如下所示：

class A:
    def method_A(self):
        return "Hi"
class B:
    def method_B(self):
        instance_A = A()
        result = instance_A.method_A()
        print(result)

在这里，我想模拟方法A，同时测试方法B，但找不到方法。提前谢谢

Answer 1

我希望这就是你要找的

class A:
    def method_A(self):
        return "Hi"
class B:
    def method_B(self):
        instance_A = A()
        result = instance_A.method_A()
        return result

import mock

def mock_A(self):
    return 'Bye'


def test_method_B():
    with mock.patch.object(A, 'method_A', new=mock_A):
        result = B().method_B()
        assert result == 'Bye'