Optimize override_mode in assembler, add more test cases

1 files changed, 13 insertions, 0 deletions

diff --git a/test/e2e/std_bump_alloc/test_bump_alloc.py b/test/e2e/std_bump_alloc/test_bump_alloc.py
index 717a122..b2274f8 100644
--- a/test/e2e/std_bump_alloc/test_bump_alloc.py
+++ b/test/e2e/std_bump_alloc/test_bump_alloc.py
@@ -20,3 +20,16 @@ def test_bump_alloc():
 
   returned_value = machine.rg.value
   assert returned_value == heap_next + 1
+
+def test_bump_alloc_overlap():
+  machine = assemble_and_run_until_halted(
+    "test/e2e/std_bump_alloc/run_bump_alloc_overlap.atk16"
+  )
+
+  heap_next_p = 0xE7EF
+  heap_next = machine.mem_read(heap_next_p)
+  assert heap_next == heap_next_p - 1 # alloc failed because of overlap
+                                      # so pointer should not have moved
+
+  returned_value = machine.rg.value
+  assert returned_value == 0          # returns 0 on failure