python - 从“开始到结束”“一遍又一遍”进行副作用迭代器循环

Question

我必须说我是python mock的新手。我有一个副作用迭代器：

myClass.do.side_effect = iter([processStatus, memoryStatus, processStatus, memoryStatus, processStatus, memoryStatus, processStatus, memoryStatus])

以上按预期工作，测试用例通过

但我正在寻找一种更好的方法来写这个。我试过[....]*4了，但没有奏效。

我该怎么做？简单地说，使迭代器在它结束时从头开始。

Answer 1

我想你可以itertools.cycle在这里使用，如果你想“一遍又一遍”：

>>> s = range(3)
>>> s
[0, 1, 2]
>>> from itertools import cycle
>>> c = cycle(s)
>>> c
<itertools.cycle object at 0xb72697cc>
>>> [next(c) for i in range(10)]
[0, 1, 2, 0, 1, 2, 0, 1, 2, 0]
>>> c = cycle(['pS', 'mS'])
>>> [next(c) for i in range(10)]
['pS', 'mS', 'pS', 'mS', 'pS', 'mS', 'pS', 'mS', 'pS', 'mS']

或者，正如@mgilson 所说，如果您想要有限数量的 2 元素项（我不完全确定您需要什么数据格式）：

>>> from itertools import repeat
>>> repeat([2,3], 3)
repeat([2, 3], 3)
>>> list(repeat([2,3], 3))
[[2, 3], [2, 3], [2, 3]]

但正如评论中所指出的，iter([1,2,3]*n)也应该有效。