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))
