[FIX] test_quick file black

aiplan4eu · Oct 25, 2023 · 4314463 · 4314463
1 parent 4919fe2
commit 4314463
Showing 1 changed file with 25 additions and 8 deletions.
diff --git a/test_quick.py b/test_quick.py
@@ -21,15 +21,32 @@
     graph = ttp_to_stn.run()
 
     options = {
-    'node_color': 'blue',
-    'node_size': 1000,
-    'width': 3,
-}
-    pos=nx.spring_layout(ttp_to_stn.stn)   
-    edge_labels=dict([((u,v,),[float(d['interval'][0]), float(d['interval'][0])])
-                 for u,v,d in ttp_to_stn.stn.edges(data=True)])
+        "node_color": "blue",
+        "node_size": 1000,
+        "width": 3,
+    }
+    pos = nx.spring_layout(ttp_to_stn.stn)
+    edge_labels = dict(
+        [
+            (
+                (
+                    u,
+                    v,
+                ),
+                [float(d["interval"][0]), float(d["interval"][0])],
+            )
+            for u, v, d in ttp_to_stn.stn.edges(data=True)
+        ]
+    )
     nx.draw_networkx_edge_labels(ttp_to_stn.stn, pos, edge_labels=edge_labels)
-    nx.draw_networkx(ttp_to_stn.stn, pos, with_labels = True, arrows=True, **options, edge_cmap=plt.cm.Reds)
+    nx.draw_networkx(
+        ttp_to_stn.stn,
+        pos,
+        with_labels=True,
+        arrows=True,
+        **options,
+        edge_cmap=plt.cm.Reds
+    )
     pylab.show()
 
     print("Done...")