From bd14debebea77a8e3772c0042dec01a2a585c30c Mon Sep 17 00:00:00 2001
From: Dean Li <deantvv@gmail.com>
Date: Sat, 13 Nov 2021 03:36:25 +0000
Subject: [PATCH] test: update test_support

---
 Lib/test/test_support.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/Lib/test/test_support.py b/Lib/test/test_support.py
index 445b29e5f..6a872d95d 100644
--- a/Lib/test/test_support.py
+++ b/Lib/test/test_support.py
@@ -642,14 +642,8 @@ class TestSupport(unittest.TestCase):
 
     def check_print_warning(self, msg, expected):
         stderr = io.StringIO()
-
-        old_stderr = sys.__stderr__
-        try:
-            sys.__stderr__ = stderr
+        with support.swap_attr(support.print_warning, 'orig_stderr', stderr):
             support.print_warning(msg)
-        finally:
-            sys.__stderr__ = old_stderr
-
         self.assertEqual(stderr.getvalue(), expected)
 
     def test_print_warning(self):