Fix marker cluster test

python-visualization · Nov 27, 2023 · 7e7b152 · 7e7b152
1 parent 633d600
commit 7e7b152
Showing 1 changed file with 19 additions and 19 deletions.
diff --git a/tests/plugins/test_marker_cluster.py b/tests/plugins/test_marker_cluster.py
@@ -23,24 +23,7 @@ def test_marker_cluster():
     m = folium.Map([45.0, 3.0], zoom_start=4)
     mc = plugins.MarkerCluster(data).add_to(m)
 
-    out = normalize(m._parent.render())
-
-    # We verify that imports
-    assert (
-        '<script src="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/leaflet.markercluster.js"></script>'  # noqa
-        in out
-    )  # noqa
-    assert (
-        '<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/MarkerCluster.css"/>'  # noqa
-        in out
-    )  # noqa
-    assert (
-        '<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/MarkerCluster.Default.css"/>'  # noqa
-        in out
-    )  # noqa
-
-    # Verify the script part is okay.
-    tmpl = Template(
+    tmpl_for_expected = Template(
         """
         var {{this.get_name()}} = L.markerClusterGroup(
             {{ this.options|tojson }}
@@ -60,7 +43,24 @@ def test_marker_cluster():
         {{ this.get_name() }}.addTo({{ this._parent.get_name() }});
     """
     )
-    expected = normalize(tmpl.render(this=mc))
+    expected = normalize(tmpl_for_expected.render(this=mc))
+
+    out = normalize(m._parent.render())
+
+    # We verify that imports
+    assert (
+        '<script src="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/leaflet.markercluster.js"></script>'  # noqa
+        in out
+    )  # noqa
+    assert (
+        '<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/MarkerCluster.css"/>'  # noqa
+        in out
+    )  # noqa
+    assert (
+        '<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/leaflet.markercluster/1.1.0/MarkerCluster.Default.css"/>'  # noqa
+        in out
+    )  # noqa
+
     assert expected in out
 
     bounds = m.get_bounds()