code: def test_split_datasets(split_dataset):
      X_train, X_test, y_train, y_test = split_dataset
  
      assert X_train.ndim == X_test.ndim == 2
      assert X_train.shape[1] == X_test.shape[1]
      assert 2 <= (len(X_train) + len(X_test)) <= 200
  
      assert y_train.ndim == y_test.ndim
      assert y_train.ndim in (0, 1, 2)
      assert (y_train.ndim == 0) or (
          2 <= (len(y_train) + len(y_test)) <= 200
      )
  
file_path: cuml/python/cuml/tests/test_strategies.py
imports: from cuml.internals.array import CumlArray
  from cuml.internals.safe_imports import cpu_only_import, gpu_only_import
  from cuml.testing.strategies import create_cuml_array_input, cuml_array_dtypes, cuml_array_input_types, cuml_array_inputs, cuml_array_orders, cuml_array_shapes, regression_datasets, split_datasets, standard_classification_datasets, standard_datasets, standard_regression_datasets
  from cuml.testing.utils import normalized_shape, series_squeezed_shape
  from hypothesis import HealthCheck, given, settings
  from hypothesis import strategies
  from hypothesis.extra.numpy import floating_dtypes, integer_dtypes
label_desc: Use image node_id 5 for calling a global function with example usage: test_split_datasets(split_dataset) without return types