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)
