Spaces:
Running
Running
File size: 2,477 Bytes
ba2f5d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
"""Tests of various renderers"""
import json
import pytest
import altair.vegalite.v4 as alt
@pytest.fixture
def chart():
return alt.Chart("data.csv").mark_point()
def test_html_renderer_embed_options(chart, renderer="html"):
"""Test that embed_options in renderer metadata are correctly manifest in html"""
# Short of parsing the javascript, it's difficult to parse out the
# actions. So we use string matching
def assert_has_options(chart, **opts):
html = chart._repr_mimebundle_(None, None)["text/html"]
for key, val in opts.items():
assert json.dumps({key: val})[1:-1] in html
with alt.renderers.enable(renderer):
assert_has_options(chart, mode="vega-lite")
with alt.renderers.enable(embed_options=dict(actions={"export": True})):
assert_has_options(chart, mode="vega-lite", actions={"export": True})
with alt.renderers.set_embed_options(actions=True):
assert_has_options(chart, mode="vega-lite", actions=True)
def test_mimetype_renderer_embed_options(chart, renderer="mimetype"):
# check that metadata is passed appropriately
mimetype = alt.display.VEGALITE_MIME_TYPE
spec = chart.to_dict()
with alt.renderers.enable(renderer):
# Sanity check: no metadata specified
bundle, metadata = chart._repr_mimebundle_(None, None)
assert bundle[mimetype] == spec
assert metadata == {}
with alt.renderers.set_embed_options(actions=False):
bundle, metadata = chart._repr_mimebundle_(None, None)
assert set(bundle.keys()) == {mimetype, "text/plain"}
assert bundle[mimetype] == spec
assert metadata == {mimetype: {"embed_options": {"actions": False}}}
def test_json_renderer_embed_options(chart, renderer="json"):
"""Test that embed_options in renderer metadata are correctly manifest in html"""
mimetype = "application/json"
spec = chart.to_dict()
with alt.renderers.enable(renderer):
# Sanity check: no options specified
bundle, metadata = chart._repr_mimebundle_(None, None)
assert bundle[mimetype] == spec
assert metadata == {}
with alt.renderers.enable(option="foo"):
bundle, metadata = chart._repr_mimebundle_(None, None)
assert set(bundle.keys()) == {mimetype, "text/plain"}
assert bundle[mimetype] == spec
assert metadata == {mimetype: {"option": "foo"}}
|