python - 如何修补基类方法？

Question

我有两个类:

class A(object):
    def get_something(self):
        return 1

class B(A):
    pass

当我尝试在测试中修补 A.get_something 时，它失败了。

from some_module.myclasses import B

...
def test_something(self):
    with patch('some_module.myclasses.A.get_something', return_value=2):
        self.assertEqual(B().get_something(), 2)

这里有什么问题吗？

Answer 1

发现了。我必须修补导入的 A

from some_module.myclasses import A, B

...
def test_something(self):
    with patch.object(A, 'get_something', return_value=2):
        self.assertEqual(B().get_something(), 2)