test_heat_map_withtime.py

"""
Test HeatMapWithTime
------------
"""

import numpy as np

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


def test_heat_map_with_time():
    np.random.seed(3141592)
    initial_data = np.random.normal(size=(100, 2)) * np.array([[1, 1]]) + np.array(
        [[48, 5]]
    )
    move_data = np.random.normal(size=(100, 2)) * 0.01
    data = [(initial_data + move_data * i).tolist() for i in range(100)]
    m = folium.Map([48.0, 5.0], zoom_start=6)
    hm = plugins.HeatMapWithTime(data).add_to(m)

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

    # We verify that the script imports are present.
    script = '<script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/leaflet.timedimension.min.js"></script>'  # noqa
    assert script in out
    script = '<script src="https://cdn.jsdelivr.net/gh/python-visualization/folium/folium/templates/pa7_hm.min.js"></script>'  # noqa
    assert script in out
    script = '<script src="https://cdn.jsdelivr.net/gh/python-visualization/folium/folium/templates/pa7_leaflet_hm.min.js"></script>'  # noqa
    assert script in out
    script = '<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/leaflet.timedimension.control.css"/>'  # noqa
    assert script in out

    # We verify that the script part is correct.
    tmpl = Template("""
        var times = {{this.times}};

        {{this._parent.get_name()}}.timeDimension = L.timeDimension(
            {times : times, currentTime: new Date(1)}
        );

        var {{this._control_name}} = new L.Control.TimeDimensionCustom({{this.index}}, {
            autoPlay: {{this.auto_play}},
            backwardButton: {{this.backward_button}},
            displayDate: {{this.display_index}},
            forwardButton: {{this.forward_button}},
            limitMinimumRange: {{this.limit_minimum_range}},
            limitSliders: {{this.limit_sliders}},
            loopButton: {{this.loop_button}},
            maxSpeed: {{this.max_speed}},
            minSpeed: {{this.min_speed}},
            playButton: {{this.play_button}},
            playReverseButton: {{this.play_reverse_button}},
            position: "{{this.position}}",
            speedSlider: {{this.speed_slider}},
            speedStep: {{this.speed_step}},
            styleNS: "{{this.style_NS}}",
            timeSlider: {{this.time_slider}},
            timeSliderDragUpdate: {{this.time_slider_drag_update}},
            timeSteps: {{this.index_steps}}
            })
            .addTo({{this._parent.get_name()}});

            var {{this.get_name()}} = new TDHeatmap({{this.data}},
            {heatmapOptions: {
                    radius: {{this.radius}},
                    blur: {{this.blur}},
                    minOpacity: {{this.min_opacity}},
                    maxOpacity: {{this.max_opacity}},
                    scaleRadius: {{this.scale_radius}},
                    useLocalExtrema: {{this.use_local_extrema}},
                    defaultWeight: 1,
                    {% if this.gradient %}gradient: {{ this.gradient }}{% endif %}
                }
            });
            {{ this.get_name() }}.addTo({{ this._parent.get_name() }});
    """)

    assert normalize(tmpl.render(this=hm)) in out