diff --git a/create_graph.py b/create_graph.py index 584c307596c09698b6987bb5f8aed246a938b87d..42b35cbae1b934e134d24055feb6d05107b97846 100644 --- a/create_graph.py +++ b/create_graph.py @@ -8,6 +8,8 @@ def create_random_graph(args): nb_nodes = args.nodes nb_links = args.links + random.seed(args.seed) + nodes = dict() for _ in range(nb_links): node_a = random.randint(0, nb_nodes - 1) @@ -88,8 +90,9 @@ if __name__ == "__main__": help="Number of nodes. Unused if '--ntf'", default=5) parser.add_argument("-l", "--links", type=int, help="Number of links. Unused if '--ntf'", default=10) - parser.add_argument("--c-header", action="store_true", + parser.add_argument("-c", "--c-header", action="store_true", help="Writes the graph as a C header file (.h) instead of a binary file") + parser.add_argument("-s", "--seed", type=int, help="Seed for random generation of the graph", default=42) args = parser.parse_args() if args.ntf: