python-3.x - Python用“=”调用模拟未调用结果

Question

我正在尝试模拟以下调用：

df_x = method() # returns a pandas dataframe
df_x.loc[df_x['atr'] < 0, 'atr'] = 0

我已经模拟了该方法，因此它返回一个 MagicMock 并将默认值设置为 MagicMock 的 __ getitem__ 属性，如下所示：

mock_df_x = mock_method.return_value
mock_df_x.__getitem__.return_value = 0

问题是当我尝试断言调用时：

mock_df_x.loc.__getitem__.assert_called_with(False, 'atr')

我得到一个不称为错误的函数。如果我在没有“= 0”部分的情况下调用这样的函数，则断言有效。

df_x.loc[df_x['atr'] < 0, 'atr']

标签： python-3.xpandasmockingpytestassertion

Answer 1

您看到这种不同行为的原因取决于= 0您在测试的调用结束时是否有 on ，因为在Python 的 data model中，它们对应于两种不同的魔术方法：__getitem__和__setitem__.

这是有道理的，因为例如做some_dictionary['nonexistent_key]'raises KeyError，而some_dictionary['nonexistent_key]' = 1没有做，并按预期设置值。

现在，为了修复你的测试，你只需要改变你的断言：

mock_df_x.loc.__getitem__.assert_called_with((False, 'atr'))

仅当您访问密钥时才有效，以：

mock_df_x.loc.__setitem__.assert_called_with((False, 'atr'), 0)

如果您尝试为该键分配值，则该方法有效。请注意额外的参数，也对应于您实际尝试分配的值。