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); }
private static void debugEdge(final Edge e) { System.out.println("edge " + e + ":"); for (String key : e.getPropertyKeys()) { System.out.println("\t" + key + ":\t'" + e.getProperty(key) + "'"); } System.out.println("\t[in vertex]: " + e.getVertex(Direction.IN)); System.out.println("\t[out vertex]: " + e.getVertex(Direction.OUT)); }
protected Map<Integer, ValueLoader> addEdgesLoaders() { Map<Integer, ValueLoader> returned = new TreeMap<Integer, ValueLoader>(); for (Edge e : strategy.getOutEdgesFor(rootVertex, property)) { if (e.getProperty(Properties.collection_index.name()) != null) { returned.put( (Integer) e.getProperty(Properties.collection_index.name()), new LoadValueBehindEdge(e)); } else { // These items are pushed to the latest element (after collectionSize) which has no value // set int index = Integer.MAX_VALUE - 1; while (returned.containsKey(index)) { index--; } returned.put(index, new LoadValueBehindEdge(e)); } } return returned; }
private void verifyLineageGraphForReplicationOrEviction( String feedName, String feedInstanceDataPath, WorkflowExecutionContext context, RelationshipLabel edgeLabel) throws Exception { String feedInstanceName = InstanceRelationshipGraphBuilder.getFeedInstanceName( feedName, context.getClusterName(), feedInstanceDataPath, context.getNominalTimeAsISO8601()); Vertex feedVertex = getEntityVertex(feedInstanceName, RelationshipType.FEED_INSTANCE); Edge edge = feedVertex.getEdges(Direction.OUT, edgeLabel.getName()).iterator().next(); Assert.assertNotNull(edge); Assert.assertEquals( edge.getProperty(RelationshipProperty.TIMESTAMP.getName()), context.getTimeStampAsISO8601()); Vertex clusterVertex = edge.getVertex(Direction.IN); Assert.assertEquals( clusterVertex.getProperty(RelationshipProperty.NAME.getName()), context.getClusterName()); }