Enhance test_timestamped_geo_json.py

Author: Martin Journois

tests/plugins/test_timestamped_geo_json.py

@@ -2,12 +2,12 @@
 """
 Test TimestampedGeoJson
 -----------------------
-
 """
+from jinja2 import Template
+import numpy as np
 
 import folium
 from folium import plugins
-import numpy as np
 
 def test_timestamped_geo_json():
     coordinates = [[[[lon-8*np.sin(theta), -47+6*np.cos(theta)] for
@@ -81,7 +81,50 @@ def test_timestamped_geo_json():
             ],
         }
 
-    mape = folium.Map([47, 3], zoom_start=1)
-    mape.add_children(plugins.TimestampedGeoJson(data))
-    mape._repr_html_()
+    m = folium.Map([47, 3], zoom_start=1)
+    tgj = plugins.TimestampedGeoJson(data)
+    m.add_children(tgj)
+    m._repr_html_()
+
+    out = m._parent.render()
+
+    # Verify the imports.
+    assert ('<script src="https://cdnjs.cloudflare.com/ajax/libs/jquery/2.0.0/'
+            'jquery.min.js"></script>'
+            ) in out
+    assert ('<script src="https://cdnjs.cloudflare.com/ajax/libs/jqueryui/'
+            '1.10.2/jquery-ui.min.js"></script>'
+            ) in out
+    assert ('<script src="https://raw.githubusercontent.com/nezasa/'
+            'iso8601-js-period/master/iso8601.min.js"></script>'
+            ) in out
+    assert ('<script src="https://raw.githubusercontent.com/socib/Leaflet.'
+            'TimeDimension/master/dist/leaflet.timedimension.min.js">'
+            '</script>'
+            ) in out
+    assert ('<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/'
+            'libs/highlight.js/8.4/styles/default.min.css" />'
+            ) in out
+    assert ('<link rel="stylesheet" href="http://apps.socib.es/Leaflet.'
+            'TimeDimension/dist/leaflet.timedimension.control.min.css" />'
+            ) in out
+
+    # Verify that the script is okay.
+    tmpl = Template("""
+        {{this._parent.get_name()}}.timeDimension = L.timeDimension();
+        {{this._parent.get_name()}}.timeDimensionControl = L.control.timeDimension({
+            position: 'bottomleft',
+            autoPlay: {{'true' if this.auto_play else 'false'}},
+            playerOptions: {
+                transitionTime: {{this.transition_time}},
+                loop: {{'true' if this.loop else 'false'}}}
+                });
+        {{this._parent.get_name()}}.addControl({{this._parent.get_name()}}.timeDimensionControl);
+
+        var {{this.get_name()}} = L.timeDimension.layer.geoJson(
+            L.geoJson({{this.data}}),
+            {updateTimeDimension: true,addlastPoint: true}
+            ).addTo({{this._parent.get_name()}});
+    """)
 
+    assert ''.join(tmpl.render(this=tgj).split()) in ''.join(out.split())