simplify get_edge_weights

Xuefeng-Zhu · Xuefeng-Zhu · commit 41f4f39c4d08 · 2015-05-11T00:27:08.000-05:00
diff --git a/graphs/graph.py b/graphs/graph.py
@@ -140,10 +140,7 @@ def get_edge_weights(self):
         unique_list = {}
         for u in self.nodes():
             for v in self.neighbors(u):
-                if not  unique_list.get(v) or u not in unique_list.get(v):
+                if u not in unique_list.get(v, []):
                     edge_list.append((self.node_neighbors[u][v], (u, v)))
-                    if u not in unique_list:
-                        unique_list[u] = [v]
-                    else:
-                        unique_list[u].append(v)
+                    unique_list.setdefault(u, [v]).append(v)
         return edge_list
