Получение значения функции класса, смоделированного в python

Question

Я пишу тестовый модуль на языке Python, используя макет и патч, где я смоделировал класс.

class Class:
    def function():
        return 'something'

@patch('Class')
def test(MockClass):
    item = MockClass()
    self.assertEqual(item.function(), 'something)

Когда вызывается item.function, он возвращает объект.Если я вызываю его с помощью item.function.return_value, он возвращает

Любая идея, как получить фактическое значение «что-то», которое должна вернуть item.function ()?

Answer 1

на самом деле это выглядит почти как пример из официальных документов

from unittest.mock import patch

class Class():
    def function(self):
        return 'something'

@patch('__main__.Class')
def test(MockClass):
    item = MockClass()
    item.function.return_value = 'something' 
    assert item.function() == 'something' 

test()