From 52bc86e76ad85a9af69f50bbc8ed9a41af7c4eeb Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 2 Dec 2023 20:27:14 +0000
Subject: Add variant to define-function instruction

---
 .../org/shedlang/compiler/patio/PatioCodeGenDefineFunctionTests.java | 5 +++++
 .../java/org/shedlang/compiler/patio/PatioCodeGenNamespaceTests.java | 1 +
 .../java/org/shedlang/compiler/shid/InstructionSerializerTests.java  | 1 +
 3 files changed, 7 insertions(+)

(limited to 'src/test/java')

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")
-- 
cgit v1.2.3