Address comments

Author: hyongtao-code

Lib/test/test_pdb.py

@@ -3666,28 +3666,6 @@ def test_print_stack_entry_uses_dynamic_line_prefix(self):
         # Check if the custom prefix appeared in the output
         self.assertIn('CUSTOM_PREFIX> ', stdout.getvalue())
 
-    def test_exec_in_closure_result_uses_pdb_stdout(self):
-        """
-        Expression results executed via _exec_in_closure() should be written
-        to the debugger output stream (pdb stdout), not to sys.stdout.
-        """
-        pdb_out = io.StringIO()
-        sys_out = io.StringIO()
-
-        p = pdb.Pdb(nosigint=True, readrc=False, stdout=pdb_out)
-
-        with redirect_stdout(sys_out):
-            self.assertTrue(p._exec_in_closure('(lambda: 123)()', {}, {}))
-            self.assertTrue(p._exec_in_closure('sum(i for i in (1, 2, 3))', {}, {}))
-
-        pdb_lines = [line.strip() for line in pdb_out.getvalue().splitlines()]
-        sys_lines = [line.strip() for line in sys_out.getvalue().splitlines()]
-
-        self.assertIn('123', pdb_lines)
-        self.assertIn('6', pdb_lines)
-        self.assertNotIn('123', sys_lines)
-        self.assertNotIn('6', sys_lines)
-
     def test_find_function_found_with_encoding_cookie(self):
         self._assert_find_function(
             """\

Lib/test/test_remote_pdb.py

@@ -1441,6 +1441,34 @@ def test_multi_line_commands(self):
             self.assertIn("Function returned: 42", stdout)
             self.assertEqual(process.returncode, 0)
 
+    def test_exec_in_closure_result_uses_pdb_stdout(self):
+        """
+        Expression results executed via _exec_in_closure() should be written
+        to the debugger output stream (pdb stdout), not to sys.stdout.
+        """
+        self._create_script()
+        process, client_file = self._connect_and_get_client_file()
+
+        with kill_on_error(process):
+            self._read_until_prompt(client_file)
+
+            self._send_command(client_file, "(lambda: 123)()")
+            messages = self._read_until_prompt(client_file)
+            result_msg = "".join(msg.get("message", "") for msg in messages)
+            self.assertIn("123", result_msg)
+
+            self._send_command(client_file, "sum(i for i in (1, 2, 3))")
+            messages = self._read_until_prompt(client_file)
+            result_msg = "".join(msg.get("message", "") for msg in messages)
+            self.assertIn("6", result_msg)
+
+            self._send_command(client_file, "c")
+            stdout, _ = process.communicate(timeout=SHORT_TIMEOUT)
+
+            self.assertNotIn("\n123\n", stdout)
+            self.assertNotIn("\n6\n", stdout)
+            self.assertEqual(process.returncode, 0)
+
 
 def _supports_remote_attaching():
     PROCESS_VM_READV_SUPPORTED = False