Add test for several location arg data types.

Author: dskkato

tests/test_folium.py

@@ -19,6 +19,7 @@
 import jinja2
 from jinja2 import Environment, PackageLoader
 
+import numpy as np
 import pandas as pd
 
 import pytest
@@ -62,6 +63,17 @@ def test_get_templates():
     assert isinstance(env, jinja2.environment.Environment)
 
 
+def test_location_args():
+    """Test some data types for a location arg."""
+    location = np.array([45.5236, -122.6750])
+    m = folium.Map(location)
+    assert m.location == [45.5236, -122.6750]
+
+    df = pd.DataFrame({"location": [45.5236, -122.6750]})
+    m = folium.Map(df["location"])
+    assert m.location == [45.5236, -122.6750]
+
+
 class TestFolium(object):
     """Test class for the Folium library."""
 
@@ -108,9 +120,9 @@ def test_init(self):
                     'name': 'TileLayer',
                     'id': '00000000000000000000000000000000',
                     'children': {}
-                    }
                 }
             }
+        }
 
     def test_cloudmade(self):
         """Test cloudmade tiles and the API key."""
@@ -197,7 +209,8 @@ def test_map_build(self):
 
         # Standard map.
         self.setup()
-        rendered = [line.strip() for line in self.m._parent.render().splitlines() if line.strip()]
+        rendered = [line.strip()
+                    for line in self.m._parent.render().splitlines() if line.strip()]
 
         html_templ = self.env.get_template('fol_template.html')
         attr = 'http://openstreetmap.org'
@@ -332,7 +345,7 @@ def test_fit_bounds(self):
                                               'padding': (3, 3), },
                                              sort_keys=True),
             'this': fitbounds,
-            })
+        })
 
         assert ''.join(fit_bounds_rendered.split()) in ''.join(out.split())