code: def test_split_3d_data():
      X_3d = np.arange(1.0, 5001.0).reshape((100, 10, 5))
      y_3d = np.arange(1.0, 101.0).reshape(100, 1)
  
      r = dask_ml.model_selection.train_test_split(X_3d, y_3d)
      X_train, X_test, y_train, y_test = r
  
      assert X_train.ndim == X_3d.ndim
      assert X_train.shape[1:] == X_3d.shape[1:]
  
imports: import dask.array
  import dask.dataframe
  import numpy
  import pandas
  import pytest
  from sklearn.datasets import fetch_20newsgroups, make_regression
  import dask_ml.model_selection
  from dask_ml._compat import DASK_2130
label_desc: Use image node_id 16 for calling a global function with example usage: test_split_3d_data() without return types