python mock-check方法是否从另一个类调用?

当方法来自不同的类时,如何检查方法是否在另一个方法中调用?

如果他们来自同一个班级,我可以这样做:

from unittest import mock

class A():
  def method_a(self):
    pass

  def method_b(self):
    self.method_a()

a = A()
a.method_a = mock.MagicMock()
a.method_b()
a.method_a.assert_called_once_with()

但是如果 method_a 从不同的阶层,然后 AssertionError 它没有被调用。

我怎么能做同样的检查,如果我有这些类代替(我想检查 method_b 电话 方法A )?:

class A():
  def method_a(self):
    pass

class B():
  def method_b(self):
    A().method_a()