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