Add tests for _repr_png_ and _repr_html_.

bdice · bdice · commit d3ca7e251846 · 2020-07-11T12:38:50.000-05:00
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1135,3 +1135,16 @@ def test_hex_shorthand_notation():
 def test_DivergingNorm_deprecated():
     with pytest.warns(cbook.MatplotlibDeprecationWarning):
         norm = mcolors.DivergingNorm(vcenter=0)
+
+
+def test_repr_png():
+    cmap = plt.get_cmap('viridis')
+    png = cmap._repr_png_()
+    assert len(png) > 0
+
+
+def test_repr_html():
+    cmap = plt.get_cmap('viridis')
+    html = cmap._repr_html_()
+    assert len(html) > 0
+    assert cmap.name in html
