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)
