Search in sources :

Example 1 with Rule

use of com.vaticle.typedb.core.logic.Rule in project grakn by graknlabs.

the class UnifyAttributeConcludableTest method literal_predicates_unify_and_filter_answers.

@Test
public void literal_predicates_unify_and_filter_answers() {
    String conjunction = "{ $a = 'john'; }";
    Set<Concludable> concludables = Concludable.create(resolvedConjunction(conjunction, logicMgr));
    Concludable.Attribute queryConcludable = concludables.iterator().next().asAttribute();
    Rule rule = createRule("isa-rule", "{ $x isa person; }", "$x has first-name \"john\"", logicMgr);
    List<Unifier> unifiers = queryConcludable.unify(rule.conclusion(), conceptMgr).toList();
    assertEquals(1, unifiers.size());
    Unifier unifier = unifiers.get(0);
    Map<String, Set<String>> result = getStringMapping(unifier.mapping());
    Map<String, Set<String>> expected = new HashMap<String, Set<String>>() {

        {
            put("$a", set("$_0"));
        }
    };
    assertEquals(expected, result);
    // test requirements
    assertEquals(0, unifier.requirements().types().size());
    assertEquals(1, unifier.requirements().isaExplicit().size());
    assertEquals(1, unifier.requirements().predicates().size());
    // test forward unification can reject an invalid partial answer
    ConceptMap unUnified = new ConceptMap(map(pair(Identifier.Variable.name("a"), instanceOf("first-name", "bob"))));
    assertFalse(unifier.unify(unUnified).isPresent());
    // test filter allows a valid answer
    Map<Identifier.Variable, Concept> concepts = map(pair(Identifier.Variable.anon(0), instanceOf("first-name", "john")));
    FunctionalIterator<ConceptMap> unified = unifier.unUnify(concepts, new Unifier.Requirements.Instance(map()));
    assertTrue(unified.hasNext());
    assertEquals(1, unified.next().concepts().size());
    concepts = map(pair(Identifier.Variable.anon(0), instanceOf("first-name", "abe")));
    unified = unifier.unUnify(concepts, new Unifier.Requirements.Instance(map()));
    assertFalse(unified.hasNext());
}
Also used : Concept(com.vaticle.typedb.core.concept.Concept) Set(java.util.Set) Variable(com.vaticle.typedb.core.pattern.variable.Variable) HashMap(java.util.HashMap) Rule(com.vaticle.typedb.core.logic.Rule) Util.createRule(com.vaticle.typedb.core.logic.resolvable.Util.createRule) ConceptMap(com.vaticle.typedb.core.concept.answer.ConceptMap) Test(org.junit.Test)

Example 2 with Rule

use of com.vaticle.typedb.core.logic.Rule in project grakn by graknlabs.

the class UnifyAttributeConcludableTest method variable_predicates_unify_value_conclusions.

@Test
public void variable_predicates_unify_value_conclusions() {
    String conjunction = "{ $x > $y; }";
    Set<Concludable> concludables = Concludable.create(resolvedConjunction(conjunction, logicMgr));
    assertEquals(2, concludables.size());
    Concludable.Attribute queryConcludable = concludables.iterator().next().asAttribute();
    // Get the name of the attribute variable from whichever concludable is chosen
    String var = Iterators.iterate(concludables).map(Concludable::pattern).flatMap(c -> Iterators.iterate(c.variables())).filter(Variable::isThing).map(v -> v.reference().syntax()).first().get();
    Rule rule = createRule("isa-rule", "{ $x isa person; }", "$x has first-name \"john\"", logicMgr);
    List<Unifier> unifiers = queryConcludable.unify(rule.conclusion(), conceptMgr).toList();
    assertEquals(1, unifiers.size());
    Unifier unifier = unifiers.get(0);
    Map<String, Set<String>> result = getStringMapping(unifier.mapping());
    Map<String, Set<String>> expected = new HashMap<String, Set<String>>() {

        {
            put(var, set("$_0"));
        }
    };
    assertEquals(expected, result);
    // test requirements
    assertEquals(0, unifier.requirements().types().size());
    assertEquals(1, unifier.requirements().isaExplicit().size());
    assertEquals(0, unifier.requirements().predicates().size());
    Rule rule2 = createRule("isa-rule-2", "{ " + var + " isa person; }", "(employee: " + var + ") isa employment", logicMgr);
    unifiers = queryConcludable.unify(rule2.conclusion(), conceptMgr).toList();
    assertEquals(0, unifiers.size());
}
Also used : Set(java.util.Set) HashMap(java.util.HashMap) Rule(com.vaticle.typedb.core.logic.Rule) Util.createRule(com.vaticle.typedb.core.logic.resolvable.Util.createRule) Test(org.junit.Test)

Example 3 with Rule

use of com.vaticle.typedb.core.logic.Rule in project grakn by graknlabs.

the class UnifyIsaConcludableTest method isa_variable_unifies_relation_variable_type.

@Test
public void isa_variable_unifies_relation_variable_type() {
    String conjunction = "{ $a isa $t; }";
    Set<Concludable> concludables = Concludable.create(resolvedConjunction(conjunction, logicMgr));
    Concludable.Isa queryConcludable = concludables.iterator().next().asIsa();
    Rule rule = createRule("isa-rule", "{ $x isa person; $role-type type employment:employee; $rel-type relates $role-type; }", "($role-type: $x) isa $rel-type", logicMgr);
    List<Unifier> unifiers = queryConcludable.unify(rule.conclusion(), conceptMgr).toList();
    assertEquals(1, unifiers.size());
    Unifier unifier = unifiers.get(0);
    Map<String, Set<String>> result = getStringMapping(unifier.mapping());
    HashMap<String, Set<String>> expected = new HashMap<String, Set<String>>() {

        {
            put("$a", set("$_0"));
            put("$t", set("$rel-type"));
        }
    };
    assertEquals(expected, result);
    // test requirements
    assertEquals(1, unifier.requirements().types().size());
    assertEquals(1, unifier.requirements().isaExplicit().size());
    assertEquals(0, unifier.requirements().predicates().size());
}
Also used : Set(java.util.Set) HashMap(java.util.HashMap) Rule(com.vaticle.typedb.core.logic.Rule) Util.createRule(com.vaticle.typedb.core.logic.resolvable.Util.createRule) Test(org.junit.Test)

Example 4 with Rule

use of com.vaticle.typedb.core.logic.Rule in project grakn by graknlabs.

the class UnifyIsaConcludableTest method isa_variable_unifies_rule_relation_exact.

@Test
public void isa_variable_unifies_rule_relation_exact() {
    String conjunction = "{ $a isa $t; }";
    Set<Concludable> concludables = Concludable.create(resolvedConjunction(conjunction, logicMgr));
    Concludable.Isa queryConcludable = concludables.iterator().next().asIsa();
    Rule rule = createRule("isa-rule", "{ $x isa person; }", "(employee: $x) isa employment", logicMgr);
    List<Unifier> unifiers = queryConcludable.unify(rule.conclusion(), conceptMgr).toList();
    assertEquals(1, unifiers.size());
    Unifier unifier = unifiers.get(0);
    Map<String, Set<String>> result = getStringMapping(unifier.mapping());
    HashMap<String, Set<String>> expected = new HashMap<String, Set<String>>() {

        {
            put("$a", set("$_0"));
            put("$t", set("$_employment"));
        }
    };
    assertEquals(expected, result);
    // test requirements
    assertEquals(1, unifier.requirements().types().size());
    assertEquals(1, unifier.requirements().isaExplicit().size());
    assertEquals(0, unifier.requirements().predicates().size());
}
Also used : Set(java.util.Set) HashMap(java.util.HashMap) Rule(com.vaticle.typedb.core.logic.Rule) Util.createRule(com.vaticle.typedb.core.logic.resolvable.Util.createRule) Test(org.junit.Test)

Example 5 with Rule

use of com.vaticle.typedb.core.logic.Rule in project grakn by graknlabs.

the class UnifyIsaConcludableTest method isa_exact_unifies_rule_relation_variable.

@Test
public void isa_exact_unifies_rule_relation_variable() {
    String conjunction = "{ $a isa relation; }";
    Set<Concludable> concludables = Concludable.create(resolvedConjunction(conjunction, logicMgr));
    Concludable.Isa queryConcludable = concludables.iterator().next().asIsa();
    Rule rule = createRule("isa-rule", "{ $x isa person; $role-type type employment:employee; $rel-type relates $role-type; }", "($role-type: $x) isa $rel-type", logicMgr);
    List<Unifier> unifiers = queryConcludable.unify(rule.conclusion(), conceptMgr).toList();
    assertEquals(1, unifiers.size());
    Unifier unifier = unifiers.get(0);
    Map<String, Set<String>> result = getStringMapping(unifier.mapping());
    HashMap<String, Set<String>> expected = new HashMap<String, Set<String>>() {

        {
            put("$a", set("$_0"));
        }
    };
    assertEquals(expected, result);
    // test requirements
    assertEquals(1, unifier.requirements().types().size());
    assertEquals(1, unifier.requirements().isaExplicit().size());
    assertEquals(set(Label.of("employment")), unifier.requirements().isaExplicit().get(Identifier.Variable.name("a")));
    assertEquals(0, unifier.requirements().predicates().size());
    // test filter allows a valid answer
    Map<Identifier.Variable, Concept> concepts = map(pair(Identifier.Variable.anon(0), instanceOf("employment")), pair(Identifier.Variable.name("rel-type"), conceptMgr.getThingType("employment")));
    FunctionalIterator<ConceptMap> unified = unifier.unUnify(concepts, new Unifier.Requirements.Instance(map()));
    assertTrue(unified.hasNext());
    assertEquals(1, unified.next().concepts().size());
    // filter out invalid type
    concepts = map(pair(Identifier.Variable.anon(0), instanceOf("age")), pair(Identifier.Variable.name("rel-type"), conceptMgr.getThingType("age")));
    unified = unifier.unUnify(concepts, new Unifier.Requirements.Instance(map()));
    assertFalse(unified.hasNext());
}
Also used : Concept(com.vaticle.typedb.core.concept.Concept) Set(java.util.Set) HashMap(java.util.HashMap) Rule(com.vaticle.typedb.core.logic.Rule) Util.createRule(com.vaticle.typedb.core.logic.resolvable.Util.createRule) ConceptMap(com.vaticle.typedb.core.concept.answer.ConceptMap) Test(org.junit.Test)

Aggregations

Rule (com.vaticle.typedb.core.logic.Rule)38 Test (org.junit.Test)37 Util.createRule (com.vaticle.typedb.core.logic.resolvable.Util.createRule)35 HashMap (java.util.HashMap)32 Set (java.util.Set)30 Concept (com.vaticle.typedb.core.concept.Concept)27 ConceptMap (com.vaticle.typedb.core.concept.answer.ConceptMap)27 ConceptManager (com.vaticle.typedb.core.concept.ConceptManager)15 Relation (com.vaticle.typedb.core.concept.thing.Relation)15 Thing (com.vaticle.typedb.core.concept.thing.Thing)15 RelationType (com.vaticle.typedb.core.concept.type.RelationType)15 LogicManager (com.vaticle.typedb.core.logic.LogicManager)15 Variable (com.vaticle.typedb.core.traversal.common.Identifier.Variable)15 HashSet (java.util.HashSet)15 Map (java.util.Map)15 AttributeType (com.vaticle.typedb.core.concept.type.AttributeType)14 RoleType (com.vaticle.typedb.core.concept.type.RoleType)14 Lists (com.google.common.collect.Lists)13 Collections.list (com.vaticle.typedb.common.collection.Collections.list)13 Collections.map (com.vaticle.typedb.common.collection.Collections.map)13