def test_bad_agg_size(self): rows = [1,2,3,4] self.assertRaises(ValueError, iter_chunks, rows, 0) self.assertRaises(ValueError, iter_chunks, rows, -1) try: for i in iter_chunks(rows, 0): pass except ValueError: pass else: self.fail() try: result = list(iter_chunks(rows, 0)) except ValueError: pass else: self.fail()
def test_bad_agg_size(self): rows = [1, 2, 3, 4] self.assertRaises(ValueError, iter_chunks, rows, 0) self.assertRaises(ValueError, iter_chunks, rows, -1) try: for i in iter_chunks(rows, 0): pass except ValueError: pass else: self.fail() try: result = list(iter_chunks(rows, 0)) except ValueError: pass else: self.fail()
def test_multi_agg(self): rows = [[1],[2],[3],[4],[5]] result = list(iter_chunks(rows, 2)) self.assertEqual(result, [[[1],[2]],[[3],[4]],[[5]]])
def test_size(self): rows = [[1],[2]] result = list(iter_chunks(rows, 2)) self.assertEqual(result, [[[1],[2]]])
def test_empty(self): rows = [] result = list(iter_chunks(rows)) self.assertEqual(result, [])
def test_multi_agg(self): rows = [[1], [2], [3], [4], [5]] result = list(iter_chunks(rows, 2)) self.assertEqual(result, [[[1], [2]], [[3], [4]], [[5]]])
def test_size(self): rows = [[1], [2]] result = list(iter_chunks(rows, 2)) self.assertEqual(result, [[[1], [2]]])