code: def test_train_test_split():
      tr, te = train_test_split(y, test_size=10)
      assert te.shape[0] == 10
      assert_array_equal(y, np.concatenate([tr, te]))
  
file_path: pmdarima/pmdarima/model_selection/tests/test_split.py
imports: from pmdarima.compat.pytest import pytest_error_str
  from pmdarima.model_selection import RollingForecastCV, SlidingWindowForecastCV, check_cv, train_test_split
  from pmdarima.datasets import load_wineind
  import pytest
  import numpy
  from numpy.testing import assert_array_equal
label_desc: Use image node_id 6 for calling a global function with example usage: test_train_test_split() without return types