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)