[dbs-leipzig#1559] add unit tests

gradoop-temporal/src/test/java/org/gradoop/temporal/model/impl/operators/metric/AvgDegreeEvolutionTest.java

@@ -0,0 +1,144 @@
+/*
+ * Copyright © 2014 - 2021 Leipzig University (Database Research Group)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.gradoop.temporal.model.impl.operators.metric;
+
+import org.apache.flink.api.common.typeinfo.TypeHint;
+import org.apache.flink.api.java.DataSet;
+import org.apache.flink.api.java.io.LocalCollectionOutputFormat;
+import org.apache.flink.api.java.tuple.Tuple2;
+import org.gradoop.common.model.impl.id.GradoopId;
+import org.gradoop.flink.model.impl.operators.sampling.functions.VertexDegree;
+import org.gradoop.temporal.model.api.TimeDimension;
+import org.gradoop.temporal.model.impl.TemporalGraph;
+import org.gradoop.temporal.util.TemporalGradoopTestBase;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.List;
+
+import static org.junit.Assert.assertTrue;
+
+@RunWith(Parameterized.class)
+public class AvgDegreeEvolutionTest extends TemporalGradoopTestBase {
+    /**
+     * The expected in-degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_IN_DEGREES = new ArrayList<>();
+    /**
+     * The expected out-degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_OUT_DEGREES = new ArrayList<>();
+    /**
+     * The expected degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_BOTH_DEGREES = new ArrayList<>();
+
+    static {
+        // IN DEGREES
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(4L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(6L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(7L, 1));
+
+        // OUT DEGREES
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(4L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(6L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(7L, 1));
+
+        // DEGREES
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(4L, 2));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(6L, 1));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(7L, 1));
+    }
+
+    /**
+     * The degree type to test.
+     */
+    @Parameterized.Parameter(0)
+    public VertexDegree degreeType;
+
+    /**
+     * The expected degree evolution fo the given type.
+     */
+    @Parameterized.Parameter(1)
+    public List<Tuple2<Long, Integer>> expectedDegrees;
+
+    /**
+     * The temporal graph to test the operator.
+     */
+    TemporalGraph testGraph;
+
+    /**
+     * The parameters to test the operator.
+     *
+     * @return three different vertex degree types with its corresponding expected degree evolution.
+     */
+    @Parameterized.Parameters(name = "Test degree type {0}.")
+    public static Iterable<Object[]> parameters() {
+        return Arrays.asList(
+                new Object[] {VertexDegree.IN, EXPECTED_IN_DEGREES},
+                new Object[] {VertexDegree.OUT, EXPECTED_OUT_DEGREES},
+                new Object[] {VertexDegree.BOTH, EXPECTED_BOTH_DEGREES});
+    }
+
+    /**
+     * Set up the test graph an create the id-label mapping.
+     *
+     * @throws Exception in case of an error
+     */
+    @Before
+    public void setUp() throws Exception {
+        testGraph = getTestGraphWithValues();
+        Collection<Tuple2<GradoopId, String>> idLabelCollection = new HashSet<>();
+        testGraph.getVertices().map(v -> new Tuple2<>(v.getId(), v.getLabel()))
+                .returns(new TypeHint<Tuple2<GradoopId, String>>() {
+                }).output(new LocalCollectionOutputFormat<>(idLabelCollection));
+        getExecutionEnvironment().execute();
+    }
+
+    /**
+     * Test the avg degree evolution operator.
+     *
+     * @throws Exception in case of an error.
+     */
+    @Test
+    public void testAvgDegree() throws Exception {
+        Collection<Tuple2<Long, Integer>> resultCollection = new ArrayList<>();
+
+        final DataSet<Tuple2<Long, Integer>> resultDataSet = testGraph
+                .callForValue(new AvgDegreeEvolution(degreeType, TimeDimension.VALID_TIME));
+
+        resultDataSet.output(new LocalCollectionOutputFormat<>(resultCollection));
+        getExecutionEnvironment().execute();
+
+        assertTrue(resultCollection.containsAll(expectedDegrees));
+        assertTrue(expectedDegrees.containsAll(resultCollection));
+    }
+}

gradoop-temporal/src/test/java/org/gradoop/temporal/model/impl/operators/metric/MaxDegreeEvolutionTest.java

@@ -0,0 +1,144 @@
+/*
+ * Copyright © 2014 - 2021 Leipzig University (Database Research Group)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.gradoop.temporal.model.impl.operators.metric;
+
+import org.apache.flink.api.common.typeinfo.TypeHint;
+import org.apache.flink.api.java.DataSet;
+import org.apache.flink.api.java.io.LocalCollectionOutputFormat;
+import org.apache.flink.api.java.tuple.Tuple2;
+import org.gradoop.common.model.impl.id.GradoopId;
+import org.gradoop.flink.model.impl.operators.sampling.functions.VertexDegree;
+import org.gradoop.temporal.model.api.TimeDimension;
+import org.gradoop.temporal.model.impl.TemporalGraph;
+import org.gradoop.temporal.util.TemporalGradoopTestBase;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.List;
+
+import static org.junit.Assert.assertTrue;
+
+@RunWith(Parameterized.class)
+public class MaxDegreeEvolutionTest extends TemporalGradoopTestBase {
+    /**
+     * The expected in-degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_IN_DEGREES = new ArrayList<>();
+    /**
+     * The expected out-degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_OUT_DEGREES = new ArrayList<>();
+    /**
+     * The expected degrees for each vertex label.
+     */
+    private static final List<Tuple2<Long, Integer>> EXPECTED_BOTH_DEGREES = new ArrayList<>();
+
+    static {
+        // IN DEGREES
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(4L, 2));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(6L, 1));
+        EXPECTED_IN_DEGREES.add(new Tuple2<>(7L, 1));
+
+        // OUT DEGREES
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(4L, 2));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(6L, 1));
+        EXPECTED_OUT_DEGREES.add(new Tuple2<>(7L, 1));
+
+        // DEGREES
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(Long.MIN_VALUE, 0));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(0L, 1));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(4L, 3));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(5L, 1));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(6L, 2));
+        EXPECTED_BOTH_DEGREES.add(new Tuple2<>(7L, 1));
+    }
+
+    /**
+     * The degree type to test.
+     */
+    @Parameterized.Parameter(0)
+    public VertexDegree degreeType;
+
+    /**
+     * The expected degree evolution fo the given type.
+     */
+    @Parameterized.Parameter(1)
+    public List<Tuple2<Long, Integer>> expectedDegrees;
+
+    /**
+     * The temporal graph to test the operator.
+     */
+    TemporalGraph testGraph;
+
+    /**
+     * The parameters to test the operator.
+     *
+     * @return three different vertex degree types with its corresponding expected degree evolution.
+     */
+    @Parameterized.Parameters(name = "Test degree type {0}.")
+    public static Iterable<Object[]> parameters() {
+        return Arrays.asList(
+                new Object[] {VertexDegree.IN, EXPECTED_IN_DEGREES},
+                new Object[] {VertexDegree.OUT, EXPECTED_OUT_DEGREES},
+                new Object[] {VertexDegree.BOTH, EXPECTED_BOTH_DEGREES});
+    }
+
+    /**
+     * Set up the test graph an create the id-label mapping.
+     *
+     * @throws Exception in case of an error
+     */
+    @Before
+    public void setUp() throws Exception {
+        testGraph = getTestGraphWithValues();
+        Collection<Tuple2<GradoopId, String>> idLabelCollection = new HashSet<>();
+        testGraph.getVertices().map(v -> new Tuple2<>(v.getId(), v.getLabel()))
+                .returns(new TypeHint<Tuple2<GradoopId, String>>() {
+                }).output(new LocalCollectionOutputFormat<>(idLabelCollection));
+        getExecutionEnvironment().execute();
+    }
+
+    /**
+     * Test the max degree evolution operator.
+     *
+     * @throws Exception in case of an error.
+     */
+    @Test
+    public void testMaxDegree() throws Exception {
+        Collection<Tuple2<Long, Integer>> resultCollection = new ArrayList<>();
+
+        final DataSet<Tuple2<Long, Integer>> resultDataSet = testGraph
+                .callForValue(new MaxDegreeEvolution(degreeType, TimeDimension.VALID_TIME));
+
+        resultDataSet.output(new LocalCollectionOutputFormat<>(resultCollection));
+        getExecutionEnvironment().execute();
+
+        assertTrue(resultCollection.containsAll(expectedDegrees));
+        assertTrue(expectedDegrees.containsAll(resultCollection));
+    }
+}