Add tests for plugins

Author: Martin Journois

folium/plugins/__init__.py

@@ -4,4 +4,5 @@
 from .scroll_zoom_toggler import ScrollZoomToggler
 from .terminator import Terminator
 from .boat_marker import BoatMarker
-from .layer import Layer, LayerControl
+from .layer import Layer, LayerControl
+from .geo_json import GeoJson

folium/plugins/boat_marker.py

@@ -37,7 +37,7 @@ def __init__(self, position=None, heading=0, wind_heading=None, wind_speed=0, **
     def render_header(self, nb):
         """Generates the HTML part of the plugin."""
         return """
-            <script src="https://tomaszbrue.github.io/leaflet.boatmarker/js/leaflet.boatmarker.min.js"></script>
+            <script src="https://thomasbrueggemann.github.io/leaflet.boatmarker/js/leaflet.boatmarker.min.js"></script>
             """  if nb==0 else ""
 
     def render_js(self, nb):

folium/plugins/geojson.py renamed to folium/plugins/geo_json.py

@@ -42,4 +42,4 @@ def render_js(self, nb):
         out = """
         var geojson_{nb} = L.geoJson({data}).addTo(map);
         """.format(nb=nb, data = self.data)
-        return out
+        return out

folium/plugins/layer.py

@@ -4,10 +4,10 @@
 
 class Layer(Plugin):
     """Adds a layer to the map."""
-    def __init__(self, layer_name = None, tile_url=None, min_zoom=1, max_zoom=18, attribution=''):
+    def __init__(self, url=None, layer_name = None, min_zoom=1, max_zoom=18, attribution=''):
         super(Layer, self).__init__()
         self.plugin_name = 'Layer'
-        self.tile_url = tile_url if tile_url is not None else '//otile1.mqcdn.com/tiles/1.0.0/osm/{z}/{x}/{y}.png'
+        self.tile_url = url
         self.attribution = attribution
         self.min_zoom = min_zoom
         self.max_zoom = max_zoom

tests/folium_tests.py

@@ -14,7 +14,7 @@
 import folium
 from folium.six import PY3
 from folium.plugins import ScrollZoomToggler, MarkerCluster
-
+from test_plugins import testPlugins
 
 def setup_data():
     '''Import economic data for testing'''

tests/test_plugins.py

@@ -0,0 +1,106 @@
+# -*- coding: utf-8 -*-
+'''
+Folium Plugins Tests
+--------------------
+
+'''
+import folium
+from folium import plugins
+import numpy as np
+import json
+
+class testPlugins(object):
+    '''Test class for Folium plugins'''
+
+    def test_scroll_zoom_toggler(self):
+        mapa = folium.Map([45.,3.], zoom_start=4)
+        mapa.add_plugin(plugins.ScrollZoomToggler())
+        mapa._build_map()
+
+        
+    def test_marker_cluster(self):
+        N = 100
+        data = np.array([
+            np.random.uniform(low=35,high=60, size=N),   # random latitudes in Europe
+            np.random.uniform(low=-12,high=30, size=N),  # random longitudes in Europe
+            range(N),                                    # popups are simple numbers 
+            ]).T
+        mapa = folium.Map([45.,3.], zoom_start=4)
+        mapa.add_plugin(plugins.MarkerCluster(data))
+        mapa._build_map()
+
+    def test_terminator(self):
+        mapa = folium.Map([45.,3.], zoom_start=1)
+        mapa.add_plugin(plugins.Terminator())
+        mapa.add_plugin(plugins.ScrollZoomToggler())
+        mapa._build_map()
+
+    def test_boat_marker(self):
+        mapa = folium.Map([30.,0.], zoom_start=3)
+        mapa.add_plugin(plugins.BoatMarker((34,-43), heading=45,
+                                            wind_heading=150, wind_speed=45, color="#8f8"))
+        mapa.add_plugin(plugins.BoatMarker((46,-30), heading=-20,
+                                            wind_heading=46, wind_speed=25, color="#88f"))
+        mapa._build_map()
+
+    def test_layer(self):
+        mapa = folium.Map([48.,5.], zoom_start=6)
+        mapa.add_plugin(plugins.Layer('//otile1.mqcdn.com/tiles/1.0.0/osm/{z}/{x}/{y}.png',
+                                      layer_name='MapQuest'))
+        mapa.add_plugin(plugins.LayerControl())
+        mapa._build_map()
+
+    def test_geo_json(self):
+        N=100
+        lons = +5 - np.random.normal(size=N)
+        lats = 48 - np.random.normal(size=N)
+
+        data = {
+            "type": "FeatureCollection",
+                "features": [
+                    {
+                        "type": "Feature",
+                        "geometry": {
+                            "type": "MultiPoint",
+                            "coordinates": [[lon, lat] for (lat,lon) in zip(lats,lons)],
+                            },
+                        "properties": {"prop0": "value0"}
+                        },
+                ],
+            }
+
+        mapa = folium.Map([48.,5.], zoom_start=6)
+        mapa.add_plugin(plugins.GeoJson(data))
+        mapa._build_map()
+
+        open('geojson_plugin_test1.json','w').write(json.dumps(data))
+        mapb = folium.Map([48.,5.], zoom_start=6)
+        mapb.add_plugin(plugins.GeoJson(open('geojson_plugin_test1.json')))
+        mapb._build_map()
+
+        data = {
+            "type": "FeatureCollection",
+                "features": [
+                    {
+                        "type": "Feature",
+                        "geometry": {
+                            "type": "MultiPolygon",
+                            "coordinates": [[[[lon+1e-4, lat+1e-4],
+                                             [lon+1e-4, lat-1e-4],
+                                             [lon-1e-4, lat-1e-4],
+                                             [lon-1e-4, lat+1e-4]]] for (lat,lon) in zip(lats,lons)],
+                            },
+                        "properties": {"prop0": "value0"}
+                        },
+                ],
+            }
+
+        mapc = folium.Map([48.,5.], zoom_start=6)
+        mapc.add_plugin(plugins.GeoJson(data))
+        mapc._build_map()
+
+        open('geojson_plugin_test2.json','w').write(json.dumps(data))
+        mapd = folium.Map([48.,5.], zoom_start=6)
+        mapd.add_plugin(plugins.GeoJson(open('geojson_plugin_test2.json')))
+        mapd._build_map()
+