Fix flake8

Author: dukebody

tests/test_dataframe_mapper.py

@@ -17,7 +17,8 @@
 from sklearn.pipeline import Pipeline
 from sklearn.svm import SVC
 from sklearn.feature_extraction.text import CountVectorizer
-from sklearn.preprocessing import Imputer, StandardScaler, OneHotEncoder, LabelBinarizer
+from sklearn.preprocessing import (
+    Imputer, StandardScaler, OneHotEncoder, LabelBinarizer)
 from sklearn.feature_selection import SelectKBest, chi2
 from sklearn.base import BaseEstimator, TransformerMixin
 import sklearn.decomposition
@@ -94,7 +95,9 @@ def test_complex_df(complex_dataframe):
     Get a dataframe from a complex mapped dataframe
     """
     df = complex_dataframe
-    mapper = DataFrameMapper([('target', None), ('feat1', None), ('feat2', None)], df_out=True)
+    mapper = DataFrameMapper(
+        [('target', None), ('feat1', None), ('feat2', None)],
+        df_out=True)
     transformed = mapper.fit_transform(df)
     assert len(transformed) == len(complex_dataframe)
     for c in df.columns:
@@ -145,7 +148,9 @@ def test_pca(complex_dataframe):
     Check multi in and out with PCA
     """
     df = complex_dataframe
-    mapper = DataFrameMapper([(['feat1', 'feat2'], sklearn.decomposition.PCA(2))], df_out=True)
+    mapper = DataFrameMapper(
+        [(['feat1', 'feat2'], sklearn.decomposition.PCA(2))],
+        df_out=True)
     transformed = mapper.fit_transform(df)
     cols = transformed.columns
     assert len(cols) == 2