code: def test_data_generator_split():
      X_train, y_train, X_test, y_test = datagen.gen_data(
          "blobs",
          "numpy",
          n_samples=100,
          n_features=10,
          test_fraction=0.20,
      )
      assert X_train.shape == (100, 10)
      assert X_test.shape == (25, 10)
  
imports: from cuml.benchmark.bench_helper_funcs import fit, fit_predict
  import time
  from sklearn import metrics
  from cuml.internals.safe_imports import gpu_only_import_from
  import pytest
  from cuml.internals.safe_imports import gpu_only_import
  from cuml.benchmark import datagen, algorithms
  from cuml.benchmark.bench_helper_funcs import _training_data_to_numpy
  from cuml.benchmark.runners import AccuracyComparisonRunner, SpeedupComparisonRunner, run_variations
  from cuml.internals.import_utils import has_umap
  from cuml.internals.import_utils import has_xgboost
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 3 for calling a global function with example usage: test_data_generator_split() without return types