__eq__ and __hash__ (TODO)

Author: markotoplak

Orange/clustering/clustering.py

@@ -57,6 +57,20 @@ def predict(self, X):
         raise NotImplementedError(
             "This clustering algorithm does not support predicting.")
 
+    def __eq__(self, other):
+        if self is other:
+            return True
+        print(self.labels, other.labels)
+        return type(self) is type(other) \
+            and self.projector == other.projector \
+            and self.domain == other.domain \
+            and self.original_domain == other.original_domain \
+            and self.labels == other.labels
+
+    def __hash__(self):
+        # TODO handle labels
+        return hash((self.projector, self.domain, self.original_domain))
+
 
 class Clustering(metaclass=WrapperMeta):
     """

Orange/clustering/kmeans.py

@@ -19,6 +19,15 @@ def __init__(self, projector):
     def predict(self, X):
         return self.projector.predict(X)
 
+    def __eq__(self, other):
+        return super().__eq__(other) \
+               and self.centroids == other.centroids \
+               and self.k == other.k
+
+    def __hash__(self):
+        # TODO self.centroids
+        return hash((super().__hash__(), self.k))
+
 
 class KMeans(Clustering):