add __eq__ method to SparseEmbedding

haystack/dataclasses/sparse_embedding.py

@@ -20,6 +20,9 @@ def __init__(self, indices: List[int], values: List[float]):
         self.indices = indices
         self.values = values
 
+    def __eq__(self, other):
+        return self.indices == other.indices and self.values == other.values
+
     def to_dict(self):
         """
         Convert the sparse embedding to a dictionary.

releasenotes/notes/sparse-emb-eq-773ef04ae3ed83ea.yaml

@@ -0,0 +1,4 @@
+---
+enhancements:
+  - |
+    Add an `__eq__` method to `SparseEmbedding` class to compare two `SparseEmbedding` objects.

test/dataclasses/test_sparse_embedding.py

@@ -21,3 +21,11 @@ def test_from_dict(self):
         se = SparseEmbedding.from_dict({"indices": [0, 2, 4], "values": [0.1, 0.2, 0.3]})
         assert se.indices == [0, 2, 4]
         assert se.values == [0.1, 0.2, 0.3]
+
+    def test_eq(self):
+        se1 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
+        se2 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
+        assert se1 == se2
+
+        se3 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.4])
+        assert se1 != se3