def test_rechunk_internals_1(): """ Test the cumdims_label and _breakpoints and _intersect_1d internal funcs to rechunk.""" new = cumdims_label(((1,1,2),(1,5,1)),'n') old = cumdims_label(((4, ),(1,) * 5),'o') breaks = tuple(_breakpoints(o, n) for o, n in zip(old, new)) answer = (('o', 0), ('n', 0), ('n', 1), ('n', 2), ('o', 4), ('n', 4)) assert breaks[0] == answer answer2 = (('o', 0), ('n', 0), ('o', 1), ('n', 1), ('o', 2), ('o', 3), ('o', 4), ('o', 5), ('n', 6), ('n', 7)) assert breaks[1] == answer2 i1d = [_intersect_1d(b) for b in breaks] answer3 = (((0, slice(0, 1, None)),), ((0, slice(1, 2, None)),), ((0, slice(2, 4, None)),)) assert i1d[0] == answer3 answer4 = (((0, slice(0, 1, None)),), ((1, slice(0, 1, None)), (2, slice(0, 1, None)), (3, slice(0, 1, None)), (4, slice(0, 1, None)), (5, slice(0, 1, None))), ((5, slice(1, 2, None)),)) assert i1d[1] == answer4
def test_rechunk_internals_1(): """ Test the cumdims_label and _breakpoints and _intersect_1d internal funcs to rechunk.""" new = cumdims_label(((1, 1, 2), (1, 5, 1)), "n") old = cumdims_label(((4,), (1,) * 5), "o") breaks = tuple(_breakpoints(o, n) for o, n in zip(old, new)) answer = (("o", 0), ("n", 0), ("n", 1), ("n", 2), ("o", 4), ("n", 4)) assert breaks[0] == answer answer2 = (("o", 0), ("n", 0), ("o", 1), ("n", 1), ("o", 2), ("o", 3), ("o", 4), ("o", 5), ("n", 6), ("n", 7)) assert breaks[1] == answer2 i1d = [_intersect_1d(b) for b in breaks] answer3 = (((0, slice(0, 1, None)),), ((0, slice(1, 2, None)),), ((0, slice(2, 4, None)),)) assert i1d[0] == answer3 answer4 = ( ((0, slice(0, 1, None)),), ( (1, slice(0, 1, None)), (2, slice(0, 1, None)), (3, slice(0, 1, None)), (4, slice(0, 1, None)), (5, slice(0, 1, None)), ), ((5, slice(1, 2, None)),), ) assert i1d[1] == answer4
def test_rechunk_internals_1(): """ Test the cumdims_label and _breakpoints and _intersect_1d internal funcs to rechunk.""" new = cumdims_label(((1, 1, 2), (1, 5, 1)), "n") old = cumdims_label(((4, ), (1, ) * 5), "o") breaks = tuple(_breakpoints(o, n) for o, n in zip(old, new)) answer = (("o", 0), ("n", 0), ("n", 1), ("n", 2), ("o", 4), ("n", 4)) assert breaks[0] == answer answer2 = ( ("o", 0), ("n", 0), ("o", 1), ("n", 1), ("o", 2), ("o", 3), ("o", 4), ("o", 5), ("n", 6), ("n", 7), ) assert breaks[1] == answer2 i1d = [_intersect_1d(b) for b in breaks] answer3 = [[(0, slice(0, 1))], [(0, slice(1, 2))], [(0, slice(2, 4))]] assert i1d[0] == answer3 answer4 = [ [(0, slice(0, 1))], [ (1, slice(0, 1)), (2, slice(0, 1)), (3, slice(0, 1)), (4, slice(0, 1)), (5, slice(0, 1)), ], [(5, slice(1, 2))], ] assert i1d[1] == answer4
def test_rechunk_internals_1(): """ Test the cumdims_label and _breakpoints and _intersect_1d internal funcs to rechunk.""" new = cumdims_label(((1,1,2),(1,5,1)),'n') old = cumdims_label(((4, ),(1,) * 5),'o') breaks = tuple(_breakpoints(o, n) for o, n in zip(old, new)) answer = (('o', 0), ('n', 0), ('n', 1), ('n', 2), ('o', 4), ('n', 4)) assert breaks[0] == answer answer2 = (('o', 0), ('n', 0), ('o', 1), ('n', 1), ('o', 2), ('o', 3), ('o', 4), ('o', 5), ('n', 6), ('n', 7)) assert breaks[1] == answer2 i1d = [_intersect_1d(b) for b in breaks] answer3 = [[(0, slice(0, 1))], [(0, slice(1, 2))], [(0, slice(2, 4))]] assert i1d[0] == answer3 answer4 = [[(0, slice(0, 1))], [(1, slice(0, 1)), (2, slice(0, 1)), (3, slice(0, 1)), (4, slice(0, 1)), (5, slice(0, 1))], [(5, slice(1, 2))]] assert i1d[1] == answer4