材料模拟python_用Python模拟无限生成器

我正在尝试使用模拟库模拟无限生成器函数. (如果您使用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)返回类似

< MagicMock名称='mock().__ next __()'id ='3043937712'>

解决方法:

在这里省略__iter__,因为您不打算遍历嘲笑对象本身：

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

标签：python-unittest,unit-testing,mocking,python-3-x,python

来源： https://codeday.me/bug/20191121/2053409.html