From cec6afb8be6b8a040a2483e621646fd6890b50c1 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Fri, 19 Jun 2026 18:55:14 +0100 Subject: Store enum variants in TypesInfo --- .../compiler/typechecker/TypeCheckerEnumDefinitionTests.java | 6 ++++++ .../compiler/typechecker/TypeCheckerSumDefinitionTests.java | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) (limited to 'src/test/java') diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java index f8a8fb4..ac85b7d 100644 --- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java +++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java @@ -57,5 +57,11 @@ public class TypeCheckerEnumDefinitionTests { ) ) )); + var typesInfo = globalContext.toTypesInfo(); + var enumType = new EnumType(NamespaceName.of("a", "b"), "X"); + assertThat(typesInfo.enumVariants(enumType), isSequence( + equalTo(new EnumVariant("a")), + equalTo(new EnumVariant("b")) + )); } } diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java index 60c2b9a..57c40f5 100644 --- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java +++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java @@ -76,7 +76,7 @@ public class TypeCheckerSumDefinitionTests { var typesInfo = globalContext.toTypesInfo(); assertThat(typesInfo.variantOf(rectangleType), isSequence(equalTo(sumType))); assertThat(typesInfo.variantOf(triangleType), isSequence(equalTo(sumType))); - assertThat(typesInfo.variants(sumType), isSequence(equalTo(rectangleType), equalTo(triangleType))); + assertThat(typesInfo.sumVariants(sumType), isSequence(equalTo(rectangleType), equalTo(triangleType))); } @Test -- cgit v1.2.3