public void testEdgesOnWeight() throws Exception { Configuration config = IntervalFilterMap.createConfiguration(Edge.class, "weight", 0.3f, 0.45f); mapReduceDriver.withConfiguration(config); Map<Long, FaunusVertex> graph = runWithGraph( startPath(generateGraph(BaseTest.ExampleGraph.TINKERGRAPH, config), Edge.class), mapReduceDriver); assertEquals(graph.size(), 6); long counter = 0; for (FaunusVertex vertex : graph.values()) { for (Edge edge : vertex.getEdges(Direction.BOTH)) { if (((StandardFaunusEdge) edge).hasPaths()) { counter = ((StandardFaunusEdge) edge).pathCount() + counter; assertEquals(edge.getProperty("weight"), 0.4d); } } } assertEquals(counter, 4); assertEquals( DEFAULT_COMPAT.getCounter(mapReduceDriver, IntervalFilterMap.Counters.VERTICES_FILTERED), 0); assertEquals( DEFAULT_COMPAT.getCounter(mapReduceDriver, IntervalFilterMap.Counters.EDGES_FILTERED), 8); identicalStructure(graph, ExampleGraph.TINKERGRAPH); }
public void testVerticesOnAge() throws Exception { Configuration config = IntervalFilterMap.createConfiguration(Vertex.class, "age", 10, 30); mapReduceDriver.withConfiguration(config); Map<Long, FaunusVertex> graph = runWithGraph( startPath(generateGraph(BaseTest.ExampleGraph.TINKERGRAPH, config), Vertex.class), mapReduceDriver); assertEquals(graph.size(), 6); assertEquals(graph.get(1l).pathCount(), 1); assertEquals(graph.get(2l).pathCount(), 1); assertEquals(graph.get(3l).pathCount(), 0); assertEquals(graph.get(4l).pathCount(), 0); assertEquals(graph.get(5l).pathCount(), 0); assertEquals(graph.get(6l).pathCount(), 0); assertEquals( DEFAULT_COMPAT.getCounter(mapReduceDriver, IntervalFilterMap.Counters.VERTICES_FILTERED), 4); assertEquals( DEFAULT_COMPAT.getCounter(mapReduceDriver, IntervalFilterMap.Counters.EDGES_FILTERED), 0); identicalStructure(graph, ExampleGraph.TINKERGRAPH); }