From 1afdae7c8b28bfb323836a855f4d757aa17d2ef9 Mon Sep 17 00:00:00 2001 From: Claudio Scheer Date: Sun, 17 May 2020 14:35:06 -0300 Subject: [PATCH] Update tests --- src/a_star.py | 4 +++- src/graph_tests.py | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/a_star.py b/src/a_star.py index 60a9856..4d3734e 100644 --- a/src/a_star.py +++ b/src/a_star.py @@ -42,6 +42,8 @@ def a_star(graph, start_node, goal_node, heuristic): graph, start_node, goal_node = get_graph_1() path = a_star(graph, start_node, goal_node, max_cost) print(f"Start node: {start_node.id}. Goal node: {goal_node.id}") +print("-" * 5, end="") +print("Solution plan", end="") +print("-" * 5) for x in path: print(x.id) -print("-" * 30) diff --git a/src/graph_tests.py b/src/graph_tests.py index 8310610..29c0e03 100644 --- a/src/graph_tests.py +++ b/src/graph_tests.py @@ -7,12 +7,16 @@ def get_graph_1(): node_2 = Node(2) node_3 = Node(3) node_4 = Node(4) + node_5 = Node(5) + node_6 = Node(6) graph.set_bidirectional_neighbor(node_1, node_2, 1) graph.set_bidirectional_neighbor(node_1, node_3, 1) graph.set_bidirectional_neighbor(node_2, node_4, 1) graph.set_bidirectional_neighbor(node_3, node_4, 1) + graph.set_bidirectional_neighbor(node_3, node_5, 1) + graph.set_bidirectional_neighbor(node_5, node_6, 1) - graph.add_nodes([node_1, node_2, node_3, node_4]) + graph.add_nodes([node_1, node_2, node_3, node_4, node_5, node_6]) - return graph, node_1, node_4 + return graph, node_1, node_6 -- GitLab