fix tests on 32-bit platforms

Author: picnixz

Lib/test/test_frame.py

@@ -1,5 +1,6 @@
 import copy
 import operator
+import platform
 import re
 import sys
 import textwrap
@@ -236,9 +237,14 @@ def test_sizeof_overflow(self):
         evil_stacksize = int(_testcapi.INT_MAX / ps - fss - co_nlocalsplus)
         # an evil code with a valid (but very large) stack size
         evil_code = f.f_code.replace(co_stacksize=evil_stacksize - 1)
-        frame = _testcapi.frame_new(evil_code, globals(), locals())
-        message = re.escape("size exceeds INT_MAX")
-        self.assertRaisesRegex(OverflowError, message, frame.__sizeof__)
+
+        if sys.maxsize == 2147483647:  # 32-bit machine
+            with self.assertRaises(MemoryError):
+                frame = _testcapi.frame_new(evil_code, globals(), locals())
+        else:
+            frame = _testcapi.frame_new(evil_code, globals(), locals())
+            message = re.escape("size exceeds INT_MAX")
+            self.assertRaisesRegex(OverflowError, message, frame.__sizeof__)
 
 
 class ReprTest(unittest.TestCase):

Lib/test/test_generators.py

@@ -288,10 +288,14 @@ def f(): yield
 
         if support.Py_GIL_DISABLED:
             self.skipTest("segmentation fault on free-threaded builds")
-        # the following crashes on free-threaded builds for now
-        evil_gi_func = types.FunctionType(evil, {})()
-        message = re.escape("size exceeds INT_MAX")
-        self.assertRaisesRegex(OverflowError, message, evil_gi.__sizeof__)
+        elif sys.maxsize == 2147483647:  # 32-bit machine
+            with self.assertRaises(MemoryError):
+                evil_gi = types.FunctionType(evil, {})()
+        else:
+            # the following crashes on free-threaded builds for now
+            evil_gi = types.FunctionType(evil, {})()
+            message = re.escape("size exceeds INT_MAX")
+            self.assertRaisesRegex(OverflowError, message, evil_gi.__sizeof__)
 
 
 class ModifyUnderlyingIterableTest(unittest.TestCase):

Objects/codeobject.c

@@ -439,7 +439,9 @@ _PyCode_Validate(struct _PyCodeConstructor *con)
     /*
      * The framesize = stacksize + nlocalsplus + FRAME_SPECIALS_SIZE is used
      * as framesize * sizeof(PyObject *) and assumed to be < INT_MAX. Thus,
-     * we need to dynamically limit the value of stacksize.
+     * we need to dynamically limit the value of stacksize. Note that this
+     * usually prevents crashes due to assertions but a MemoryError may still
+     * be triggered later.
      *
      * See https://github.com/python/cpython/issues/126119 for details.
      */