test_feature_group_sub_group.py

"""
Test MarkerCluster
------------------
"""

import folium
from folium import plugins
from folium.template import Template
from folium.utilities import normalize


def test_feature_group_sub_group():
    m = folium.Map([0.0, 0.0], zoom_start=6)
    fg = folium.FeatureGroup()
    m.add_child(fg)
    g1 = plugins.FeatureGroupSubGroup(fg, "g1")
    m.add_child(g1)
    folium.Marker([1, 1]).add_to(g1)
    folium.Marker([-1, -1]).add_to(g1)
    g2 = plugins.FeatureGroupSubGroup(fg, "g2")
    folium.Marker([-1, 1]).add_to(g2)
    folium.Marker([1, -1]).add_to(g2)
    m.add_child(g2)
    folium.LayerControl().add_to(m)

    out = normalize(m._parent.render())

    # We verify that imports
    assert (
        '<script src="https://unpkg.com/[email protected]/dist/leaflet.featuregroup.subgroup.js"></script>'  # noqa
        in out
    )  # noqa

    # Verify the script part is okay.
    tmpl = Template(
        """
        var {{ this.get_name() }} = L.featureGroup.subGroup(
            {{ this._group.get_name() }}
        );
    """
    )
    assert normalize(tmpl.render(this=g1)) in out
    assert normalize(tmpl.render(this=g2)) in out

    tmpl = Template("{{ this.get_name() }}.addTo({{ this._parent.get_name() }});")
    assert normalize(tmpl.render(this=g1)) in out
    assert normalize(tmpl.render(this=g2)) in out