Fix Marker location validation for numpy array (#1647)

python-visualization · Nov 10, 2022 · 35b2ca4 · 35b2ca4
1 parent 79f4599
commit 35b2ca4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/folium/map.py b/folium/map.py
@@ -275,7 +275,7 @@ def __init__(self, location=None, popup=None, tooltip=None, icon=None,
  draggable=False, **kwargs):
  super(Marker, self).__init__()
  self._name = 'Marker'
- self.location = validate_location(location) if location else None
+ self.location = validate_location(location) if location is not None else None
  self.options = parse_options(
  draggable=draggable or None,
  autoPan=draggable or None,

diff --git a/tests/test_map.py b/tests/test_map.py
@@ -3,7 +3,7 @@
 ----------------
 
 """
-
+import numpy as np
 import pytest
 
 from folium import Map
@@ -148,6 +148,10 @@ def test_marker_valid_location():
  m.render()
 
 
+def test_marker_numpy_array_as_location():
+ Marker(np.array([0, 0]))
+
+
 @pytest.mark.filterwarnings('ignore::UserWarning')
 def test_icon_invalid_marker_colors():
  pytest.warns(UserWarning, Icon, color='lila')