fix tests

pandas-dev · attack68 · May 24, 2021 · May 26, 2021 · May 29, 2021 · Jun 1, 2021
commit 6ea94bfa1600369b0a1d220976941dcb078f2ce4
diff --git a/pandas/core/generic.py b/pandas/core/generic.py
@@ -3367,14 +3367,18 @@ def to_latex(
             escape="latex" if escape else None,
         )
 
-        for ax in [0, 1]:
+        for ax in [0, 1]:  #
+            fmt: callable | str | None = None
+            if isinstance(formatter, dict):
+                fmt = formatter.get("__index__" if ax == 0 else "__columns__")
             styler.format_index(
                 axis=ax,
                 escape="latex" if escape else None,
                 na_rep=na_rep,
                 decimal=decimal,
                 precision=precision,
                 thousands=thousands,
+                formatter=fmt,
             )
 
         if header is False:

diff --git a/pandas/tests/io/formats/test_to_latex.py b/pandas/tests/io/formats/test_to_latex.py
@@ -951,17 +951,17 @@ def test_to_latex_with_formatters(self):
             "object": lambda x: f"-{x!s}-",
             "__index__": lambda x: f"index: {x}",
         }
-        result = df.to_latex(formatters=dict(formatters))
+        result = df.to_latex(formatter=formatters, hrules=True)
 
         expected = _dedent(
             r"""
             \begin{tabular}{llrrl}
             \toprule
-            {} & datetime64 &  float & int &    object \\
+            {} & {datetime64} & {float} & {int} & {object} \\
             \midrule
-            index: 0 &    2016-01 & [ 1.0] & 0x1 &  -(1, 2)- \\
-            index: 1 &    2016-02 & [ 2.0] & 0x2 &    -True- \\
-            index: 2 &    2016-03 & [ 3.0] & 0x3 &   -False- \\
+            index: 0 & 2016-01 & [ 1.0] & 0x1 & -(1, 2)- \\
+            index: 1 & 2016-02 & [ 2.0] & 0x2 & -True- \\
+            index: 2 & 2016-03 & [ 3.0] & 0x3 & -False- \\
             \bottomrule
             \end{tabular}
             """