Update tests/test.py

Correct references to nodes that were previously hard-coded.

Author: JackAtOmenApps

tests/test.py

@@ -252,7 +252,9 @@ def test_02_dag(self):
         log.debug("Check attributes")
         self.assertEqual(nx_out.graph, {"test": "test"})
         self.assertEqual(nx_out.nodes[root.pk], {"id": root.pk, "name": "root"})
-        self.assertEqual(nx_out.edges[root.pk, a3.pk], {"id": NetworkEdge.objects.get(parent=root, child=a3).pk, "name": "root a3"})
+        self.assertEqual(
+            nx_out.edges[root.pk, a3.pk], {"id": NetworkEdge.objects.get(parent=root, child=a3).pk, "name": "root a3"}
+        )
 
         """
         Simulate a basic irrigation canal network