fix formatting

Author: alessioarena

folium/features.py

@@ -1064,9 +1064,10 @@ def __init__(self, geo_data, data=None, columns=None, key_on=None,  # noqa
         if hasattr(data, 'to_dict'):
             # This is a pd.Dataframe or pd.Series
             # check for numbers in future color_data keys
-            future_keys = data[columns[0]] if columns is not None else data.index
-            any_numeric_key = any([isinstance(e, (int, float, np.floating, np.integer))
-                                    for e in future_keys])
+            future_keys = data[columns[0]] if columns is not None else data.index  # noqa
+            any_numeric_key = any([isinstance(e,
+                                   (int, float, np.floating, np.integer))
+                                   for e in future_keys])
 
             if hasattr(data, 'set_index'):
             # This is a pd.DataFrame

tests/test_folium.py

@@ -255,16 +255,16 @@ def test_choropleth_geopandas_numeric(self):
         """Test to make sure that Choropleth function does complete the lookup
         between a GeoJSON generated from a GeoDataFrame and data from the GeoDataFrame itself.
 
-        key_on field is dtype = str, while column 0 is dtype = int 
+        key_on field is dtype = str, while column 0 is dtype = int
         All geometries have matching values (no nan_fill_color allowed)
         """
         self.setup()
 
         with open(os.path.join(rootpath, 'geo_grid.json')) as f:
             geo_data = json.load(f)
-        
+
         geo_data_frame = gpd.GeoDataFrame.from_features(geo_data['features'])
-        geo_data_frame.crs = {'init':'epsg:4326'}
+        geo_data_frame.crs = {'init': 'epsg:4326'}
         fill_color = 'BuPu'
         key_on = 'feature.id'
 
@@ -294,9 +294,9 @@ def test_choropleth_geopandas_mixed(self):
 
         with open(os.path.join(rootpath, 'geo_grid.json')) as f:
             geo_data = json.load(f)
-        
+
         geo_data_frame = gpd.GeoDataFrame.from_features(geo_data['features'])
-        geo_data_frame.crs = {'init':'epsg:4326'}
+        geo_data_frame.crs = {'init': 'epsg:4326'}
         data = pd.DataFrame({'idx': {'0': 0, '1': '1', '2': 2, '3': 3, '4': 4, '5': 5},
                              'value': {'0': 78.0, '1': 39.0, '2': 0.0, '3': 81.0, '4': 42.0, '5': 68.0}})
         fill_color = 'BuPu'
@@ -330,9 +330,9 @@ def test_choropleth_geopandas_str(self):
 
         with open(os.path.join(rootpath, 'geo_grid.json')) as f:
             geo_data = json.load(f)
-        
+
         geo_data_frame = gpd.GeoDataFrame.from_features(geo_data['features'])
-        geo_data_frame.crs = {'init':'epsg:4326'}
+        geo_data_frame.crs = {'init': 'epsg:4326'}
         data = pd.DataFrame({'idx': {'0': '0', '1': '1', '2': '2', '3': '3', '4': '4', '5': '5'},
                              'value': {'0': 78.0, '1': 39.0, '2': 0.0, '3': 81.0, '4': 42.0, '5': 68.0}})
         fill_color = 'BuPu'