Merge pull request #16893 from anntzer/affineq

QuLogic · web-flow · commit ff365ec4354f · 2020-03-26T18:49:59.000-04:00
Robustify `AffineBase.__eq__` against comparing to other classes.
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -1611,8 +1611,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
 
@@ -1625,7 +1624,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
 
