code: def test_split_column():
      data = cudf.DataFrame(
          {
              "x": range(100),
              "y": ([0] * (100 // 2)) + ([1] * (100 // 2)),
          }
      )
      train_size = 0.8
  
      X_train, X_test, y_train, y_test = train_test_split(
          data, "y", train_size=train_size
      )
  
      assert (
          len(X_train)
          == len(y_train)
          == pytest.approx(train_size * len(data))
      )
      assert (
          len(X_test)
          == len(y_test)
          == pytest.approx((1 - train_size) * len(data))
      )
  
      X_reconstructed = cudf.concat([X_train, X_test]).sort_values(
          by=["x"]
      )
      y_reconstructed = y_train.append(y_test).sort_values()
  
      assert all(
          data
          == X_reconstructed.assign(y=y_reconstructed).reset_index(
              drop=True
          )
      )
  
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 3 for calling a global function with example usage: test_split_column() without return types