[ 2173204 ] implot leaves environment unstable after bad image data

mdboom · mdboom · commit 55ac6dda9a3a · 2008-10-20T14:16:05.000Z
Report errors about invalid data sooner when using imshow.

svn path=/trunk/matplotlib/; revision=6283
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -285,6 +285,13 @@ def set_data(self, A, shape=None):
         else:
             self._A = np.asarray(A) # assume array
 
+        if self._A.dtype != np.uint8 and not np.can_cast(self._A.dtype, np.float):
+            raise TypeError("Image data can not convert to float")
+
+        if (self._A.ndim not in (2, 3) or
+            (self._A.ndim == 3 and self._A.shape[-1] not in (3, 4))):
+            raise TypeError("Invalid dimensions for image data")
+
         self._imcache =None
         self._rgbacache = None
         self._oldxslice = None
