From 1c0b88e256de6b4ea1b0c34ad525dfdad471f5ee Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Mon, 25 Dec 2023 11:54:21 +0000 Subject: Implement type substitution for constructed types --- .../compiler/types/TypeSubstitutionTests.java | 28 ++++++++++++++++++++++ 1 file changed, 28 insertions(+) (limited to 'src/test/java') diff --git a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java index cd676d71..e9b04a95 100644 --- a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java +++ b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java @@ -35,6 +35,34 @@ public class TypeSubstitutionTests { assertThat(result, equalTo(Types.BOTTOM)); } + @Test + public void substitutingConstructedTypeSubstitutesArgs() { + var typeSubstitutionMap = TypeSubstitutionMap.of( + Map.entry(TypeParam.forStructType(STRUCT_TYPE, "T"), Types.INT) + ); + + var result = TypeSubstitution.substitute( + Types.constructedType( + Types.typeConstructor( + List.of(TypeParam.forStructType(STRUCT_TYPE_2, "T")), + STRUCT_TYPE_2 + ), + List.of(TypeParam.forStructType(STRUCT_TYPE, "T")) + ), + typeSubstitutionMap + ); + + assertThat(result, equalTo( + Types.constructedType( + Types.typeConstructor( + List.of(TypeParam.forStructType(STRUCT_TYPE_2, "T")), + STRUCT_TYPE_2 + ), + List.of(Types.INT) + ) + )); + } + @Test public void substitutingFunctionSubstitutesPositionalArgs() { var typeSubstitutionMap = TypeSubstitutionMap.of( -- cgit v1.2.3