gh-104012: Ensure test_calendar.CalendarTestCase.test_deprecation_war…

…ning consistently passes (#104014)
python · Apr 30, 2023 · 7d3931e · 7d3931e
1 parent f186557
commit 7d3931e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Lib/test/test_calendar.py b/Lib/test/test_calendar.py
@@ -493,11 +493,11 @@ def test_format(self):
 class CalendarTestCase(unittest.TestCase):
 
     def test_deprecation_warning(self):
-        with warnings.catch_warnings(record=True) as w:
+        with self.assertWarnsRegex(
+            DeprecationWarning,
+            "The 'January' attribute is deprecated, use 'JANUARY' instead"
+        ):
             calendar.January
-        self.assertEqual(len(w), 1)
-        self.assertEqual(w[0].category, DeprecationWarning)
-        self.assertIn("The 'January' attribute is deprecated, use 'JANUARY' instead", str(w[0].message))
 
     def test_isleap(self):
         # Make sure that the return is right for a few years, and