Add variant to define-function instruction

3 files changed, 7 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 b5b9d0b5..bb8fb3ab 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java
@@ -29,6 +29,7 @@ public class PatioCodeGenDefineFunctionTests {
                 Shid.pushUnit(),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var context = PatioCodeGenContext.EMPTY;
@@ -49,6 +50,7 @@ public class PatioCodeGenDefineFunctionTests {
                 Shid.localLoad("a"),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var context = PatioCodeGenContext.EMPTY
@@ -71,6 +73,7 @@ public class PatioCodeGenDefineFunctionTests {
                 Shid.localLoad("a"),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var context = PatioCodeGenContext.EMPTY
@@ -105,6 +108,7 @@ public class PatioCodeGenDefineFunctionTests {
                 Shid.localLoad("a"),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var context = PatioCodeGenContext.EMPTY
@@ -134,6 +138,7 @@ public class PatioCodeGenDefineFunctionTests {
                 Shid.intAdd(),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var context = PatioCodeGenContext.EMPTY
diff --git a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenNamespaceTests.java b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenNamespaceTests.java
index 852eda6f..c7dcf094 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioCodeGenNamespaceTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioCodeGenNamespaceTests.java
@@ -33,6 +33,7 @@ public class PatioCodeGenNamespaceTests {
                 Shid.pushUnit(),
                 Shid.return_()
             ),
+            ShidDefineFunction.Variant.FUNCTION,
             Sources.NULL
         );
         var instructions = List.<ShidInstruction>of(instruction);
diff --git a/src/test/java/org/shedlang/compiler/shid/InstructionSerializerTests.java b/src/test/java/org/shedlang/compiler/shid/InstructionSerializerTests.java
index 9f6bdcf1..f423442a 100644
--- a/src/test/java/org/shedlang/compiler/shid/InstructionSerializerTests.java
+++ b/src/test/java/org/shedlang/compiler/shid/InstructionSerializerTests.java
@@ -55,6 +55,7 @@ public class InstructionSerializerTests {
         assertThat(string, equalTo("""
             (
                 define-function
+                variant="function"
                 name="f"
                 positional-params=("p1" "p2")
                 named-params=("n1" "n2")