test for raises ValueError

pandas-dev · jreback · Aug 11, 2021 · Jun 9, 2021 · Jun 9, 2021 · Jun 9, 2021
commit 6f5b46c8071f76c8d4153e64e066b55f15449bb8
diff --git a/pandas/tests/io/formats/style/test_style.py b/pandas/tests/io/formats/style/test_style.py
@@ -190,6 +190,11 @@ def test_apply_map_header_mi(mi_styler, method, axis):
     assert getattr(result, f"ctx_{axis}") == expected
 
 
+def test_apply_map_header_raises(mi_styler):
+    with pytest.raises(ValueError, match="`axis` must be one of 0, 1, 'index', 'col"):
+        mi_styler.applymap_header(lambda v: "attr: val;", axis="bad-axis")._compute()
+
+
 class TestStyler:
     def setup_method(self, method):
         np.random.seed(24)