def test_get_subedges_from_edge(self): """ test getting the subedges from a graph mapper from a edge :return: """ subvertices = list() subedges = list() subvertices.append(PartitionedVertex(None, "")) subvertices.append(PartitionedVertex(None, "")) subedges.append(MultiCastPartitionedEdge(subvertices[0], subvertices[1])) subedges.append(MultiCastPartitionedEdge(subvertices[1], subvertices[1])) sube = MultiCastPartitionedEdge(subvertices[1], subvertices[0]) subedges.append(sube) graph = GraphMapper() edge = TestPartitionableEdge(TestVertex(10, "pre"), TestVertex(5, "post")) graph.add_partitioned_edge(sube, edge) graph.add_partitioned_edge(subedges[0], edge) subedges_from_edge = \ graph.get_partitioned_edges_from_partitionable_edge(edge) self.assertIn(sube, subedges_from_edge) self.assertIn(subedges[0], subedges_from_edge) self.assertNotIn(subedges[1], subedges_from_edge)