def test_emptyRange(self): self.assertEqual(list(iterage.take(iterage.irange(0), 0)), []) self.assertEqual(list(iterage.take([], 0)), []) self.assertEqual(list(iterage.take(emptyGenerator, 0)), []) self.assertEqual(list(iterage.take(iterage.irange(0), 3)), []) self.assertEqual(list(iterage.take([], 3)), []) self.assertEqual(list(iterage.take(emptyGenerator, 3)), []) self.assertEqual(list(iterage.take(iterage.irange(3), 0)), []) self.assertEqual(list(iterage.take([0, 1, 2], 0)), []) self.assertEqual(list(iterage.take(createNGenerator(3), 0)), [])
def test_unsuitableRange(self): self.assertEqual(list(iterage.chunk(iterage.irange(8), 3)), [(0, 1, 2), (3, 4, 5), (6, 7)]) self.assertEqual(list(iterage.chunk(iterage.irange(1), 3)), [(0,)])
def test_emptyRange(self): self.assertEqual(list(iterage.chunk(iterage.irange(0), 3)), [])
def test_simpleUseCases(self): self.assertEqual(list(iterage.chunk(iterage.irange(9), 3)), [(0, 1, 2), (3, 4, 5), (6, 7, 8)]) self.assertEqual(list(iterage.chunk(iterage.irange(3), 3)), [(0, 1, 2)])
def test_simpleUseCases(self): self.assertEqual(list(iterage.take(iterage.cycle(iterage.irange(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 iterage.irange(6): exp = list(iterage.irange(i)) if i < 3 else list(iterage.irange(3)) self.assertEqual(list(iterage.take(iterage.irange(i), 3)), exp) self.assertEqual(list(iterage.take(list(iterage.irange(i)), 3)), exp) self.assertEqual(list(iterage.take(createNGenerator(i), 3)), exp)
def test_emptyRange(self): self.assertEqual(list(iterage.drop(iterage.irange(6), 0)), list(iterage.irange(6))) self.assertEqual(list(iterage.drop([], 3)), []) self.assertEqual(list(iterage.drop([], 0)), [])
def createNGenerator(n): return (x for x in iterage.irange(n))