Python возвращает объект MagicMock вместо return_value ⇐ Python

[Anonymous]

У меня есть файл Python a.py, который содержит два класса A и B.

Код: Выделить всё

class A(object):
def method_a(self):
return "Class A method a"

class B(object):
def method_b(self):
a = A()
print a.method_a()

Я хотел бы протестировать метод_b в классе B, имитируя A. Вот содержимое файла testa.py для этой цели:

Код: Выделить всё

import unittest
import mock
import a

class TestB(unittest.TestCase):

@mock.patch('a.A')
def test_method_b(self, mock_a):
mock_a.method_a.return_value = 'Mocked A'
b = a.B()
b.method_b()

if __name__ == '__main__':
unittest.main()

Я ожидаю получить на выходе Mocked A. Но я получаю следующее:

Код: Выделить всё

Где я делаю не так?

Подробнее здесь: https://stackoverflow.com/questions/381 ... turn-value