Separate the data from the logic (chap07 graph)

Author: codingEzio

chap07_graphs/data.py

@@ -0,0 +1,11 @@
+edges_with_weights = (
+    ("a", "b", 7),
+    ("a", "c", 9),
+    ("a", "f", 14),
+    ("b", "c", 10),
+    ("b", "d", 15),
+    ("c", "d", 11),
+    ("c", "f", 2),
+    ("d", "e", 6),
+    ("e", "f", 9),
+)

chap07_graphs/dijkstra.py

@@ -1,4 +1,5 @@
 # Ref: https://dev.to/mxl/dijkstras-algorithm-in-python-algorithms-for-beginners-dkc
+from data import edges_with_weights as edges
 
 from collections import deque, namedtuple
 from math import inf
@@ -25,8 +26,7 @@ def __init__(self, edges):
 
     @property
     def vertices(self):
-        """Return a set of strings that stands for vertices/nodes.
-        """
+        """Return a set of strings that stands for vertices/nodes."""
         return set(sum(([edge.start, edge.end] for edge in self.edges), []))
 
     @property
@@ -137,19 +137,7 @@ def remove_edge(self, node1, node2, both_ends=True):
 
 
 def main() -> None:
-    graph = Graph(
-        [
-            ("a", "b", 7),
-            ("a", "c", 9),
-            ("a", "f", 14),
-            ("b", "c", 10),
-            ("b", "d", 15),
-            ("c", "d", 11),
-            ("c", "f", 2),
-            ("d", "e", 6),
-            ("e", "f", 9),
-        ]
-    )
+    graph = Graph(edges)
 
     assert graph.dijkstra("a", "e") == deque(["a", "c", "d", "e"])
 

chap07_graphs/plot_weighted_graph.py

@@ -1,25 +1,14 @@
 # References:
 # https://networkx.github.io/documentation/stable/auto_examples/drawing/plot_weighted_graph.html
 # https://stackoverflow.com/questions/56597840/drawing-weighted-graph-from-adjacency-matrix-with-edge-labels
+from data import edges_with_weights as edges
 
 import matplotlib.pyplot as plt
 import networkx as nx
 
 # Graph = nx.Graph()
 Graph = nx.DiGraph(directed=True)
 
-edges = (
-    ("a", "b", 7),
-    ("a", "c", 9),
-    ("a", "f", 14),
-    ("b", "c", 10),
-    ("b", "d", 15),
-    ("c", "d", 11),
-    ("c", "f", 2),
-    ("d", "e", 6),
-    ("e", "f", 9),
-)
-
 for orig, dest, weight in edges:
     Graph.add_edge(orig, dest, weight=weight)