diff --git a/groupConnectivityNaive.py b/groupConnectivityNaive.py index af4c4bc734ef0d0aa455b6f84d6223364473f532..5bbd2fed41a301845b214a0670be0a65f0a94c5c 100644 --- a/groupConnectivityNaive.py +++ b/groupConnectivityNaive.py @@ -1,5 +1,12 @@ from sage.graphs.graph import Graph, DiGraph +def prettifyGraph(G_): + """Convert graph into directed one and number its edges.""" + + E = [ (u, v, i) for i, (u, v, _) in enumerate(G_.edges()) ] + return DiGraph([G_.vertices(), E], format='vertices_and_edges') + + def flowEnumerator(G, group): """Enumerate all flows of given graph.