diff options
| -rw-r--r-- | rumba/model.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/rumba/model.py b/rumba/model.py index 5c22185..45d0473 100644 --- a/rumba/model.py +++ b/rumba/model.py @@ -91,6 +91,15 @@ class DIF: s = "DIF %s" % self.name return s + def __hash__(self): + return hash(self.name) + + def __eq__(self, other): + return other != None and self.name == other.name + + def __neq__(self, other): + return not (self == other) + def add_member(self, node): self.members.append(node) @@ -199,6 +208,15 @@ class Node: return s + def __hash__(self): + return hash(self.name) + + def __eq__(self, other): + return other != None and self.name == other.name + + def __neq__(self, other): + return not (self == other) + def add_dif(self, dif): self.difs.append(dif) dif.add_member(self) |