Add free variables to define function instruction

1 files changed, 2 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
index 4e0be235..1fa6c41a 100644
--- a/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
+++ b/src/test/java/org/shedlang/compiler/stackir/StackCompilerFunctionDefinitionTests.java
@@ -17,6 +17,7 @@ public class StackCompilerFunctionDefinitionTests {
             .withName("f")
             .addPositionalParam("x", Typed.typeLevelReference("String", Types.STRING))
             .addNamedParam("y", Typed.typeLevelReference("Int", Types.INT))
+            .addFreeVariable("z")
             .addStatement(Typed.expressionStatement(Typed.stringLiteral("hello")))
             .build();
 
@@ -32,6 +33,7 @@ public class StackCompilerFunctionDefinitionTests {
                 has("namedParams", x -> x.namedParams(), isSequence(
                     isDefineFunctionInstructionParam("y")
                 )),
+                has("freeVariables", x -> x.freeVariables(), containsExactly(equalTo("z"))),
                 has("body", x -> x.body(), isSequence(
                     isPushStringInstruction("hello"),
                     isDiscardInstruction()