Modify the tests

Author: ZKaoChi

pandas/tests/io/excel/test_style.py

@@ -9,6 +9,9 @@
 
 from pandas import (
     DataFrame,
+    MultiIndex,
+    Period,
+    period_range,
     read_excel,
 )
 import pandas._testing as tm
@@ -333,3 +336,39 @@ def test_styler_to_s3(s3_public_bucket, s3so):
             f"s3://{mock_bucket_name}/{target_file}", index_col=0, storage_options=s3so
         )
         tm.assert_frame_equal(result, df)
+
+
+def test_format_hierarchical_rows_periodindex_merge_cells():
+    # GH#60099
+    df = DataFrame(
+        {"A": [1, 2]},
+        index=MultiIndex.from_arrays(
+            [period_range("2023-01", "2023-02", freq="M"), ["X", "Y"]],
+            names=["date", "category"],
+        ),
+    )
+    formatter = ExcelFormatter(df, merge_cells=True)
+    formatted_cells = list(formatter._format_hierarchical_rows())
+
+    for cell in formatted_cells:
+        assert not isinstance(
+            cell.val, Period
+        ), "Period should be converted to Timestamp"
+
+
+def test_format_hierarchical_rows_periodindex_no_merge_cells():
+    # GH#60099
+    df = DataFrame(
+        {"A": [1, 2]},
+        index=MultiIndex.from_arrays(
+            [period_range("2023-01", "2023-02", freq="M"), ["X", "Y"]],
+            names=["date", "category"],
+        ),
+    )
+    formatter = ExcelFormatter(df, merge_cells=False)
+    formatted_cells = list(formatter._format_hierarchical_rows())
+
+    for cell in formatted_cells:
+        assert not isinstance(
+            cell.val, Period
+        ), "Period should be converted to Timestamp"

pandas/tests/io/formats/test_to_excel.py

@@ -9,13 +9,6 @@
 
 from pandas.errors import CSSWarning
 
-from pandas import (
-    DataFrame,
-    MultiIndex,
-    Timestamp,
-    period_range,
-    to_datetime,
-)
 import pandas._testing as tm
 
 from pandas.io.formats.excel import (
@@ -435,35 +428,3 @@ def test_css_excel_cell_cache(styles, cache_hits, cache_misses):
 
     assert cache_info.hits == cache_hits
     assert cache_info.misses == cache_misses
-
-
-def test_format_hierarchical_rows_with_periodindex():
-    # GH#60099
-    period_index = period_range(start="2006-10-06", end="2006-10-07", freq="D")
-    df = DataFrame({"A": [0, 0]}, index=period_index)
-    converter = CSSToExcelConverter()
-    cells: list[CssExcelCell] = list(converter._format_hierarchical_rows(df))
-    for cell in cells:
-        assert isinstance(cell.val, Timestamp), "Expected cell value to be a Timestamp"
-        assert cell.val in to_datetime(
-            ["2006-10-06", "2006-10-07"]
-        ), "Unexpected cell value"
-
-
-def test_format_hierarchical_rows_with_period():
-    # GH#60099
-    period_index = period_range(start="2006-10-06", end="2006-10-07", freq="D")
-    number_index = ["1", "2"]
-    df = DataFrame(
-        {"A": [0, 0]}, index=MultiIndex.from_arrays([period_index, number_index])
-    )
-    converter = CSSToExcelConverter()
-    cells: list[CssExcelCell] = list(converter._format_hierarchical_rows(df))
-    for cell in cells:
-        if cell.css_col == 0:
-            assert isinstance(
-                cell.val, Timestamp
-            ), "Expected cell value to be a Timestamp"
-            assert cell.val in to_datetime(
-                ["2006-10-06", "2006-10-07"]
-            ), "Unexpected cell value"