org.umlg.sqlg.test.reducing.TestReducingVertexStep Maven / Gradle / Ivy

Go to download
package org.umlg.sqlg.test.reducing;

import org.apache.tinkerpop.gremlin.process.traversal.Traversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * @author Pieter Martin (https://github.com/pietermartin)
 * Date: 2018/12/31
 */
@SuppressWarnings("rawtypes")
public class TestReducingVertexStep extends BaseTest {

    @Test
    public void testGroupOverOnePropertyMax() {
        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "name", "a1", "age", 1);
        Vertex b1 = this.sqlgGraph.addVertex(T.label, "B", "name", "b1", "age", 1);
        Vertex b2 = this.sqlgGraph.addVertex(T.label, "B", "name", "b1", "age", 2);
        Vertex b3 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 3);
        Vertex b4 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 4);

        Vertex a2 = this.sqlgGraph.addVertex(T.label, "A", "name", "a2", "age", 1);
        Vertex b21 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 5);
        Vertex b22 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 6);
        Vertex b23 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 7);
        Vertex b24 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 8);

        a1.addEdge("ab", b1);
        a1.addEdge("ab", b2);
        a1.addEdge("ab", b3);
        a1.addEdge("ab", b4);

        a2.addEdge("ab", b21);
        a2.addEdge("ab", b22);
        a2.addEdge("ab", b23);
        a2.addEdge("ab", b24);
        this.sqlgGraph.tx().commit();

        Traversal> traversal = this.sqlgGraph.traversal()
                .V().has("name", __.or(__.is("a1"), __.is("a2"))).out()
                .group().by("name").by(__.values("age").max());

        printTraversalForm(traversal);
        Assert.assertTrue(traversal.hasNext());
        Map result = traversal.next();
        Assert.assertFalse(traversal.hasNext());
        Assert.assertTrue(result.containsKey("b1"));
        Assert.assertTrue(result.containsKey("b2"));
        Assert.assertEquals(2, result.get("b1"), 0);
        Assert.assertEquals(8, result.get("b2"), 0);
    }

    @Test
    public void testGroupOverOnePropertyLocalMax() {
        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "name", "a1", "age", 1);
        Vertex b1 = this.sqlgGraph.addVertex(T.label, "B", "name", "b1", "age", 1);
        Vertex b2 = this.sqlgGraph.addVertex(T.label, "B", "name", "b1", "age", 2);
        Vertex b3 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 3);
        Vertex b4 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2", "age", 4);

        Vertex a2 = this.sqlgGraph.addVertex(T.label, "A", "name", "a2", "age", 1);
        Vertex b21 = this.sqlgGraph.addVertex(T.label, "B", "name", "b21", "age", 1);
        Vertex b22 = this.sqlgGraph.addVertex(T.label, "B", "name", "b21", "age", 2);
        Vertex b23 = this.sqlgGraph.addVertex(T.label, "B", "name", "b22", "age", 3);
        Vertex b24 = this.sqlgGraph.addVertex(T.label, "B", "name", "b22", "age", 4);

        a1.addEdge("ab", b1);
        a1.addEdge("ab", b2);
        a1.addEdge("ab", b3);
        a1.addEdge("ab", b4);

        a2.addEdge("ab", b21);
        a2.addEdge("ab", b22);
        a2.addEdge("ab", b23);
        a2.addEdge("ab", b24);
        this.sqlgGraph.tx().commit();

        Traversal> traversal = this.sqlgGraph.traversal()
                .V().hasLabel("A")
                .local(
                        __.out().group().by("name").by(__.values("age").max())
                );

        printTraversalForm(traversal);
        List> result = traversal.toList();
        Assert.assertEquals(2, result.size());
        for (int i = 0; i < 2; i++) {
            if (result.get(i).containsKey("b1") && result.get(i).containsKey("b2")) {
                Assert.assertEquals(2, result.get(i).get("b1"), 0);
                Assert.assertEquals(4, result.get(i).get("b2"), 0);
            } else if (result.get(i).containsKey("b21") && result.get(i).containsKey("b22")) {
                Assert.assertEquals(2, result.get(i).get("b21"), 0);
                Assert.assertEquals(4, result.get(i).get("b22"), 0);
            } else {
                Assert.fail();
            }
        }

    }

    @Test
    public void g_VX1_2X_unionXoutE_count__inE_count__outE_weight_sumX() {
        loadModern();
        @SuppressWarnings("unchecked") final Traversal traversal = this.sqlgGraph.traversal()
                .V(convertToVertexId("marko"), convertToVertexId("vadas"))
                .union(
                        __.outE().count(),
                        __.inE().count(),
                        (Traversal) __.outE().values("weight").sum()
                );
        printTraversalForm(traversal);
        checkResults(Arrays.asList(3L, 1.9D, 1L), traversal);
    }

    @Test
    public void g_VX1_2X_unionX_outE_weight_sumX() {
        loadModern();
        @SuppressWarnings("unchecked") final Traversal traversal = this.sqlgGraph.traversal()
                .V(convertToVertexId("marko"), convertToVertexId("vadas"))
                .union(
                        (Traversal) __.outE().values("weight").sum()
                );
        printTraversalForm(traversal);
        checkResults(Collections.singletonList(1.9D), traversal);
    }

    @Test
    public void g_V_hasLabelXsongX_group_byXnameX_byXproperties_groupCount_byXlabelXX() {
        loadGratefulDead();
        final Traversal>> traversal = this.sqlgGraph.traversal()
                .V().hasLabel("song")
                .>group().by("name").by(__.properties().groupCount().by(T.label));
        printTraversalForm(traversal);
        final Map> map = traversal.next();
        Assert.assertEquals(584, map.size());
        for (final Map.Entry> entry : map.entrySet()) {
            Assert.assertEquals(entry.getKey().toUpperCase(), entry.getKey());
            final Map countMap = entry.getValue();
            Assert.assertEquals(3, countMap.size());
            Assert.assertEquals(1L, countMap.get("name").longValue());
            Assert.assertEquals(1L, countMap.get("songType").longValue());
            Assert.assertEquals(1L, countMap.get("performances").longValue());
        }
        Assert.assertFalse(traversal.hasNext());
    }
}