Как смоделировать / исправить функцию без возвращаемого значения, которое изменяет значения параметров?

Question

скажи, что у меня есть функция

def change_dict_value(abc):
  # doing a lot of calculations here
  # and finally generate a new dictionary efg
  abc.update(efg)

Как мне смоделировать / пропатчить функцию, чтобы установить возвращаемое значение abc после обработки в change_dict_value?

Answer 1

Вы можете использовать параметр side-effect .

def mock_change_dict_value(abc):
    # Change abc
    abc.update(efg)

Тогда вы можете использовать его так:

>>> mock = Mock()
>>> mock.side_effect = mock_change_dict_value
>>> mock()