code: def test_safe_indexing(x, i, exp):
      res = sk.safe_indexing(x, i)
      if hasattr(res, "values"):  # pd.Series
          res = res.values
      assert_array_equal(exp, res)
  
file_path: pmdarima/pmdarima/compat/tests/test_sklearn.py
imports: from pmdarima.arima import ARIMA
  from pmdarima.compat.pytest import pytest_error_str
  from pmdarima.compat import sklearn
  import numpy
  import pandas
  from numpy.testing import assert_array_equal
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: test_safe_indexing(x, i, exp) without return types