code: def test_subset_1():
      """[Preprocessing | Subset]: assert correct subset."""
      assert sub.fit_transform(X).shape[1] == 2
  
imports: import numpy
  from mlens.preprocessing import Subset, Shift
  from mlens.testing.dummy import Data
label_desc: Use image node_id 1 for calling a global function with example usage: test_subset_1() without return types