code: def test_column_selector_nd_should_transform_with_column_selector_2d():
      step = ColumnsSelectorND(0, n_dimension=2)
      data_inputs, expected_outputs = _create_data_source((20, 20))
  
      outputs = step.transform(data_inputs)
  
      assert np.array_equal(
          outputs, np.expand_dims(data_inputs[..., 0], axis=-1)
      )
  
file_path: Neuraxle/testing_neuraxle/steps/test_column_selector_2d.py
imports: import numpy
  import pytest
  from neuraxle.steps.column_transformer import ColumnSelector2D, ColumnsSelectorND, NumpyColumnSelector2D
label_desc: Use image node_id 5 for calling a global function with example usage: test_column_selector_nd_should_transform_with_column_selector_2d() without return types