diff --git a/graph.py b/graph.py index 42a6402..5c2d187 100644 --- a/graph.py +++ b/graph.py @@ -27,7 +27,7 @@ def __eq__(self, other): ################################################################################ class DirectedAdjList: - def __init__(self, edge_list=[], vertex_label=None, + def __init__(self, edge_list=None, vertex_label=None, vertex_text=None, edge_label=None, edge_color=None): self.out = {} @@ -42,6 +42,8 @@ def __init__(self, edge_list=[], vertex_label=None, self.edge_label = edge_label self.edge_color = edge_color self.edge_set = set() + if not edge_list: + return for e in edge_list: if isinstance(e, Edge): self.add_edge(e.source, e.target)