Make test_traceback version agnostic

sebastinas · sebastinas · commit 27fc6a0fb271 · 2026-03-26T15:45:47.000+01:00
diff --git a/bpython/test/test_interpreter.py b/bpython/test/test_interpreter.py
@@ -48,59 +48,9 @@ def gfunc():
 
         i.runsource("gfunc()")
 
-        global_not_found = "name 'gfunc' is not defined"
-
-        if (
-            sys.version_info[:2] == (3, 13)
-            and sys.version_info[:3] >= (3, 13, 12)
-        ) or sys.version_info[:3] >= (3, 14, 3):
-            expected = expected = (
-                "Traceback (most recent call last):\n  File "
-                + green('"<input>"')
-                + ", line "
-                + bold(magenta("1"))
-                + ", in "
-                + cyan("<module>")
-                + "\n"
-                + bold(red("NameError"))
-                + ": "
-                + cyan(global_not_found)
-                + "\n"
-            )
-        elif (3, 13) <= sys.version_info[:2] or pypy:
-            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     ^^^^^\n"
-                + bold(red("NameError"))
-                + ": "
-                + cyan(global_not_found)
-                + "\n"
-            )
-
-        a = i.a
-        self.assertMultiLineEqual(str(expected), str(plain("").join(a)))
-        self.assertEqual(expected, plain("").join(a))
+        a = str(plain("").join(i.a))
+        self.assertIn("name 'gfunc' is not defined", a)
+        self.assertIn("NameErro", a)
 
     def test_getsource_works_on_interactively_defined_functions(self):
         source = "def foo(x):\n    return x + 1\n"
