From 0d531b61a74872da604aa35e367392b4f8c59219 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 10 Jun 2023 15:22:08 +0100 Subject: Type check function definition params --- .../TypeCheckerFunctionDefinitionTests.java | 29 +++++++++++++++++++++- 1 file changed, 28 insertions(+), 1 deletion(-) (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java index 889698ac..1215389b 100644 --- a/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java +++ b/src/test/java/org/shedlang/compiler/typechecker/TypeCheckerFunctionDefinitionTests.java @@ -36,10 +36,37 @@ public class TypeCheckerFunctionDefinitionTests { ); } + @Test + public void paramsAreTypeChecked() { + var untyped = UntypedFunctionDefinitionNode.builder() + .addPositionalParam("x", Untyped.typeLevelReference("String")) + .addNamedParam("y", Untyped.typeLevelReference("Int")) + .withReturnType(Untyped.typeLevelReference("Unit")) + .build(); + var context = TypeCheckerContext.EMPTY + .enter(P.map( + Map.entry("Int", Types.metaType(Types.INT)), + Map.entry("String", Types.metaType(Types.STRING)), + Map.entry("Unit", Types.metaType(Types.UNIT)) + )); + + var typed = TypeChecker.typeCheckNamespaceStatement(untyped, context); + + assertThat( + typed, + isTypedFunctionDefinitionNode() + .withParams( + isTypedParamsNode( + isSequence(isTypedParamNode("x", isTypedTypeLevelReferenceNode("String", Types.STRING))), + isSequence(isTypedParamNode("y", isTypedTypeLevelReferenceNode("Int", Types.INT))) + ) + ) + ); + } + @Test public void bodyIsTypeChecked() { var untyped = UntypedFunctionDefinitionNode.builder() - .withName("f") .withReturnType(Untyped.typeLevelReference("Unit")) .addStatement(Untyped.expressionStatement(Untyped.stringLiteral("hello"))) .build(); -- cgit v1.2.3