From 28049a39ca32332a01c4dd1857384372c10b464f Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 2 Mar 2024 14:49:45 +0000 Subject: Directly test effect subsitution --- .../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 2e2bd08e..210e4c18 100644 --- a/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java +++ b/src/test/java/org/shedlang/compiler/types/TypeSubstitutionTests.java @@ -13,6 +13,34 @@ public class TypeSubstitutionTests { private static final SimpleStructType STRUCT_TYPE = Types.struct(NAMESPACE_NAME, "Struct"); private static final SimpleStructType STRUCT_TYPE_2 = Types.struct(NAMESPACE_NAME, "Struct2"); + @Test + public void substitutingEffectParamDoesNothingIfEffectParamIsNotInSubstitutionMap() { + var typeSubstitutionMap = TypeSubstitutionMap.of( + Map.entry(EffectParam.forStructType(STRUCT_TYPE, "E"), Effects.IO) + ); + + var result = TypeSubstitution.substituteEffect( + EffectParam.forStructType(STRUCT_TYPE_2, "E"), + typeSubstitutionMap + ); + + assertThat(result, equalTo(EffectParam.forStructType(STRUCT_TYPE_2, "E"))); + } + + @Test + public void substitutingEffectParamChangesEffectParamToCorrespondingEffect() { + var typeSubstitutionMap = TypeSubstitutionMap.of( + Map.entry(EffectParam.forStructType(STRUCT_TYPE, "E"), Effects.IO) + ); + + var result = TypeSubstitution.substituteEffect( + EffectParam.forStructType(STRUCT_TYPE, "E"), + typeSubstitutionMap + ); + + assertThat(result, equalTo(Effects.IO)); + } + @Test public void substitutingBoolDoesNothing() { var typeSubstitutionMap = TypeSubstitutionMap.of( -- cgit v1.2.3