diff options
| -rw-r--r-- | analysis/scalarmults/epare/prob_map.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/analysis/scalarmults/epare/prob_map.py b/analysis/scalarmults/epare/prob_map.py index f0df3ed..3846661 100644 --- a/analysis/scalarmults/epare/prob_map.py +++ b/analysis/scalarmults/epare/prob_map.py @@ -33,6 +33,14 @@ class ProbMap: def __contains__(self, item): return item in self.probs + def __eq__(self, other): + if not isinstance(other, ProbMap): + return False + return self.divisors_hash == other.divisors_hash and self.probs == other.probs + + def id(self): + return hash((frozenset(self.probs.items()), self.divisors_hash)) + def keys(self): return self.probs.keys() |