def test_consume(self): it = iter(range(10)) iterators.consume(it, 3) self.assertTrue(next(it) == 3) iterators.consume(it) self.assertRaises(StopIteration, next, it)
def test_pick(self): it = iterators.pick(range(10)) iterators.consume(it, 10) picks = [next(it)] * 10 self.assertTrue(picks.count(9) == 10)