Use assertWarnsRegex

AlexWaygood · AlexWaygood · commit a24e9b4bf56b · 2023-04-30T16:27:18.000+01:00
diff --git a/Lib/test/test_calendar.py b/Lib/test/test_calendar.py
@@ -493,12 +493,11 @@ def test_format(self):
 class CalendarTestCase(unittest.TestCase):
 
     def test_deprecation_warning(self):
-        warnings.resetwarnings()
-        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
