code: def test_split_size_mismatch():
      X = cudf.DataFrame({"x": range(3)})
      y = cudf.Series([0, 1])
  
      with pytest.raises(ValueError):
          train_test_split(X, y)
  
imports: from cuml.internals.safe_imports import gpu_only_import_from
  from cuml.datasets import make_classification
  from cuml.model_selection import train_test_split
  import pytest
  from cuml.internals.safe_imports import cpu_only_import
  from cuml.internals.safe_imports import gpu_only_import
label_desc: Use image node_id 4 for calling a global function with example usage: test_split_size_mismatch() without return types