From a42524702975068a687947c4c6d5cd03d4a89912 Mon Sep 17 00:00:00 2001
From: jorenham <jhammudoglu@gmail.com>
Date: Sat, 8 Mar 2025 16:39:04 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20also=20rewrite=20the=20`stubtest?=
 =?UTF-8?q?`=20stderr=20output?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tool/stubtest.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tool/stubtest.py b/tool/stubtest.py
index 3f92adde..b0d88f10 100644
--- a/tool/stubtest.py
+++ b/tool/stubtest.py
@@ -128,10 +128,10 @@ def main() -> int:
         capture_output=True,
         env={"FORCE_COLOR": "1"} | os.environ,
     )
-    sys.stderr.buffer.write(result.stderr)
-    output = _rewrite_mypy_output(result.stdout)
-    sys.stdout.buffer.write(output)
-    sys.stdout.buffer.flush()
+    for std in "stdout", "stderr":
+        stream = getattr(sys, std)
+        stream.buffer.write(_rewrite_mypy_output(getattr(result, std)))
+        stream.buffer.flush()
 
     return result.returncode