From 5ab307fa798626d7f852ebec65f87e2af7487c53 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 29 Oct 2023 18:23:36 +0000
Subject: Infer return type of function expressions properly

---
 .../compiler/typechecker/TypeCheckerFunctionExpressionTests.java      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/test/java')

diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionExpressionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionExpressionTests.java
index 80627018..35d61efd 100644
--- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionExpressionTests.java
+++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionExpressionTests.java
@@ -43,7 +43,7 @@ public class TypeCheckerFunctionExpressionTests {
     public void returnTypeCanBeInferred() {
         var untyped = UntypedFunctionExpressionNode.builder()
             .withoutReturnType()
-            .addStatement(Untyped.expressionStatement(Untyped.stringLiteral("")))
+            .addStatement(Untyped.functionReturn(Untyped.stringLiteral("")))
             .build();
         var context = TypeCheckerContext.EMPTY;
 
@@ -52,7 +52,7 @@ public class TypeCheckerFunctionExpressionTests {
         assertThat(
             typed.value(),
             isTypedFunctionExpressionNode()
-                .withType(new FunctionType(List.of(), List.of(), Effects.NONE, Types.UNIT))
+                .withType(new FunctionType(List.of(), List.of(), Effects.NONE, Types.STRING))
         );
     }
 }
-- 
cgit v1.2.3