def test_merge_3_dims(self):
self.assertListEqual(
list(
merge_permute_order_dimensions([1, 2, 3],
int64_array([3, 0, 1, 2, 4]))),
[1, 0, 2])
def test_merge_start_indices(self):
self.assertListEqual(
list(
merge_permute_order_dimensions([0, 1],
int64_array([1, 2, 4, 3, 0]))),
[1, 3, 2, 0])
def test_merge_all_dims(self):
self.assertListEqual(
list(
merge_permute_order_dimensions([0, 1, 2],
int64_array([0, 1, 2]))), [0])
def test_merge_last_indices(self):
self.assertListEqual(
list(
merge_permute_order_dimensions([3, 4],
int64_array([0, 3, 4, 1, 2]))),
[0, 2, 3, 1])
