diff options
author | dvora-h <67596500+dvora-h@users.noreply.github.com> | 2023-01-08 10:42:19 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-01-08 10:42:19 +0200 |
commit | c7600b4c7e558d3c05485f76cfcdd3e807c6b13e (patch) | |
tree | 7376b115b09cb37073a35e30cc587fd4cf63f968 | |
parent | decd1f6ae5434b111c5b125263549d8743275e51 (diff) | |
download | redis-py-c7600b4c7e558d3c05485f76cfcdd3e807c6b13e.tar.gz |
add type checking for graph __eq__ (#2531)
-rw-r--r-- | redis/commands/graph/edge.py | 4 | ||||
-rw-r--r-- | redis/commands/graph/node.py | 4 | ||||
-rw-r--r-- | redis/commands/graph/path.py | 4 |
3 files changed, 12 insertions, 0 deletions
diff --git a/redis/commands/graph/edge.py b/redis/commands/graph/edge.py index b0141d9..6ee195f 100644 --- a/redis/commands/graph/edge.py +++ b/redis/commands/graph/edge.py @@ -61,6 +61,10 @@ class Edge: return res def __eq__(self, rhs): + # Type checking + if not isinstance(rhs, Edge): + return False + # Quick positive check, if both IDs are set. if self.id is not None and rhs.id is not None and self.id == rhs.id: return True diff --git a/redis/commands/graph/node.py b/redis/commands/graph/node.py index 0ebe510..4546a39 100644 --- a/redis/commands/graph/node.py +++ b/redis/commands/graph/node.py @@ -65,6 +65,10 @@ class Node: return res def __eq__(self, rhs): + # Type checking + if not isinstance(rhs, Node): + return False + # Quick positive check, if both IDs are set. if self.id is not None and rhs.id is not None and self.id != rhs.id: return False diff --git a/redis/commands/graph/path.py b/redis/commands/graph/path.py index 6f2214a..ee22dc8 100644 --- a/redis/commands/graph/path.py +++ b/redis/commands/graph/path.py @@ -54,6 +54,10 @@ class Path: return self def __eq__(self, other): + # Type checking + if not isinstance(other, Path): + return False + return self.nodes() == other.nodes() and self.edges() == other.edges() def __str__(self): |