类似于this question ,但其中的答案不足以满足我正在做的事情。
我正在尝试测试这样的方法:
import mock
def stack_overflow_desired_output():
print_a_few_times(['upvote', 'this', 'question!'])
def stack_overflow_mocked():
the_mock = mock.Mock()
the_mock.__iter__ = mock.Mock(return_value=iter(["upvote", "this", "question"]))
print_a_few_times(the_mock)
def print_a_few_times(fancy_object):
for x in [1, 2, 3]:
for y in fancy_object:
print("{}.{}".format(x, y))
当我调用 stack_overflow_desired_output()
时,我得到了这个:
1.upvote
1.this
1.question!
2.upvote
2.this
2.question!
3.upvote
3.this
3.question!
但是当我调用 stack_overflow_mocked()
时,我只得到这个:
1.upvote
1.this
1.question!
有没有办法让迭代器在 for 循环结束时用尽时重置自身?将重置放在 print_a_few_times
函数中,as described in the aforementioned question , 将是侵入性的。
最佳答案
将模拟对象包裹在实际列表的 __iter__
方法周围。
def stack_overflow_mocked():
the_mock = mock.Mock()
the_mock.__iter__ = mock.Mock(wraps=["upvote", "this", "question"].__iter__)
print_a_few_times(the_mock)
关于python - 如何在 for 循环中重置模拟迭代器?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45494371/