Handle changed output in Python 3.11

sebastinas · sebastinas · commit 742085633868 · 2022-10-26T17:37:41.000+02:00
diff --git a/bpython/test/test_interpreter.py b/bpython/test/test_interpreter.py
@@ -112,19 +112,35 @@ def gfunc():
 
         global_not_found = "name 'gfunc' is not defined"
 
-        expected = (
-            "Traceback (most recent call last):\n  File "
-            + green('"<input>"')
-            + ", line "
-            + bold(magenta("1"))
-            + ", in "
-            + cyan("<module>")
-            + "\n    gfunc()\n"
-            + bold(red("NameError"))
-            + ": "
-            + cyan(global_not_found)
-            + "\n"
-        )
+        if (3, 11) <= sys.version_info[:2]:
+            expected = (
+                "Traceback (most recent call last):\n  File "
+                + green('"<input>"')
+                + ", line "
+                + bold(magenta("1"))
+                + ", in "
+                + cyan("<module>")
+                + "\n    gfunc()"
+                + "\n     ^^^^^\n"
+                + bold(red("NameError"))
+                + ": "
+                + cyan(global_not_found)
+                + "\n"
+            )
+        else:
+            expected = (
+                "Traceback (most recent call last):\n  File "
+                + green('"<input>"')
+                + ", line "
+                + bold(magenta("1"))
+                + ", in "
+                + cyan("<module>")
+                + "\n    gfunc()\n"
+                + bold(red("NameError"))
+                + ": "
+                + cyan(global_not_found)
+                + "\n"
+            )
 
         self.assertMultiLineEqual(str(plain("").join(a)), str(expected))
         self.assertEqual(plain("").join(a), expected)
