fix add_edge to pass properties to edge instead of the SLelement #69

BridgesUNCC · Apr 10, 2020 · 90438cf · 90438cf
1 parent 35ff7b3
commit 90438cf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/bridges/edge.py b/bridges/edge.py
@@ -14,7 +14,7 @@
 
 class Edge():
 
-    def __init__(self, v1, v2, data=None) -> None:
+    def __init__(self, v1, v2, data=None, label: str = None, color: Color = None, thickness: float = None) -> None:
         """
         Constructor for a edge
         Args:
@@ -29,6 +29,15 @@ def __init__(self, v1, v2, data=None) -> None:
         self._edge_data = data
         self._lvis = LinkVisualizer()
 
+        if label:
+            self.label = label
+
+        if color:
+            self.color = color
+
+        if thickness:
+            self.thickness = thickness
+
     @property
     def tov(self):
         """

diff --git a/bridges/graph_adj_list.py b/bridges/graph_adj_list.py
@@ -115,8 +115,8 @@ def add_edge(self,
                                  " does not exist! Add the vertex before creating the edge.")
         except Exception as e:
             traceback.print_tb(e.__traceback__)
-        self.adj_list[source_id] = SLelement(e=Edge(source_id, dest_id, data), next=self.adj_list.get(src), label=label,
-                                       color=color, thickness=thickness, opacity=opacity)
+        self.adj_list[source_id] = SLelement(e=Edge(source_id, dest_id, data, label=label, color=color,
+                                                    thickness=thickness), next=self.adj_list.get(src))
 
         return self.adj_list[source_id].value