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