Adding eq method to collection

pipeline/src/base.py

@@ -34,6 +34,19 @@ def has_property(self, name):
             if property.name == name:
                 return True
         return False
+
+    def __eq__(self, other: Node) -> bool:
+
+        eq = True
+        for property in self.properties:
+            property_other = getattr(other, property.name, None)
+            property_self = getattr(self, property.name, None)
+            if property_other == property_self:
+                pass
+            else:
+                eq = False
+
+        return eq
 
     def to_jsonld(self, include_empty_properties=True, embed_linked_nodes=True, with_context=True):
         """

pipeline/src/collection.py

@@ -34,6 +34,19 @@ def __len__(self):
 
     def __iter__(self):
         return iter(self.nodes.values())
+
+    def __eq__(self, other):
+
+        eq = True
+        # The current implementation assumes that nodes in both graphs are connected with the same link number.
+        for key in self.nodes.keys():
+            if key in other.nodes.keys():
+                if self.nodes[key] != other.nodes[key]:
+                    eq = False
+            else:
+                eq = False
+
+        return eq
 
     def add(self, *nodes):
         """

pipeline/tests/test_collections.py

@@ -65,6 +65,7 @@ def test_round_trip_multi_file():
     new_collection.load(test_output_dir)
 
     assert len(collection) == len(new_collection)
+    assert collection==new_collection
 
     for node in new_collection:
         if node.id == person.id: