Question

我正在编写一个函数测试，该函数使用数据框的to_sql()方法将熊猫数据框发送到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...

我希望这对您有用