2

我正在尝试模拟一个库（matplotlib为了它的价值），并且遇到了一个问题，即当调用模拟并期望返回一个元组时它会失败。有一个更好的方法吗？

Python 3.7.2 (default, Jan 13 2019, 12:50:15) 
[Clang 10.0.0 (clang-1000.11.45.5)] on darwin

>>> import mock
>>> foo = mock.MagicMock()
>>> a, b = foo()

这是收到的错误消息：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: not enough values to unpack (expected 2, got 0)
>>>
4

2 回答 2

1

（对于那些对一个模拟返回不同的不同值感到好奇的人，请参阅我side_effect在下面答案末尾的使用。）

我将从您问题中的第一个陈述开始：

我正在尝试模拟一个图书馆（matplotlib为了它的价值）

如果你能够正确地模拟这个库，你就不必担心任何子函数的行为——因为它们都会被模拟，你可以确定模拟函数的行为方式。我如何实现类似的行为是通过使用patch- 具体来说，patch.object应该有助于：

import mock

mock_matplotlib = mock.MagicMock()
mock_matplotlib.return_value = whatever_mock_behavior_you_want

with mock.patch.object(my_app, "matplotlib", mock_matplotlib):
    # Call the function you are testing which uses matplotlib

您现在可以自由地替换您想要的任何内容，mock_matplotlib以便它以您寻求的方式运行......（即mock_matplotlib.return_value = mock_tuple）让我们探索其中的一些可能性。

当您提到MagicMock总是MagicMock为调用它的任何内容返回 a 时，您是正确的-但是，使用 的语句，Pythona, b = foo()期望找到一个元组，并且在MagicMock可以创建默认的单个MagicMock返回值之前触发错误，因此错误表示它找到可以0使用的值。

您创建MagicMocks 元组的解决方案可以采用多种不同的方式 - 从您的问题中不清楚分配值是否a, b =对您更重要。如果是，那么你可以这样做：

import mock
foo = mock.MagicMock()
a, b = (mock.MagicMock(), mock.MagicMock())

或这个：

a, b = (foo(), foo())

或者，如果您对让一个函数调用为您填充这些值更感兴趣，如前所述，您可以像这样分配return_value：

foo = mock.MagicMock()
foo.return_value = (mock.MagicMock(), mock.MagicMock())
a, b = foo()

甚至像这样：

foo = mock.MagicMock(return_value=(mock.MagicMock(), mock.MagicMock()))
a, b = foo()

对于那些对此评论感到好奇的人，OP 是正确的，这与每次MagicMock调用时返回不同的值不同。那将使用该side_effect选项-但是，即使是这样也可以用来实现 OP 的目标-尽管是以一种迂回的方式：

mock_1 = mock.MagicMock()
mock_2 = mock.MagicMock()
foo = mock.MagicMock(side_effect=[mock_1, mock_2])

a, b = (foo(), foo())

在最后一个示例中，mock_1分配给a，并且mock_2分配给b。

于 2021-05-11T21:46:14.973 回答
0

您可以使用您选择的元组设置对象的return_value属性：MagicMock

import mock
foo = mock.MagicMock()
foo.return_value = 1, 2
a, b = foo()
assert a == 1
assert b == 2
于 2019-03-14T22:05:06.880 回答