code: def test_column_drop(spec, expected):
      df = pd.DataFrame(
          {
              "x": _make_arr(na=1),
              "y": _make_arr(na=2),
              "z": _make_arr(na=3),
              "prefix_a": _make_arr(na=4),
              "prefix_b": _make_arr(na=5),
              "a_suffix": _make_arr(na=6),
              "b_suffix": _make_arr(na=7),
          }
      )
  
      selector = DataSelector({**spec, "kind": "column_drop"})
      out, _ = selector.transform(df, return_summary=True)
  
      assert set(out.columns) == expected
  
file_path: sklearn-evaluation/tests/test_DataSelector.py
imports: from pathlib import Path
  import pytest
  import numpy
  import pandas
  from sklearn_evaluation.training import DataSelector
  from sklearn_evaluation.training.selector import expand_arguments, _with_prefix
  from sklearn_evaluation.exceptions import DataSelectorError
label_desc: Use image node_id 4 for calling a global function with example usage: test_column_drop(spec, expected) without return types