@@ -35,7 +35,7 @@ def chart():
3535 return alt .Chart ("data.csv" ).mark_point ()
3636
3737
38- def test_html_renderer_embed_options (chart , renderer = "html" ):
38+ def test_html_renderer_embed_options (chart ):
3939 """Test that embed_options in renderer metadata are correctly manifest in html."""
4040 # Short of parsing the javascript, it's difficult to parse out the
4141 # actions. So we use string matching
@@ -45,7 +45,7 @@ def assert_has_options(chart, **opts):
4545 for key , val in opts .items ():
4646 assert json .dumps ({key : val })[1 :- 1 ] in html
4747
48- with alt .renderers .enable (renderer ):
48+ with alt .renderers .enable ("html" ):
4949 assert_has_options (chart , mode = "vega-lite" )
5050
5151 with alt .renderers .enable (embed_options = {"actions" : {"export" : True }}):
@@ -55,11 +55,13 @@ def assert_has_options(chart, **opts):
5555 assert_has_options (chart , mode = "vega-lite" , actions = True )
5656
5757
58- def test_mimetype_renderer_embed_options (chart , renderer = "mimetype" ):
58+ def test_mimetype_renderer_embed_options (chart ):
5959 # check that metadata is passed appropriately
60- mimetype = alt .display .VEGALITE_MIME_TYPE
60+ from altair .vegalite .v5 .display import VEGALITE_MIME_TYPE
61+
62+ mimetype = VEGALITE_MIME_TYPE
6163 spec = chart .to_dict ()
62- with alt .renderers .enable (renderer ):
64+ with alt .renderers .enable ("mimetype" ):
6365 # Sanity check: no metadata specified
6466 bundle , metadata = chart ._repr_mimebundle_ (None , None )
6567 assert bundle [mimetype ] == spec
@@ -71,11 +73,11 @@ def test_mimetype_renderer_embed_options(chart, renderer="mimetype"):
7173 assert metadata == {mimetype : {"embed_options" : {"actions" : False }}}
7274
7375
74- def test_json_renderer_embed_options (chart , renderer = "json" ):
76+ def test_json_renderer_embed_options (chart ):
7577 """Test that embed_options in renderer metadata are correctly manifest in html."""
7678 mimetype = "application/json"
7779 spec = chart .to_dict ()
78- with alt .renderers .enable (renderer ):
80+ with alt .renderers .enable ("json" ):
7981 # Sanity check: no options specified
8082 bundle , metadata = chart ._repr_mimebundle_ (None , None )
8183 assert bundle [mimetype ] == spec
@@ -89,12 +91,12 @@ def test_json_renderer_embed_options(chart, renderer="json"):
8991
9092
9193@skip_requires_vl_convert
92- def test_renderer_with_none_embed_options (chart , renderer = "mimetype" ):
94+ def test_renderer_with_none_embed_options (chart ):
9395 # Check that setting embed_options to None doesn't crash
9496 from altair .utils .mimebundle import spec_to_mimebundle
9597
9698 spec = chart .to_dict ()
97- with alt .renderers .enable (renderer , embed_options = None ):
99+ with alt .renderers .enable ("mimetype" , embed_options = None ):
98100 bundle = spec_to_mimebundle (
99101 spec = spec ,
100102 mode = "vega-lite" ,
@@ -105,9 +107,9 @@ def test_renderer_with_none_embed_options(chart, renderer="mimetype"):
105107
106108
107109@jupyter_marks
108- def test_jupyter_renderer_mimetype (chart , renderer = "jupyter" ) -> None :
110+ def test_jupyter_renderer_mimetype (chart ) -> None :
109111 """Test that we get the expected widget mimetype when the jupyter renderer is enabled."""
110- with alt .renderers .enable (renderer ):
112+ with alt .renderers .enable ("jupyter" ):
111113 assert (
112114 "application/vnd.jupyter.widget-view+json"
113115 in chart ._repr_mimebundle_ (None , None )[0 ]
0 commit comments