Examples with GraknTx ai.grakn.GraknTx used on opensource projects

Example 56 with GraknTx

use of ai.grakn.GraknTx in project grakn by graknlabs.

the class DegreeTest method testDegreesSimple.

@Test
public void testDegreesSimple() {
    // create instances
    EntityType thingy = tx.putEntityType("thingy");
    EntityType anotherThing = tx.putEntityType("another");
    ConceptId entity1 = thingy.addEntity().getId();
    ConceptId entity2 = thingy.addEntity().getId();
    ConceptId entity3 = thingy.addEntity().getId();
    ConceptId entity4 = anotherThing.addEntity().getId();
    Role role1 = tx.putRole("role1");
    Role role2 = tx.putRole("role2");
    thingy.plays(role1).plays(role2);
    anotherThing.plays(role1).plays(role2);
    RelationshipType related = tx.putRelationshipType("related").relates(role1).relates(role2);
    // relate them
    related.addRelationship().addRolePlayer(role1, tx.getConcept(entity1)).addRolePlayer(role2, tx.getConcept(entity2));
    related.addRelationship().addRolePlayer(role1, tx.getConcept(entity2)).addRolePlayer(role2, tx.getConcept(entity3));
    related.addRelationship().addRolePlayer(role1, tx.getConcept(entity2)).addRolePlayer(role2, tx.getConcept(entity4));
    tx.commit();
    tx = session.open(GraknTxType.READ);
    Map<ConceptId, Long> correctDegrees = new HashMap<>();
    correctDegrees.put(entity1, 1L);
    correctDegrees.put(entity2, 3L);
    correctDegrees.put(entity3, 1L);
    correctDegrees.put(entity4, 1L);
    // compute degrees
    List<Long> list = new ArrayList<>(4);
    long workerNumber = 4L;
    if (GraknTestUtil.usingTinker())
        workerNumber = 1L;
    for (long i = 0L; i < workerNumber; i++) {
        list.add(i);
    }
    tx.close();
    Set<Map<Long, Set<String>>> result = list.parallelStream().map(i -> {
        try (GraknTx graph = session.open(GraknTxType.READ)) {
            return graph.graql().compute().centrality().usingDegree().execute();
        }
    }).collect(Collectors.toSet());
    assertEquals(1, result.size());
    Map<Long, Set<String>> degrees0 = result.iterator().next();
    assertEquals(2, degrees0.size());
    degrees0.forEach((key, value) -> value.forEach(id -> {
        assertTrue(correctDegrees.containsKey(ConceptId.of(id)));
        assertEquals(correctDegrees.get(ConceptId.of(id)), key);
    }));
    try (GraknTx graph = session.open(GraknTxType.READ)) {
        Map<Long, Set<String>> degrees1 = graph.graql().compute().centrality().usingDegree().of("thingy").execute();
        assertEquals(2, degrees1.size());
        assertEquals(2, degrees1.get(1L).size());
        assertEquals(1, degrees1.get(3L).size());
        degrees1.forEach((key, value) -> value.forEach(id -> {
            assertTrue(correctDegrees.containsKey(ConceptId.of(id)));
            assertEquals(correctDegrees.get(ConceptId.of(id)), key);
        }));
        Map<Long, Set<String>> degrees2 = graph.graql().compute().centrality().usingDegree().of("thingy", "related").execute();
        assertEquals(degrees1, degrees2);
        degrees2 = graph.graql().compute().centrality().usingDegree().of().execute();
        assertEquals(degrees0, degrees2);
        // compute degrees on subgraph
        Map<Long, Set<String>> degrees3 = graph.graql().compute().centrality().usingDegree().in("thingy", "related").execute();
        assertEquals(degrees1, degrees3);
        degrees3 = graph.graql().compute().centrality().usingDegree().of("thingy").in("related").execute();
        assertEquals(degrees1, degrees3);
    }
}

Example 57 with GraknTx

use of ai.grakn.GraknTx in project grakn by graknlabs.

the class DegreeTest method testSubIsAccountedForInSubgraph.

@Test
public void testSubIsAccountedForInSubgraph() {
    Role pet = tx.putRole("pet");
    Role owner = tx.putRole("owner");
    Entity person = tx.putEntityType("person").plays(owner).addEntity();
    EntityType animal = tx.putEntityType("animal").plays(pet);
    Entity dog = tx.putEntityType("dog").sup(animal).addEntity();
    tx.putRelationshipType("mans-best-friend").relates(pet).relates(owner).addRelationship().addRolePlayer(pet, dog).addRolePlayer(owner, person);
    Map<Long, Set<String>> correctDegrees = new HashMap<>();
    correctDegrees.put(1L, Sets.newHashSet(person.getId().getValue(), dog.getId().getValue()));
    tx.commit();
    try (GraknTx graph = session.open(GraknTxType.READ)) {
        // set subgraph, use animal instead of dog
        Set<Label> ct = Sets.newHashSet(Label.of("person"), Label.of("animal"), Label.of("mans-best-friend"));
        Map<Long, Set<String>> degrees = graph.graql().compute().centrality().usingDegree().in(ct).execute();
        // check that dog has a degree to confirm sub has been inferred
        assertEquals(correctDegrees, degrees);
    }
}

Example 58 with GraknTx

use of ai.grakn.GraknTx in project grakn by graknlabs.

the class KCoreTest method testImplicitTypeShouldBeExcluded.

@Test
public void testImplicitTypeShouldBeExcluded() {
    addSchemaAndEntities();
    try (GraknTx graph = session.open(GraknTxType.WRITE)) {
        String aResourceTypeLabel = "aResourceTypeLabel";
        AttributeType<String> attributeType = graph.putAttributeType(aResourceTypeLabel, AttributeType.DataType.STRING);
        graph.getEntityType(thing).attribute(attributeType);
        Attribute aAttribute = attributeType.putAttribute("blah");
        graph.getConcept(entityId1).asEntity().attribute(aAttribute);
        graph.getConcept(entityId2).asEntity().attribute(aAttribute);
        graph.commit();
    }
    Map<String, Set<String>> result;
    try (GraknTx graph = session.open(GraknTxType.READ)) {
        result = graph.graql().compute().cluster().usingKCore().includeAttribute().kValue(2L).execute();
        assertEquals(1, result.size());
        assertEquals(5, result.values().iterator().next().size());
        result = graph.graql().compute().cluster().usingKCore().kValue(3L).execute();
        assertEquals(1, result.size());
        assertEquals(4, result.values().iterator().next().size());
    }
}

Example 59 with GraknTx

use of ai.grakn.GraknTx in project grakn by graknlabs.

the class KCoreTest method testOnGraphWithoutRelationships_ReturnsEmptyMap.

@Test
public void testOnGraphWithoutRelationships_ReturnsEmptyMap() {
    try (GraknTx graph = session.open(GraknTxType.WRITE)) {
        graph.putEntityType(thing).addEntity();
        graph.putEntityType(anotherThing).addEntity();
        Map<String, Set<String>> result = graph.graql().compute().cluster().usingKCore().kValue(2L).execute();
        assertTrue(result.isEmpty());
    }
}

Example 60 with GraknTx

use of ai.grakn.GraknTx in project grakn by graknlabs.

the class KCoreTest method testDisconnectedCores.

@Test
public void testDisconnectedCores() {
    try (GraknTx graph = session.open(GraknTxType.WRITE)) {
        EntityType entityType1 = graph.putEntityType(thing);
        EntityType entityType2 = graph.putEntityType(anotherThing);
        Role role1 = graph.putRole("role1");
        Role role2 = graph.putRole("role2");
        RelationshipType relationshipType1 = graph.putRelationshipType(related).relates(role1).relates(role2);
        Role role3 = graph.putRole("role3");
        Role role4 = graph.putRole("role4");
        RelationshipType relationshipType2 = graph.putRelationshipType(veryRelated).relates(role3).relates(role4);
        entityType1.plays(role1).plays(role2).plays(role3).plays(role4);
        entityType2.plays(role1).plays(role2).plays(role3).plays(role4);
        Entity entity0 = entityType1.addEntity();
        Entity entity1 = entityType1.addEntity();
        Entity entity2 = entityType1.addEntity();
        Entity entity3 = entityType1.addEntity();
        Entity entity4 = entityType1.addEntity();
        Entity entity5 = entityType1.addEntity();
        Entity entity6 = entityType1.addEntity();
        Entity entity7 = entityType1.addEntity();
        Entity entity8 = entityType1.addEntity();
        relationshipType1.addRelationship().addRolePlayer(role1, entity1).addRolePlayer(role2, entity2);
        relationshipType1.addRelationship().addRolePlayer(role1, entity2).addRolePlayer(role2, entity3);
        relationshipType1.addRelationship().addRolePlayer(role1, entity3).addRolePlayer(role2, entity4);
        relationshipType1.addRelationship().addRolePlayer(role1, entity1).addRolePlayer(role2, entity3);
        relationshipType1.addRelationship().addRolePlayer(role1, entity1).addRolePlayer(role2, entity4);
        relationshipType1.addRelationship().addRolePlayer(role1, entity2).addRolePlayer(role2, entity4);
        relationshipType1.addRelationship().addRolePlayer(role1, entity5).addRolePlayer(role2, entity6);
        relationshipType2.addRelationship().addRolePlayer(role3, entity5).addRolePlayer(role4, entity7);
        relationshipType2.addRelationship().addRolePlayer(role3, entity5).addRolePlayer(role4, entity8);
        relationshipType2.addRelationship().addRolePlayer(role3, entity6).addRolePlayer(role4, entity7);
        relationshipType2.addRelationship().addRolePlayer(role3, entity6).addRolePlayer(role4, entity8);
        relationshipType2.addRelationship().addRolePlayer(role3, entity7).addRolePlayer(role4, entity8);
        relationshipType1.addRelationship().addRolePlayer(role1, entity0).addRolePlayer(role2, entity1);
        relationshipType1.addRelationship().addRolePlayer(role1, entity0).addRolePlayer(role2, entity8);
        graph.commit();
    }
    Map<String, Set<String>> result;
    try (GraknTx graph = session.open(GraknTxType.READ)) {
        result = graph.graql().compute().cluster().usingKCore().kValue(3L).execute();
        assertEquals(2, result.size());
        assertEquals(4, result.values().iterator().next().size());
        System.out.println("result = " + result);
        result = graph.graql().compute().cluster().usingKCore().kValue(2L).execute();
        assertEquals(1, result.size());
        assertEquals(9, result.values().iterator().next().size());
    }
}