diff options
Diffstat (limited to 'subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/containment/DefaultContainmentTest.java')
-rw-r--r-- | subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/containment/DefaultContainmentTest.java | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/containment/DefaultContainmentTest.java b/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/containment/DefaultContainmentTest.java new file mode 100644 index 00000000..e80f9744 --- /dev/null +++ b/subprojects/store-reasoning/src/test/java/tools/refinery/store/reasoning/translator/containment/DefaultContainmentTest.java | |||
@@ -0,0 +1,106 @@ | |||
1 | /* | ||
2 | * SPDX-FileCopyrightText: 2024 The Refinery Authors <https://refinery.tools/> | ||
3 | * | ||
4 | * SPDX-License-Identifier: EPL-2.0 | ||
5 | */ | ||
6 | package tools.refinery.store.reasoning.translator.containment; | ||
7 | |||
8 | import org.jetbrains.annotations.NotNull; | ||
9 | import org.junit.jupiter.params.ParameterizedTest; | ||
10 | import org.junit.jupiter.params.provider.ValueSource; | ||
11 | import tools.refinery.logic.term.cardinalityinterval.CardinalityIntervals; | ||
12 | import tools.refinery.logic.term.truthvalue.TruthValue; | ||
13 | import tools.refinery.store.model.ModelStore; | ||
14 | import tools.refinery.store.query.interpreter.QueryInterpreterAdapter; | ||
15 | import tools.refinery.store.reasoning.ReasoningAdapter; | ||
16 | import tools.refinery.store.reasoning.ReasoningStoreAdapter; | ||
17 | import tools.refinery.store.reasoning.literal.Concreteness; | ||
18 | import tools.refinery.store.reasoning.representation.PartialRelation; | ||
19 | import tools.refinery.store.reasoning.seed.ModelSeed; | ||
20 | import tools.refinery.store.reasoning.translator.multiobject.MultiObjectTranslator; | ||
21 | import tools.refinery.store.reasoning.translator.multiplicity.UnconstrainedMultiplicity; | ||
22 | import tools.refinery.store.reasoning.translator.typehierarchy.TypeHierarchy; | ||
23 | import tools.refinery.store.reasoning.translator.typehierarchy.TypeHierarchyTranslator; | ||
24 | import tools.refinery.store.tuple.Tuple; | ||
25 | |||
26 | import java.util.LinkedHashMap; | ||
27 | |||
28 | import static org.hamcrest.MatcherAssert.assertThat; | ||
29 | import static org.hamcrest.Matchers.is; | ||
30 | import static tools.refinery.store.reasoning.translator.containment.ContainmentHierarchyTranslator.CONTAINED_SYMBOL; | ||
31 | import static tools.refinery.store.reasoning.translator.containment.ContainmentHierarchyTranslator.CONTAINS_SYMBOL; | ||
32 | import static tools.refinery.store.reasoning.translator.multiobject.MultiObjectTranslator.COUNT_SYMBOL; | ||
33 | |||
34 | class DefaultContainmentTest { | ||
35 | private final PartialRelation c1 = new PartialRelation("C1", 1); | ||
36 | private final PartialRelation c2 = new PartialRelation("C2", 1); | ||
37 | private final PartialRelation c3 = new PartialRelation("C2", 1); | ||
38 | private final PartialRelation r1 = new PartialRelation("r1", 2); | ||
39 | private final PartialRelation r2 = new PartialRelation("r2", 2); | ||
40 | |||
41 | @ParameterizedTest | ||
42 | @ValueSource(booleans = {false, true}) | ||
43 | void defaultContainmentTest(boolean reverse) { | ||
44 | var typeHierarchy = TypeHierarchy.builder() | ||
45 | .type(CONTAINED_SYMBOL, true) | ||
46 | .type(c1) | ||
47 | .type(c2, CONTAINED_SYMBOL) | ||
48 | .type(c3, CONTAINED_SYMBOL) | ||
49 | .build(); | ||
50 | |||
51 | var containmentHierarchy = getContainmentHierarchy(reverse); | ||
52 | |||
53 | var store = ModelStore.builder() | ||
54 | .with(QueryInterpreterAdapter.builder()) | ||
55 | .with(ReasoningAdapter.builder()) | ||
56 | .with(new MultiObjectTranslator()) | ||
57 | .with(new TypeHierarchyTranslator(typeHierarchy)) | ||
58 | .with(new ContainmentHierarchyTranslator(containmentHierarchy)) | ||
59 | .build(); | ||
60 | |||
61 | var modelSeed = ModelSeed.builder(3) | ||
62 | .seed(COUNT_SYMBOL, builder -> builder.reducedValue(CardinalityIntervals.ONE)) | ||
63 | .seed(CONTAINED_SYMBOL, builder -> builder.reducedValue(TruthValue.UNKNOWN)) | ||
64 | .seed(CONTAINS_SYMBOL, builder -> builder.reducedValue(TruthValue.UNKNOWN)) | ||
65 | .seed(c1, builder -> builder | ||
66 | .reducedValue(TruthValue.UNKNOWN) | ||
67 | .put(Tuple.of(0), TruthValue.TRUE)) | ||
68 | .seed(c2, builder -> builder | ||
69 | .reducedValue(TruthValue.UNKNOWN) | ||
70 | .put(Tuple.of(1), TruthValue.TRUE)) | ||
71 | .seed(c3, builder -> builder | ||
72 | .reducedValue(TruthValue.UNKNOWN) | ||
73 | .put(Tuple.of(2), TruthValue.TRUE)) | ||
74 | .seed(r1, builder -> builder | ||
75 | // The {@code FALSE} value here should not affect {@code r2}. | ||
76 | .reducedValue(TruthValue.FALSE) | ||
77 | .put(Tuple.of(0, 1), TruthValue.TRUE)) | ||
78 | .seed(r2, builder -> builder | ||
79 | .reducedValue(TruthValue.UNKNOWN) | ||
80 | .put(Tuple.of(1, 2), TruthValue.TRUE)) | ||
81 | .build(); | ||
82 | |||
83 | var model = store.getAdapter(ReasoningStoreAdapter.class).createInitialModel(modelSeed); | ||
84 | var reasoningAdapter = model.getAdapter(ReasoningAdapter.class); | ||
85 | var r1Interpretation = reasoningAdapter.getPartialInterpretation(Concreteness.PARTIAL, r1); | ||
86 | var r2Interpretation = reasoningAdapter.getPartialInterpretation(Concreteness.PARTIAL, r2); | ||
87 | |||
88 | assertThat(r1Interpretation.get(Tuple.of(0, 1)), is(TruthValue.TRUE)); | ||
89 | assertThat(r1Interpretation.get(Tuple.of(1, 2)), is(TruthValue.FALSE)); | ||
90 | assertThat(r2Interpretation.get(Tuple.of(0, 1)), is(TruthValue.FALSE)); | ||
91 | assertThat(r2Interpretation.get(Tuple.of(1, 2)), is(TruthValue.TRUE)); | ||
92 | } | ||
93 | |||
94 | private @NotNull LinkedHashMap<PartialRelation, ContainmentInfo> getContainmentHierarchy(boolean reverse) { | ||
95 | // Make sure the order of relations is retained. | ||
96 | var containmentHierarchy = new LinkedHashMap<PartialRelation, ContainmentInfo>(); | ||
97 | if (reverse) { | ||
98 | containmentHierarchy.put(r2, new ContainmentInfo(c2, UnconstrainedMultiplicity.INSTANCE, c3)); | ||
99 | containmentHierarchy.put(r1, new ContainmentInfo(c1, UnconstrainedMultiplicity.INSTANCE, c2)); | ||
100 | } else { | ||
101 | containmentHierarchy.put(r1, new ContainmentInfo(c1, UnconstrainedMultiplicity.INSTANCE, c2)); | ||
102 | containmentHierarchy.put(r2, new ContainmentInfo(c2, UnconstrainedMultiplicity.INSTANCE, c3)); | ||
103 | } | ||
104 | return containmentHierarchy; | ||
105 | } | ||
106 | } | ||