Merge pull request #12093 from yangyangxcf/catch_memory_error

catch `MemoryError` raised by `compile`

IPython/core/async_helpers.py

@@ -146,20 +146,20 b' def _should_be_async(cell: str) -> bool:'
 
     If it works, assume it should be async. Otherwise Return False.
 
-    Not handled yet: If the block of code has a return statement as  the top
+    Not handled yet: If the block of code has a return statement as the top
     level, it will be seen as async. This is a know limitation.
     """
     if sys.version_info > (3, 8):
         try:
             code = compile(cell, "<>", "exec", flags=getattr(ast,'PyCF_ALLOW_TOP_LEVEL_AWAIT', 0x0))
             return inspect.CO_COROUTINE & code.co_flags == inspect.CO_COROUTINE
-        except SyntaxError:
+        except (SyntaxError, MemoryError):
             return False
     try:
         # we can't limit ourself to ast.parse, as it __accepts__ to parse on
         # 3.7+, but just does not _compile_
         code = compile(cell, "<>", "exec")
-    except SyntaxError:
+    except (SyntaxError, MemoryError):
         try:
             parse_tree = _async_parse_cell(cell)
 
@@ -167,7 +167,7 b' def _should_be_async(cell: str) -> bool:'
             v = _AsyncSyntaxErrorVisitor()
             v.visit(parse_tree)
 
-        except SyntaxError:
+        except (SyntaxError, MemoryError):
             return False
         return True
     return False

IPython/core/tests/test_async_helpers.py

@@ -276,6 +276,10 b' class AsyncTest(TestCase):'
         """
         )
 
+    def test_memory_error(self):
+        with self.assertRaises(MemoryError):
+            iprc("(" * 200 + ")" * 200)
+
     @skip_without('curio')
     def test_autoawait_curio(self):
         iprc("%autoawait curio")

IPython/core/tests/test_ultratb.py

@@ -253,6 +253,13 b' bar()'
                 ip.showsyntaxerror()
 
 
+class MemoryErrorTest(unittest.TestCase):
+    def test_memoryerror(self):
+        memoryerror_code = "(" * 200 + ")" * 200
+        with tt.AssertPrints("MemoryError"):
+            ip.run_cell(memoryerror_code)
+
+
 class Python3ChainedExceptionsTest(unittest.TestCase):
     DIRECT_CAUSE_ERROR_CODE = """
 try: