1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph.edge;
20
21 import com.google.common.collect.Lists;
22 import org.apache.hadoop.io.DoubleWritable;
23 import org.apache.hadoop.io.LongWritable;
24 import org.junit.Before;
25 import org.junit.Test;
26
27 import java.util.Collection;
28 import java.util.List;
29
30 import static org.apache.giraph.graph.TestVertexAndEdges.instantiateOutEdges;
31 import static org.junit.Assert.assertEquals;
32
33
34
35
36 public class TestStrictGraphEdges {
37
38 private Collection<Class<? extends OutEdges>> edgesClasses =
39 Lists.newArrayList();
40
41 @Before
42 public void setUp() {
43 edgesClasses.add(HashMapEdges.class);
44 edgesClasses.add(LongDoubleHashMapEdges.class);
45 }
46
47
48
49
50
51 @Test
52 public void testParallelEdges() {
53 for (Class<? extends OutEdges> edgesClass : edgesClasses) {
54 testParallelEdgesClass(edgesClass);
55 }
56 }
57
58 private void testParallelEdgesClass(
59 Class<? extends OutEdges> edgesClass) {
60 OutEdges<LongWritable, DoubleWritable> edges =
61 instantiateOutEdges(edgesClass);
62
63
64 List<Edge<LongWritable, DoubleWritable>> initialEdges = Lists.newArrayList(
65 EdgeFactory.create(new LongWritable(1), new DoubleWritable(1)),
66 EdgeFactory.create(new LongWritable(2), new DoubleWritable(2)),
67 EdgeFactory.create(new LongWritable(3), new DoubleWritable(3)),
68 EdgeFactory.create(new LongWritable(2), new DoubleWritable(20)));
69
70 edges.initialize(initialEdges);
71
72
73 assertEquals(3, edges.size());
74
75
76 edges.add(EdgeFactory.create(new LongWritable(3), new DoubleWritable(30)));
77 assertEquals(3, edges.size());
78 }
79 }