Re applied mods to TestPeriodIndexFormat, since it had moved somewhere else

Author: Sylvain MARIE

pandas/tests/indexes/period/test_formats.py

@@ -239,31 +239,78 @@ def test_period_format_and_strftime_default(self):
         assert formatted[0] == "2003-01-01 12:01:01.123456789"
         assert formatted[1] == "2003-01-01 12:01:01.123456790"
 
-    def test_period_custom(self):
+    @pytest.mark.parametrize("fast_strftime", (False, True))
+    def test_period_custom(self, fast_strftime):
         # GH#46252 custom formatting directives %l (ms) and %u (us)
         msg = "PeriodIndex.format is deprecated"
 
         # 3 digits
-        per = pd.period_range("2003-01-01 12:01:01.123", periods=2, freq="ms")
+        per = pd.period_range("2003-01-01 12:01:01.123", periods=2, freq="l")
         with tm.assert_produces_warning(FutureWarning, match=msg):
-            formatted = per.format(date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)")
+            formatted = per.format(
+                date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)",
+                fast_strftime=fast_strftime,
+            )
         assert formatted[0] == "03 12:01:01 (ms=123 us=123000 ns=123000000)"
         assert formatted[1] == "03 12:01:01 (ms=124 us=124000 ns=124000000)"
 
         # 6 digits
-        per = pd.period_range("2003-01-01 12:01:01.123456", periods=2, freq="us")
+        per = pd.period_range("2003-01-01 12:01:01.123456", periods=2, freq="u")
         with tm.assert_produces_warning(FutureWarning, match=msg):
-            formatted = per.format(date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)")
+            formatted = per.format(
+                date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)",
+                fast_strftime=fast_strftime,
+            )
         assert formatted[0] == "03 12:01:01 (ms=123 us=123456 ns=123456000)"
         assert formatted[1] == "03 12:01:01 (ms=123 us=123457 ns=123457000)"
 
         # 9 digits
-        per = pd.period_range("2003-01-01 12:01:01.123456789", periods=2, freq="ns")
+        per = pd.period_range("2003-01-01 12:01:01.123456789", periods=2, freq="n")
         with tm.assert_produces_warning(FutureWarning, match=msg):
-            formatted = per.format(date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)")
+            formatted = per.format(
+                date_format="%y %I:%M:%S (ms=%l us=%u ns=%n)",
+                fast_strftime=fast_strftime,
+            )
         assert formatted[0] == "03 12:01:01 (ms=123 us=123456 ns=123456789)"
         assert formatted[1] == "03 12:01:01 (ms=123 us=123456 ns=123456790)"
 
+    @pytest.mark.parametrize("fast_strftime", (False, True))
+    @pytest.mark.parametrize(
+        "locale_str",
+        [
+            pytest.param(None, id=str(locale.getlocale())),
+            "it_IT.utf8",
+            "it_IT",  # Note: encoding will be 'ISO8859-1'
+            "zh_CN.utf8",
+            "zh_CN",  # Note: encoding will be 'gb2312'
+        ],
+    )
+    def test_period_custom_pm(self, fast_strftime, locale_str):
+        """Test that using %p in the custom format work well"""
+
+        # Skip if locale cannot be set
+        if locale_str is not None and not tm.can_set_locale(locale_str, locale.LC_ALL):
+            pytest.skip(f"Skipping as locale '{locale_str}' cannot be set on host.")
+
+        # Change locale temporarily for this test.
+        with tm.set_locale(locale_str, locale.LC_ALL) if locale_str else nullcontext():
+            # Get locale-specific reference
+            am_local, pm_local = get_local_am_pm()
+
+            # 9 digits
+            p = pd.period_range("2003-01-01 12:01:01.123456789", periods=2, freq="n")
+            formatted = p.format(
+                date_format="%y %I:%M:%S%p (ms=%l us=%u ns=%n)",
+                fast_strftime=fast_strftime,
+            )
+            assert (
+                formatted[0] == f"03 12:01:01{pm_local} (ms=123 us=123456 ns=123456789)"
+            )
+            assert (
+                formatted[1] == f"03 12:01:01{pm_local} (ms=123 us=123456 ns=123456790)"
+            )
+            # fmt: on
+
     def test_period_tz(self):
         # Formatting periods created from a datetime with timezone.
         msg = r"PeriodIndex\.format is deprecated"