Как исправить метод базового класса?

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) 

Answer 2

То, что вы делаете правильно, но если он не работает, это может быть из-за других модулей импорта A или B в их имен и не использует тот же пакет, как то, что вы заделаны.

Я видел это происходит, когда вы делаете что-то вроде этого:

mypackage/ 
    __init__.py 
    mod.py 

модули Python

mypackage/mod.py

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

mypackage/__init__.py

from mod import * 

Тогда mypackage.mod.A может быть импортирован как mypackage.A, но ваше исправление mypackage.mod.A