test _repr_png_() for MultivarColormap

trygvrad · trygvrad · commit 438923f5a3d7 · 2024-08-19T09:34:56.000+02:00
diff --git a/lib/matplotlib/tests/test_multivariate_colormaps.py b/lib/matplotlib/tests/test_multivariate_colormaps.py
@@ -478,6 +478,19 @@ def test_bivariate_repr_html():
     assert html.endswith('</div>')
 
 
+def test_multivariate_repr_png():
+    cmap = mpl.multivar_colormaps['3VarAddA']
+    png = cmap._repr_png_()
+    assert len(png) > 0
+    img = Image.open(BytesIO(png))
+    assert img.width > 0
+    assert img.height > 0
+    assert 'Title' in img.text
+    assert 'Description' in img.text
+    assert 'Author' in img.text
+    assert 'Software' in img.text
+
+
 def test_multivariate_repr_html():
     cmap = mpl.multivar_colormaps['3VarAddA']
     html = cmap._repr_html_()
