@@ -157,24 +157,34 @@ def test_altair_formatter_mimebundle():
157157
158158
159159@pytest .mark .skipif (not HAS_DEPS , reason = "altair not installed" )
160- def test_altair_formatter_svg ():
160+ @pytest .mark .parametrize (
161+ ("inline" , "expected" ),
162+ [
163+ (True , "<svg></svg>" ),
164+ (False , "data:image/svg+xml;base64,PHN2Zz48L3N2Zz4=" ),
165+ ],
166+ )
167+ def test_altair_formatter_svg (inline : bool , expected : str ):
161168 AltairFormatter ().register ()
162169
163170 import altair as alt
164171
165172 # Create a mock chart with a _repr_mimebundle_ method that returns SVG
166173 mock_chart = alt .Chart (get_data ()).mark_point ()
167- with patch .object (
168- alt .Chart ,
169- "_repr_mimebundle_" ,
170- return_value = {"image/svg+xml" : "<svg></svg>" },
174+ with (
175+ patch .dict (alt .renderers .options , {"inline" : inline }),
176+ patch .object (
177+ alt .Chart ,
178+ "_repr_mimebundle_" ,
179+ return_value = {"image/svg+xml" : "<svg></svg>" },
180+ ),
171181 ):
172182 formatter = get_formatter (mock_chart )
173183 assert formatter is not None
174184 mime , content = formatter (mock_chart )
175185
176186 assert mime == "image/svg+xml"
177- assert content == "data:image/svg+xml;base64,PHN2Zz48L3N2Zz4="
187+ assert content == expected
178188
179189
180190@pytest .mark .skipif (not HAS_DEPS , reason = "altair not installed" )
0 commit comments