def test_simpleUseCases(self):
self.assertEqual(list(iterage.take(iterage.cycle(range(2)), 6)),
[0, 1, 0, 1, 0, 1])
self.assertEqual(list(iterage.take(iterage.cycle([0, 1]), 6)),
[0, 1, 0, 1, 0, 1])
self.assertEqual(
list(iterage.take(iterage.cycle(createNGenerator(2)), 6)),
[0, 1, 0, 1, 0, 1])
def test_emptyRange(self):
self.assertEqual(list(iterage.take(range(0), 0)), [])
self.assertEqual(list(iterage.take([], 0)), [])
self.assertEqual(list(iterage.take(emptyGenerator, 0)), [])
self.assertEqual(list(iterage.take(range(0), 3)), [])
self.assertEqual(list(iterage.take([], 3)), [])
self.assertEqual(list(iterage.take(emptyGenerator, 3)), [])
self.assertEqual(list(iterage.take(range(3), 0)), [])
self.assertEqual(list(iterage.take([0, 1, 2], 0)), [])
self.assertEqual(list(iterage.take(createNGenerator(3), 0)), [])
def test_simpleUseCases(self): self.assertEqual(list(iterage.take(iterage.cycle(range(2)), 6)), [0, 1, 0, 1, 0, 1]) self.assertEqual(list(iterage.take(iterage.cycle([0, 1]), 6)), [0, 1, 0, 1, 0, 1]) self.assertEqual(list(iterage.take(iterage.cycle(createNGenerator(2)), 6)), [0, 1, 0, 1, 0, 1])
def test_simpleUseCases(self):
for i in range(6):
exp = list(range(i)) if i < 3 else list(range(3))
self.assertEqual(list(iterage.take(range(i), 3)), exp)
self.assertEqual(list(iterage.take(list(range(i)), 3)), exp)
self.assertEqual(list(iterage.take(createNGenerator(i), 3)), exp)
