PNG repr tests

Author: ocefpaf

tests/test_repr.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+""""
+Folium _repr_*_ Tests
+---------------------
+
+"""
+
+import io
+
+import folium
+
+import pytest
+
+import PIL.Image
+
+
+@pytest.fixture
+def make_map():
+    m = folium.Map()
+    return m
+
+
+def test__repr_html_is_str():
+    html = make_map()._repr_html_()
+    assert isinstance(html, str)
+
+
+def test_valid_html():
+    html = make_map()._repr_html_()
+    parts = html.split('><')
+    assert len(parts) == 6
+    assert parts[0].lstrip('<div ') == 'style="width:100%;"'
+    assert parts[1].lstrip('<div ') == 'style="position:relative;width:100%;height:0;padding-bottom:60%;"'  # noqa
+    assert parts[2].startswith('iframe')
+    assert parts[3] == '/iframe'
+    assert parts[4] == '/div'
+    assert parts[5] == '/div>'
+
+
+def test__repr_png_is_bytes():
+    png = make_map()._repr_png_()
+    assert isinstance(png, bytes)
+
+
+def test_valid_png():
+    png = make_map()._repr_png_()
+    img = PIL.Image.open(io.BytesIO(png))
+    isinstance(img, PIL.PngImagePlugin.PngImageFile)