def test_multi_dimensions(self):
old = ((5, 5, 5), (4, 4, 4), (7, 7, 7))
self.assertEqual(((4, 4, 4, 3), (4, 4, 4), (6, 2, 6, 6, 1)),
calprocs_dask._align_chunks(old, {
0: 4,
2: 2
}))
def test_align_1(self):
old = ((1, 2, 3, 4, 1), )
self.assertEqual(old, calprocs_dask._align_chunks(old, {0: 1}))
def test_already_aligned(self):
old = ((5, 10, 15, 5), )
self.assertEqual(old, calprocs_dask._align_chunks(old, {0: 5}))
def test_small_chunks(self):
old = ((1, ) * 12, )
self.assertEqual(((5, 5, 2), ),
calprocs_dask._align_chunks(old, {0: 5}))
def test_simple(self):
old = ((15, 18, 15, 16, 8), ) # Boundaries: 0, 15, 33, 48, 64, 72
self.assertEqual(((12, 4, 16, 4, 12, 16, 8), ),
calprocs_dask._align_chunks(old, {0: 4}))