com.tinkerpop.gremlin.process.graph.step.map.FoldTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gremlin-test Show documentation
The newest version!
package com.tinkerpop.gremlin.process.graph.step.map;

import com.tinkerpop.gremlin.LoadGraphWith;
import com.tinkerpop.gremlin.process.AbstractGremlinProcessTest;
import com.tinkerpop.gremlin.process.Traversal;
import com.tinkerpop.gremlin.structure.Vertex;
import org.junit.Test;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static com.tinkerpop.gremlin.LoadGraphWith.GraphData.MODERN;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public abstract class FoldTest extends AbstractGremlinProcessTest {

    public abstract Traversal> get_g_V_fold();

    public abstract Traversal get_g_V_fold_unfold();

    public abstract Traversal get_g_V_age_foldX0_plusX();

    @Test
    @LoadGraphWith(MODERN)
    public void g_V_fold() {
        final Traversal> traversal = get_g_V_fold();
        printTraversalForm(traversal);
        List list = traversal.next();
        assertFalse(traversal.hasNext());
        Set vertices = new HashSet<>(list);
        assertEquals(6, vertices.size());
    }

    @Test
    @LoadGraphWith(MODERN)
    public void g_V_fold_unfold() {
        final Traversal traversal = get_g_V_fold_unfold();
        printTraversalForm(traversal);
        int count = 0;
        Set vertices = new HashSet<>();
        while (traversal.hasNext()) {
            vertices.add(traversal.next());
            count++;
        }
        assertFalse(traversal.hasNext());
        assertEquals(6, count);
        assertEquals(6, vertices.size());
    }

    @Test
    @LoadGraphWith(MODERN)
    public void g_V_age_foldX0_plusX() {
        final Traversal traversal = get_g_V_age_foldX0_plusX();
        printTraversalForm(traversal);
        final Integer ageSum = traversal.next();
        assertFalse(traversal.hasNext());
        assertEquals(Integer.valueOf(123), ageSum);
    }

    public static class StandardTest extends FoldTest {

        @Override
        public Traversal> get_g_V_fold() {
            return g.V().fold();
        }

        @Override
        public Traversal get_g_V_fold_unfold() {
            return g.V().fold().unfold();
        }

        @Override
        public Traversal get_g_V_age_foldX0_plusX() {
            return g.V().values("age").fold(0, (seed, age) -> seed + age);
        }
    }

    public static class ComputerTest extends FoldTest {

        public ComputerTest() {
            requiresGraphComputer = true;
        }

        @Override
        public Traversal> get_g_V_fold() {
            return g.V().fold().submit(g.compute());
        }

        @Override
        public Traversal get_g_V_fold_unfold() {
            return (Traversal) g.V().fold().unfold().submit(g.compute());
        }

        @Override
        public Traversal get_g_V_age_foldX0_plusX() {
            return g.V().values("age").fold(0, (seed, age) -> seed + age).submit(g.compute());
        }
    }
}