Python shufflecycle示例

示例#1

文件： test_functional.py 项目： uploadcare/kombu

 def test_shuffles(self):
     prev_repeat, utils.repeat = utils.repeat, Mock()
     try:
         utils.repeat.return_value = list(range(10))
         values = {'A', 'B', 'C'}
         cycle = utils.shufflecycle(values)
         seen = set()
         for i in range(10):
             next(cycle)
         utils.repeat.assert_called_with(None)
         assert seen.issubset(values)
         with pytest.raises(StopIteration):
             next(cycle)
             next(cycle)
     finally:
         utils.repeat = prev_repeat

示例#2

文件： test_functional.py 项目： Erve1879/kombu

 def test_shuffles(self):
     prev_repeat, utils.repeat = utils.repeat, Mock()
     try:
         utils.repeat.return_value = list(range(10))
         values = {'A', 'B', 'C'}
         cycle = utils.shufflecycle(values)
         seen = set()
         for i in range(10):
             next(cycle)
         utils.repeat.assert_called_with(None)
         assert seen.issubset(values)
         with pytest.raises(StopIteration):
             next(cycle)
             next(cycle)
     finally:
         utils.repeat = prev_repeat