<s> import numpy as np <EOL> from mlxtend . feature_selection import ColumnSelector <EOL> def test_ColumnSelector ( ) : <EOL> X1_in = np . ones ( ( <NUM_LIT:4> , <NUM_LIT:8> ) ) <EOL> X1_out = ColumnSelector ( cols = ( <NUM_LIT:1> , <NUM_LIT:3> ) ) . transform ( X1_in ) <EOL> assert ( X1_out . shape == ( <NUM_LIT:4> , <NUM_LIT:2> ) ) </s>
