11"""
22Test the ColumnTransformer.
33"""
4- import re
54
65import numpy as np
76from scipy import sparse
@@ -597,8 +596,7 @@ def test_column_transformer_get_set_params():
597596 'trans2__with_mean' : True ,
598597 'trans2__with_std' : True ,
599598 'transformers' : ct .transformers ,
600- 'transformer_weights' : None ,
601- 'verbose' : False }
599+ 'transformer_weights' : None }
602600
603601 assert_dict_equal (ct .get_params (), exp )
604602
@@ -615,8 +613,7 @@ def test_column_transformer_get_set_params():
615613 'trans2__with_mean' : True ,
616614 'trans2__with_std' : True ,
617615 'transformers' : ct .transformers ,
618- 'transformer_weights' : None ,
619- 'verbose' : False }
616+ 'transformer_weights' : None }
620617
621618 assert_dict_equal (ct .get_params (), exp )
622619
@@ -947,8 +944,7 @@ def test_column_transformer_get_set_params_with_remainder():
947944 'trans1__with_mean' : True ,
948945 'trans1__with_std' : True ,
949946 'transformers' : ct .transformers ,
950- 'transformer_weights' : None ,
951- 'verbose' : False }
947+ 'transformer_weights' : None }
952948
953949 assert ct .get_params () == exp
954950
@@ -964,8 +960,7 @@ def test_column_transformer_get_set_params_with_remainder():
964960 'sparse_threshold' : 0.3 ,
965961 'trans1' : 'passthrough' ,
966962 'transformers' : ct .transformers ,
967- 'transformer_weights' : None ,
968- 'verbose' : False }
963+ 'transformer_weights' : None }
969964
970965 assert ct .get_params () == exp
971966
@@ -986,56 +981,6 @@ def test_column_transformer_no_estimators():
986981 assert ct .transformers_ [- 1 ][2 ] == [0 , 1 , 2 ]
987982
988983
989- @pytest .mark .parametrize (
990- ['est' , 'pattern' ],
991- [(ColumnTransformer ([('trans1' , Trans (), [0 ]), ('trans2' , Trans (), [1 ])],
992- remainder = DoubleTrans ()),
993- (r'\[ColumnTransformer\].*\(1 of 3\) Processing trans1.* total=.*\n'
994- r'\[ColumnTransformer\].*\(2 of 3\) Processing trans2.* total=.*\n'
995- r'\[ColumnTransformer\].*\(3 of 3\) Processing remainder.* total=.*\n$'
996- )),
997- (ColumnTransformer ([('trans1' , Trans (), [0 ]), ('trans2' , Trans (), [1 ])],
998- remainder = 'passthrough' ),
999- (r'\[ColumnTransformer\].*\(1 of 3\) Processing trans1.* total=.*\n'
1000- r'\[ColumnTransformer\].*\(2 of 3\) Processing trans2.* total=.*\n'
1001- r'\[ColumnTransformer\].*\(3 of 3\) Processing remainder.* total=.*\n$'
1002- )),
1003- (ColumnTransformer ([('trans1' , Trans (), [0 ]), ('trans2' , 'drop' , [1 ])],
1004- remainder = 'passthrough' ),
1005- (r'\[ColumnTransformer\].*\(1 of 2\) Processing trans1.* total=.*\n'
1006- r'\[ColumnTransformer\].*\(2 of 2\) Processing remainder.* total=.*\n$'
1007- )),
1008- (ColumnTransformer ([('trans1' , Trans (), [0 ]),
1009- ('trans2' , 'passthrough' , [1 ])],
1010- remainder = 'passthrough' ),
1011- (r'\[ColumnTransformer\].*\(1 of 3\) Processing trans1.* total=.*\n'
1012- r'\[ColumnTransformer\].*\(2 of 3\) Processing trans2.* total=.*\n'
1013- r'\[ColumnTransformer\].*\(3 of 3\) Processing remainder.* total=.*\n$'
1014- )),
1015- (ColumnTransformer ([('trans1' , Trans (), [0 ])], remainder = 'passthrough' ),
1016- (r'\[ColumnTransformer\].*\(1 of 2\) Processing trans1.* total=.*\n'
1017- r'\[ColumnTransformer\].*\(2 of 2\) Processing remainder.* total=.*\n$'
1018- )),
1019- (ColumnTransformer ([('trans1' , Trans (), [0 ]), ('trans2' , Trans (), [1 ])],
1020- remainder = 'drop' ),
1021- (r'\[ColumnTransformer\].*\(1 of 2\) Processing trans1.* total=.*\n'
1022- r'\[ColumnTransformer\].*\(2 of 2\) Processing trans2.* total=.*\n$' )),
1023- (ColumnTransformer ([('trans1' , Trans (), [0 ])], remainder = 'drop' ),
1024- (r'\[ColumnTransformer\].*\(1 of 1\) Processing trans1.* total=.*\n$' ))])
1025- @pytest .mark .parametrize ('method' , ['fit' , 'fit_transform' ])
1026- def test_column_transformer_verbose (est , pattern , method , capsys ):
1027- X_array = np .array ([[0 , 1 , 2 ], [2 , 4 , 6 ], [8 , 6 , 4 ]]).T
1028-
1029- func = getattr (est , method )
1030- est .set_params (verbose = False )
1031- func (X_array )
1032- assert not capsys .readouterr ().out , 'Got output for verbose=False'
1033-
1034- est .set_params (verbose = True )
1035- func (X_array )
1036- assert re .match (pattern , capsys .readouterr ()[0 ])
1037-
1038-
1039984def test_column_transformer_no_estimators_set_params ():
1040985 ct = ColumnTransformer ([]).set_params (n_jobs = 2 )
1041986 assert ct .n_jobs == 2
0 commit comments