Example #1
0
 def test_shuffles(self):
     prev_repeat, utils.repeat = utils.repeat, Mock()
     try:
         utils.repeat.return_value = list(range(10))
         values = set(['A', 'B', 'C'])
         cycle = utils.shufflecycle(values)
         seen = set()
         for i in range(10):
             next(cycle)
         utils.repeat.assert_called_with(None)
         self.assertTrue(seen.issubset(values))
         with self.assertRaises(StopIteration):
             next(cycle)
             next(cycle)
     finally:
         utils.repeat = prev_repeat
Example #2
0
 def test_shuffles(self):
     prev_repeat, utils.repeat = utils.repeat, Mock()
     try:
         utils.repeat.return_value = list(range(10))
         values = set(['A', 'B', 'C'])
         cycle = utils.shufflecycle(values)
         seen = set()
         for i in range(10):
             next(cycle)
         utils.repeat.assert_called_with(None)
         self.assertTrue(seen.issubset(values))
         with self.assertRaises(StopIteration):
             next(cycle)
             next(cycle)
     finally:
         utils.repeat = prev_repeat