Python макет AttributeError: объект не имеет атрибута ⇐ Python

[Anonymous]

Я пытаюсь написать модульный тест, чтобы убедиться, что метод класса из другого класса вызывается в определенном методе класса.
Я пробовал издеваться, но получаю сообщение об ошибке: AttributeError: объект «A» не имеет атрибута «b». в строке b.bar(). См. упрощенное воспроизведение ошибки:

class A:
def __init__(self, has_b=True):
if has_b:
self.b = b()
def foo(self):
return self.b.bar()


class B:
def __init__(self):
pass
def bar(self):
return True


#Outside the B class:
def b() -> B:
return B()


#Unit Test
class TestA(TestCase):
def setUp(self):
self.mock_b_methods = mock.MagicMock()
self.A = A()
def test_foo(self):
self.mock_b_methods.bar.return_value = True
res = self.A.foo()
assertTrue(res)


Подробнее здесь: https://stackoverflow.com/questions/793 ... -attribute