Split dijkstra.py into three files: node, edge, graph

src/graph/edge.py

@@ -0,0 +1,5 @@
+class Edge:
+    def __init__(self, start: int, end: int, length: float):
+        self.length: float = length
+        self.start: int = start
+        self.end: int = end

src/graph/graph.py

@@ -1,19 +1,8 @@
 from math import inf
 from typing import Iterator, Optional
 
-
+from src.graph.node import Node
-class Node:
+from src.graph.edge import Edge
-    def __init__(self, x: int, y: int):
-        self.x: int = x
-        self.y: int = y
-
-
-class Edge:
-    def __init__(self, start: int, end: int, length: float):
-        self.length: float = length
-        self.start: int = start
-        self.end: int = end
-
 
 class Graph:
     def __init__(self):
@@ -63,26 +52,4 @@ class Graph:
                     node_sequences[end] = node_sequences[start].copy()
                     node_sequences[end].append(end)
 
-        return node_sequences[target_index]
+        return node_sequences[target_index]
-
-
-def main() -> None:
-    graph = Graph()
-
-    graph.add_node(1, 2)
-    graph.add_node(4, 7)
-    graph.add_node(3, 1)
-    graph.add_node(-2, 0)
-    graph.add_node(0, 0)
-
-    graph.add_edge(0, 1, 1)
-    graph.add_edge(1, 2, 2)
-    graph.add_edge(2, 3, 3)
-    graph.add_edge(3, 0, 1)
-    graph.add_edge(1, 3, 3)
-
-    print(graph.dijkstra(0, 3))
-
-
-if __name__ == "__main__":
-    main()

src/graph/node.py

@@ -0,0 +1,4 @@
+class Node:
+    def __init__(self, x: int, y: int):
+        self.x: int = x
+        self.y: int = y