minor py2 cleanup for sys.stderr.buffer

asottile · asottile · commit f637ac860312 · 2021-12-20T17:01:51.000-05:00
diff --git a/tests/languages/helpers_test.py b/tests/languages/helpers_test.py
@@ -72,8 +72,8 @@ def test_basic_healthy():
 def test_failed_setup_command_does_not_unicode_error():
     script = (
         'import sys\n'
-        "getattr(sys.stderr, 'buffer', sys.stderr).write(b'\\x81\\xfe')\n"
-        'exit(1)\n'
+        "sys.stderr.buffer.write(b'\\x81\\xfe')\n"
+        'raise SystemExit(1)\n'
     )
 
     # an assertion that this does not raise `UnicodeError`
