Check all of node's labels when comparing nodes for equality (#159)

Author: MelomanCool

redisgraph/node.py

@@ -54,8 +54,8 @@ def __eq__(self, rhs):
         if self.id is not None and rhs.id is not None and self.id != rhs.id:
             return False
 
-        # Label should match.
-        if self.label != rhs.label:
+        # Labels should match.
+        if self.labels != rhs.labels:
             return False
 
         # Quick check for number of properties.

tests/unit/test_node.py

@@ -50,6 +50,12 @@ def test_comparision(self):
         self.assertNotEqual(node.Node(alias="a", label="l"),
                             node.Node(alias="a", label="l1"))
 
+        self.assertEqual(node.Node(alias="a", label=["a", "b"]),
+                         node.Node(alias="a", label=["a", "b"]))
+
+        self.assertNotEqual(node.Node(alias="a", label=["a", "b"]),
+                            node.Node(alias="a", label=["a", "c"]))
+
         self.assertEqual(node.Node(properties={"a": 10}),
                          node.Node(properties={"a": 10}))
         self.assertNotEqual(node.Node(), node.Node(properties={"a": 10}))