save memory and modify format

Author: ZKaoChi

pandas/tests/io/excel/test_style.py

@@ -352,7 +352,7 @@ def test_format_hierarchical_rows_periodindex(merge_cells):
         ),
     )
     formatter = ExcelFormatter(df, merge_cells=merge_cells)
-    formatted_cells = list(formatter._format_hierarchical_rows())
+    formatted_cells = formatter._format_hierarchical_rows()
 
     for cell in formatted_cells:
         assert not isinstance(

pandas/tests/io/excel/test_writers.py

@@ -366,14 +366,12 @@ def test_excel_round_trip_with_periodindex(self, tmp_excel, merge_cells):
                 names=["date", "category"],
             ),
         )
-        if tmp_excel.endswith(".ods"):
-            expected.index = expected.index.set_levels(
-                expected.index.levels[0].astype("datetime64[s]"), level=0
-            )
-        else:
-            expected.index = expected.index.set_levels(
-                expected.index.levels[0].astype("datetime64[us]"), level=0
-            )
+        time_format = (
+            "datetime64[s]" if tmp_excel.endswith(".ods") else "datetime64[us]"
+        )
+        expected.index = expected.index.set_levels(
+            expected.index.levels[0].astype(time_format), level=0
+        )
 
         tm.assert_frame_equal(result, expected)