TST Fixed typo in test_column_transformer (#14128)

lrjball · jnothman · commit 0c110701f9a8 · 2019-06-20T10:43:19.000+10:00
In test_column_transformer_dataframe(), X_df2 was created but never used. X_df was mistakely being used in its place. The PR fixes that typo.
diff --git a/sklearn/compose/tests/test_column_transformer.py b/sklearn/compose/tests/test_column_transformer.py
@@ -241,8 +241,8 @@ def transform(self, X, y=None):
     X_df2 = X_df.copy()
     X_df2.columns = [1, 0]
     ct = ColumnTransformer([('trans', Trans(), 0)], remainder='drop')
-    assert_array_equal(ct.fit_transform(X_df), X_res_first)
-    assert_array_equal(ct.fit(X_df).transform(X_df), X_res_first)
+    assert_array_equal(ct.fit_transform(X_df2), X_res_first)
+    assert_array_equal(ct.fit(X_df2).transform(X_df2), X_res_first)
 
     assert len(ct.transformers_) == 2
     assert ct.transformers_[-1][0] == 'remainder'
