fix: abstract type chain elimination

1 files changed, 19 insertions, 0 deletions

diff --git a/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeHierarchyTest.java b/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeHierarchyTest.java
index 1f922a79..9fbf2334 100644
--- a/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeHierarchyTest.java
+++ b/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/typehierarchy/TypeHierarchyTest.java
@@ -13,6 +13,7 @@ import tools.refinery.store.representation.TruthValue;
 import java.util.Set;
 
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.hasEntry;
 import static org.junit.jupiter.api.Assertions.assertAll;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
@@ -201,4 +202,22 @@ class TypeHierarchyTest {
 
                 assertThrows(IllegalArgumentException.class, builder::build);
         }
+
+        @Test
+        void chainedEliminationTest() {
+                var a1 = new PartialRelation("A1", 1);
+                var a2 = new PartialRelation("A2", 1);
+                var c1 = new PartialRelation("C1", 1);
+
+                var sut = TypeHierarchy.builder()
+                                .type(a1, true)
+                                .type(a2, true, a1)
+                                .type(c1, a2)
+                                .build();
+
+                assertAll(
+                                () -> assertThat(sut.getEliminatedTypes(), hasEntry(a1, c1)),
+                                () -> assertThat(sut.getEliminatedTypes(), hasEntry(a2, c1))
+                );
+        }
 }