Осмотреть объект класса смоделированного метода

Question

Я пишу тест функции, которая отправляет фрейм данных pandas в базу данных sql, используя метод to_sql() фрейма данных.

Чтобы вообще ничего не отправлять, я издеваюсь над методом, используя mock.patch('pandas.DataFrame.to_sql').

Я хотел бы видеть состояние кадра данных на момент вызова to_sql(). Есть ли способ сделать это с помощью макета? Т.е. известно ли об объекте, из которого была вызвана максируемая функция?

Функция, которую я тестирую, в настоящее время ничего не возвращает. Будет ли это необходимо для проверки данных? Например. возвращать фрейм данных после вызова to_sql()? Мне не нужно возвращаемое значение ни для чего, кроме выполнения этого теста, поэтому я предпочел бы другой способ, если это возможно.

Answer 1

У меня была такая же проблема, и решение таково:

@mock.patch('pandas.DataFrame.to_sql', return_value=mock.MagicMock())
def test_method(self: my_to_sql_mock):
  #your code...

Я надеюсь, что это работает для вас