def test_2d_input_with_sequential_dims(self):
self.assertListEqual(sequential_dims(int64_array([0, 1])), [0, 1])
def test_2d_input_without_sequential_dims(self):
self.assertIsNone(sequential_dims(int64_array([1, 0])))
def test_no_sequential_dims(self):
self.assertIsNone(sequential_dims(int64_array([2, 1, 3, 0, 4])))
def test_returns_from_the_beginning(self):
self.assertListEqual(sequential_dims(int64_array([1, 2, 3, 0, 4])),
[0, 1, 2])
def test_returns_full_list(self):
self.assertListEqual(sequential_dims(int64_array([0, 1, 2, 3, 4])),
[0, 1, 2, 3, 4])
def test_returns_last_indices(self):
self.assertListEqual(sequential_dims(int64_array([4, 0, 3, 1, 2])),
[3, 4])
def test_returns_first_instance(self):
self.assertListEqual(sequential_dims(int64_array([0, 3, 4, 1, 2])),
[1, 2])
