diff --git a/src/algoritmia/datastructures/digraphs/weightingfunction.py b/src/algoritmia/datastructures/digraphs/weightingfunction.py index 2086ea4..e6e743f 100644 --- a/src/algoritmia/datastructures/digraphs/weightingfunction.py +++ b/src/algoritmia/datastructures/digraphs/weightingfunction.py @@ -11,11 +11,16 @@ def __init__(self, data: "Iterable<((T, T), K)> or IMap<(T, T), K>" =[], #?data? self._map = self.createMap(data) self.symmetrical = symmetrical if symmetrical: + duplicates = set() for (u, v) in self._map.keys(): if (v, u) in self._map: if self._map[u, v] != self._map[v, u]: raise ValueError("{!r} is different from {!r}".format((u,v), (v,u))) - if v != u: del self._map[v, u] + if v != u and (u, v) not in duplicates: + duplicates.add((v, u)) + + for dup in duplicates: + self._map.pop(dup) def __contains__(self, key: "(T, T)") -> "bool": return key in self._map