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