add tests

Author: attack68

pandas/tests/io/formats/style/test_bar.py

@@ -305,13 +305,3 @@ def test_bar_value_error_raises():
     msg = r"`height` must be a value in \[0, 100\]"
     with pytest.raises(ValueError, match=msg):
         df.style.bar(height=200).to_html()
-
-
-def test_bar_color_raises(df_pos):
-    msg = "`colors` must be a matplotlib Colormap if not string"
-    with pytest.raises(ValueError, match=msg):
-        df_pos.style.bar(color={"a", "b"}).to_html()
-
-    msg = "`color` must be string, list-like of 2 strings, or matplotlib Colormap"
-    with pytest.raises(ValueError, match=msg):
-        df_pos.style.bar(color=["a", "b", "c"]).to_html()

pandas/tests/io/formats/style/test_matplotlib.py

@@ -10,6 +10,8 @@
 pytest.importorskip("matplotlib")
 pytest.importorskip("jinja2")
 
+import matplotlib as mpl
+
 from pandas.io.formats.style import Styler
 
 
@@ -256,3 +258,26 @@ def test_background_gradient_gmap_wrong_series(styler_blank):
     gmap = Series([1, 2], index=["X", "Y"])
     with pytest.raises(ValueError, match=msg):
         styler_blank.background_gradient(gmap=gmap, axis=None)._compute()
+
+
+def test_bar_colormap():
+    data = DataFrame([[1, 2], [3, 4]])
+    ctx = data.style.bar(color=mpl.cm.get_cmap("PuBu"), axis=None)._compute().ctx
+    pubu_colors = {
+        (0, 0): "#d0d1e6",
+        (1, 0): "#056faf",
+        (0, 1): "#73a9cf",
+        (1, 1): "#023858",
+    }
+    for k, v in pubu_colors.items():
+        assert v in ctx[k][1][1]
+
+
+def test_bar_color_raises(df):
+    msg = "`colors` must be a matplotlib Colormap if not string or list of strings"
+    with pytest.raises(ValueError, match=msg):
+        df.style.bar(color={"a", "b"}).to_html()
+
+    msg = "`color` must be string, list-like of 2 strings, or matplotlib Colormap"
+    with pytest.raises(ValueError, match=msg):
+        df.style.bar(color=["a", "b", "c"]).to_html()