def _validate_node_neighbor(self, node_neighbor, zone_vertex_id): expected_node_neighbor = tbase.create_node_placeholder_vertex() self.assertEqual(expected_node_neighbor, node_neighbor.vertex) # Validate neighbor edge edge = node_neighbor.edge self.assertEqual(edge.source_id, node_neighbor.vertex.vertex_id) self.assertEqual(edge.target_id, zone_vertex_id) self.assertEqual(edge.label, EdgeLabels.CONTAINS)
def _create_node_neighbor(self, zone_vertex_id): node_vertex = base.create_node_placeholder_vertex() relation_edge = graph_utils.create_edge( source_id=node_vertex.vertex_id, target_id=zone_vertex_id, relationship_type=EdgeLabels.CONTAINS ) return base.Neighbor(node_vertex, relation_edge)