fix tests

Author: attack68

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

@@ -335,7 +335,7 @@ def test_1level_multiindex():
     midx = MultiIndex.from_product([[1, 2]], names=[""])
     df = DataFrame(-1, index=midx, columns=[0, 1])
     ctx = df.style._translate(True, True)
-    assert ctx["body"][0][0]["display_value"] == 1
+    assert ctx["body"][0][0]["display_value"] == "1"
     assert ctx["body"][0][0]["is_visible"] is True
-    assert ctx["body"][1][0]["display_value"] == 2
+    assert ctx["body"][1][0]["display_value"] == "2"
     assert ctx["body"][1][0]["is_visible"] is True

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

@@ -54,6 +54,8 @@ def mi_styler_comp(mi_styler):
     mi_styler.hide_index(names=True)
     mi_styler.set_table_attributes('class="box"')
     mi_styler.format(na_rep="MISSING", precision=3)
+    mi_styler.format_index(precision=2, axis=0)
+    mi_styler.format_index(precision=4, axis=1)
     mi_styler.highlight_max(axis=None)
     mi_styler.applymap_index(lambda x: "color: white;", axis=0)
     mi_styler.applymap_index(lambda x: "color: black;", axis=1)
@@ -1051,31 +1053,31 @@ def test_mi_sparse_column_names(self):
             },
             {
                 "class": "col_heading level1 col0",
-                "display_value": 1,
+                "display_value": "1",
                 "is_visible": True,
                 "type": "th",
                 "value": 1,
                 "attributes": "",
             },
             {
                 "class": "col_heading level1 col1",
-                "display_value": 0,
+                "display_value": "0",
                 "is_visible": True,
                 "type": "th",
                 "value": 0,
                 "attributes": "",
             },
             {
                 "class": "col_heading level1 col2",
-                "display_value": 1,
+                "display_value": "1",
                 "is_visible": True,
                 "type": "th",
                 "value": 1,
                 "attributes": "",
             },
             {
                 "class": "col_heading level1 col3",
-                "display_value": 0,
+                "display_value": "0",
                 "is_visible": True,
                 "type": "th",
                 "value": 0,
@@ -1174,7 +1176,7 @@ def test_hide_columns_index_mult_levels(self):
         # column headers
         assert ctx["head"][0][2]["is_visible"]
         assert ctx["head"][1][2]["is_visible"]
-        assert ctx["head"][1][3]["display_value"] == 1
+        assert ctx["head"][1][3]["display_value"] == "1"
         # indices
         assert ctx["body"][0][0]["is_visible"]
         # data