code: def test_data_arrow_head():
      """Create sample data for univariate classification."""
      X_train, y_train = load_arrow_head(split="TRAIN")
      X_test, y_test = load_arrow_head(split="TEST")
      return y_train.astype(int), y_test.astype(int), X_train, X_test
  
imports: import os
  import sys
  from pathlib import Path
  from unittest import mock
  import numpy
  import pandas
  import pytest
  from sktime.datasets import load_airline, load_arrow_head, load_longley
  from sktime.forecasting.arima import AutoARIMA
  from sktime.forecasting.naive import NaiveForecaster
  from sktime.split import temporal_train_test_split
  from sktime.utils.multiindex import flatten_multiindex
  from sktime.utils.validation._dependencies import _check_soft_dependencies
label_desc: Use image node_id 6 for calling a global function with example usage: test_data_arrow_head() and returns: y_train,  y_test,  X_train,  X_test