Update test_run.py

python · Locked-chess-official · Jun 15, 2025 · Jun 15, 2025 · Jun 15, 2025 · Jun 15, 2025
commit 2b182013df7bed9154d3ab238750cceac26805b9
@@ -45,17 +45,24 @@ def __eq__(self, other):
             ('int.reel', AttributeError,
                  "type object 'int' has no attribute 'reel'. "
                  "Did you mean: 'real'?\n"),
+            ('raise NameError("123\\n456")', NameError, "123\n456"),
             )
 
     @force_not_colorized
     def test_get_message(self):
         for code, exc, msg in self.data:
             with self.subTest(code=code):
                 try:
-                    eval(compile(code, '', 'eval'))
+                    if "raise " not in code:
+                        eval(compile(code, '', 'eval'))
+                    else:
+                        exec(compile(code, '', 'exec'))
                 except exc:
-                    typ, val, tb = sys.exc_info()
-                    actual = run.get_message_lines(typ, val, tb)[0]
+                    if "raise " in code:
+                        actual = run.print_exception(in_test=True)
+                    else:
+                        typ, val, tb = sys.exc_info()
+                        actual = run.get_message_lines(typ, val, tb)[0]
                     expect = f'{exc.__name__}: {msg}'
                     self.assertEqual(actual, expect)