tests/stress/bytecode_limit.py: Make test more robust with low memory.

A target may have enough RAM to run the n=433 test but then run out of RAM on the n=432 test. So allow the test to skip on the n=432 case before it prints any output. Signed-off-by: Damien George <[email protected]>
samdeleu · Jul 5, 2024 · 358e501 · 358e501
1 parent 633586a
commit 358e501
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tests/stress/bytecode_limit.py b/tests/stress/bytecode_limit.py
@@ -3,14 +3,18 @@
 body = " with f()()() as a:\n  try:\n   f()()()\n  except Exception:\n   pass\n"
 
 # Test overflow of jump offset.
+# Print results at the end in case an intermediate value of n fails with MemoryError.
+results = []
 for n in (433, 432, 431, 430):
     try:
         exec("cond = 0\nif cond:\n" + body * n + "else:\n print('cond false')\n")
+        results.append((n, "ok"))
     except MemoryError:
         print("SKIP")
         raise SystemExit
     except RuntimeError:
-        print("RuntimeError")
+        results.append((n, "RuntimeError"))
+print(results)
 
 # Test changing size of code info (source line/bytecode mapping) due to changing
 # bytecode size in the final passes.  This test is very specific to how the

diff --git a/tests/stress/bytecode_limit.py.exp b/tests/stress/bytecode_limit.py.exp
@@ -1,5 +1,4 @@
-RuntimeError
-RuntimeError
 cond false
 cond false
+[(433, 'RuntimeError'), (432, 'RuntimeError'), (431, 'ok'), (430, 'ok')]
 [123]