Robustify AffineBase.__eq__ against comparing to other classes.

anntzer · anntzer · commit 5652163fae70 · 2020-03-24T10:33:29.000+01:00
We already check whether `other` has an is_affine attribute, so we may
as well check whether it has a `get_matrix`.
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -1685,8 +1685,7 @@ def set(self, child):
 
 class AffineBase(Transform):
     """
-    The base class of all affine transformations of any number of
-    dimensions.
+    The base class of all affine transformations of any number of dimensions.
     """
     is_affine = True
 
@@ -1699,7 +1698,7 @@ def __array__(self, *args, **kwargs):
         return self.get_matrix()
 
     def __eq__(self, other):
-        if getattr(other, "is_affine", False):
+        if getattr(other, "is_affine", False) and hasattr(other, "get_matrix"):
             return np.all(self.get_matrix() == other.get_matrix())
         return NotImplemented
 
