Add instructions after defining function but before defining vars

1 files changed, 3 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
index e0d9ed9d..c758fc57 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
@@ -3,6 +3,7 @@ package org.shedlang.compiler.patio;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.shedlang.compiler.ast.typed.FreeVariables;
+import org.shedlang.compiler.patio.instructions.Patio;
 import org.shedlang.compiler.sources.Sources;
 import org.shedlang.compiler.stackir.instructions.DefineFunctionInstruction;
 import org.shedlang.compiler.stackir.instructions.Instructions;
@@ -74,7 +75,9 @@ public class PatioCodeGenDefineFunctionTests {
         var context = PatioCodeGenContext.EMPTY;
 
         context = PatioCodeGen.generateInstruction(instruction, context);
+        context = context.addInstruction(Patio.localStore(Patio.localRef("a"), Patio.int_(42)));
         context = context.addStackIrParam("a");
+        context = context.addInstruction(Patio.localStore(Patio.localRef("b"), Patio.int_(47)));
         context = context.addStackIrParam("b");
 
         snapshotter.assertSnapshot(serialize(context));