Merge pull request #1 from JoshuaC3/clean_names-multiindex

Fix formatting errors.
pyjanitor-devs · Mar 22, 2018 · d292b05 · d292b05
2 parents f98934a + b401d8b
commit d292b05
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/janitor/functions.py b/janitor/functions.py
@@ -32,8 +32,8 @@ def clean_names(df):
     :returns: A pandas DataFrame.
     """
     df = df.rename(
-    	columns=lambda x: x.lower()
-	                       .replace(' ', '_')
+        columns=lambda x: x.lower()
+                           .replace(' ', '_')
                            .replace('/', '_')
                            .replace(':', '_')
                            .replace("'", '')
@@ -45,7 +45,7 @@ def clean_names(df):
                            .replace(')', '_')
                            .replace('.', '_')
     )
-    
+
     df = df.rename(columns=lambda x: re.sub('_+', '_', x))
     return df
 

diff --git a/tests/test_functions.py b/tests/test_functions.py
@@ -128,7 +128,7 @@ def test_multiindex_clean_names_functional(multiindex_dataframe):
     ]
 
     labels = [[1, 0, 2], [1, 0, 2]]
-    
+
     expected_columns = pd.MultiIndex(levels=levels, labels=labels)
     assert set(df.columns) == set(expected_columns)
 
@@ -142,7 +142,7 @@ def test_multiindex_clean_names_method_chain(multiindex_dataframe):
     ]
 
     labels = [[0, 1, 2], [0, 1, 2]]
-    
+
     expected_columns = pd.MultiIndex(levels=levels, labels=labels)
     assert set(df.columns) == set(expected_columns)
 
@@ -156,7 +156,6 @@ def test_multiindex_clean_names_pipe(multiindex_dataframe):
     ]
 
     labels = [[0, 1, 2], [0, 1, 2]]
-    
+
     expected_columns = pd.MultiIndex(levels=levels, labels=labels)
     assert set(df.columns) == set(expected_columns)
-