Make @include idempotent

3 files changed, 24 insertions, 0 deletions

diff --git a/test/e2e/include_idempotency/p1.atk16 b/test/e2e/include_idempotency/p1.atk16
new file mode 100644
index 0000000..f9f6363
--- /dev/null
+++ b/test/e2e/include_idempotency/p1.atk16
@@ -0,0 +1,4 @@
+@include %bootstrap
+
+@label main
+  hlt
diff --git a/test/e2e/include_idempotency/p2.atk16 b/test/e2e/include_idempotency/p2.atk16
new file mode 100644
index 0000000..c6d46b9
--- /dev/null
+++ b/test/e2e/include_idempotency/p2.atk16
@@ -0,0 +1,6 @@
+@include %bootstrap
+@include %bootstrap
+@include %bootstrap
+
+@label main
+  hlt
diff --git a/test/e2e/include_idempotency/test_include_idempotent.py b/test/e2e/include_idempotency/test_include_idempotent.py
new file mode 100644
index 0000000..e7b4fc8
--- /dev/null
+++ b/test/e2e/include_idempotency/test_include_idempotent.py
@@ -0,0 +1,14 @@
+import os.path
+from test.utils import assemble_and_run_until_halted
+
+def test_include_idempotent():
+  machine1 = assemble_and_run_until_halted(
+    os.path.join(os.path.dirname(__file__), "p1.atk16")
+  )
+
+  machine2 = assemble_and_run_until_halted(
+    os.path.join(os.path.dirname(__file__), "p2.atk16")
+  )
+
+  for i in range(len(machine1.rom.memory)):
+    assert machine1.rom.memory[i] == machine2.rom.memory[i]