Fix problem in (Vertex/Edge)PropertyMap.__eq__() when compared to other types

Fix problem in (Vertex/Edge)PropertyMap.eq() when compared to other types
f9684ed2 · Tiago Peixoto · 9e70cee8 · f9684ed2
Commit f9684ed2 authored Jun 25, 2020 by Tiago Peixoto
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/graph_tool/__init__.py src/graph_tool/__init__.py +4 -0

No files found.
--- a/src/graph_tool/__init__.py
+++ b/src/graph_tool/__init__.py
@@ -856,6 +856,8 @@ class VertexPropertyMap(PropertyMap):
        return PropertyArray(a, self)

    def __eq__(self, other):
+        if not isinstance(other, VertexPropertyMap):
+            return False
        g = self.get_graph()
        if g.base is not other.get_graph().base:
            return False
@@ -898,6 +900,8 @@ class EdgePropertyMap(PropertyMap):
        return PropertyArray(a, self)

    def __eq__(self, other):
+        if not isinstance(other, EdgePropertyMap):
+            return False
        g = self.get_graph()
        if g.base is not other.get_graph().base:
            return False