在Python中模拟无限生成器

Question

我正在尝试使用mock库来模拟无限生成器函数。 （如果你有Python 3.3，还是unittest.mock ）

这是无限发电机的最小工作示例。 如果我能够成功地模拟这个，那么我希望能够模拟我正在使用的实际功能。

import itertools
def infinite_generator():
    thing = itertools.cycle([1, 2])
    while True:
        yield next(thing)

这是我到目前为止所尝试的：

import mock
import itertools
mock_func = mock.MagicMock()
mock_func.__iter__.return_value = itertools.cycle([1, 2])

我希望mock_func完全像infinite_generator函数一样运行。

例如，我希望能够做到以下几点：

>>> a = mock_func()
>>> next(a)
1
>>> next(a)
2
>>> next(a)
1
>>> next(a)
2

等等

然而，在next(a)的那一刻next(a)返回类似的东西

<MagicMock name='mock().__next__()' id='3043937712'>

Answer 1

在这里省略__iter__ ，因为你不打算迭代mock_func对象本身：

mock_func.__iter__.return_value = itertools.cycle([1, 2])

代替：

>>> mock_func = mock.Mock()
>>> mock_func.return_value = itertools.cycle([1, 2])
>>> a = mock_func()
>>> next(a)
1
>>> next(a)
2
>>> next(a)
1
>>> next(a)
2