У меня есть файл 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()
Я хотел бы unittest method_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
на выходе. Но я получаю:
<MagicMock name='A().method_a()' id='4326621392'>
Где я делаю неправильно?