Store free variables in closure

2 files changed, 8 insertions, 3 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
index ac510295..4f09ff3b 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
@@ -49,7 +49,9 @@ public class PatioCodeGenDefineFunctionTests {
             ),
             Sources.NULL
         );
-        var context = PatioCodeGenContext.EMPTY;
+        var context = PatioCodeGenContext.EMPTY
+            .addStackIrLocal("a").context()
+            .addStackIrLocal("b").context();
 
         var result = PatioCodeGen.generateInstruction(instruction, context);
 
@@ -73,7 +75,9 @@ public class PatioCodeGenDefineFunctionTests {
             ),
             Sources.NULL
         );
-        var context = PatioCodeGenContext.EMPTY;
+        var context = PatioCodeGenContext.EMPTY
+            .addStackIrLocal("a").context()
+            .addStackIrLocal("b").context();
 
         var result = PatioCodeGen.generateInstruction(instruction, context);
 
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenLocalLoadTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenLocalLoadTests.java
index 8ad1ba64..643c6d72 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenLocalLoadTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenLocalLoadTests.java
@@ -15,7 +15,8 @@ public class PatioCodeGenLocalLoadTests {
     @Test
     public void localLoadPushesLocalOntoStack(Snapshotter snapshotter) throws IOException {
         var stackIr = Instructions.localLoad("x");
-        var context = PatioCodeGenContext.EMPTY;
+        var context = PatioCodeGenContext.EMPTY
+            .addStackIrLocal("x").context();
 
         var result = PatioCodeGen.generateInstruction(stackIr, context);