Merge pull request #9146 from tacaswell/fix_bool_imshow

tacaswell · web-flow · commit b7228106c7e2 · 2017-09-01T01:05:02.000-04:00
FIX: cast max/min to scaled dtype
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -367,8 +367,8 @@ def _make_image(self, A, in_bbox, out_bbox, clip_bbox, magnification=1.0,
                     scaled_dtype = np.float32
                 # old versions of numpy do not work with `np.nammin`
                 # and `np.nanmax` as inputs
-                a_min = np.ma.min(A)
-                a_max = np.ma.max(A)
+                a_min = np.ma.min(A).astype(scaled_dtype)
+                a_max = np.ma.max(A).astype(scaled_dtype)
                 # scale the input data to [.1, .9].  The Agg
                 # interpolators clip to [0, 1] internally, use a
                 # smaller input scale to identify which of the
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -802,3 +802,8 @@ def test_empty_imshow():
 def test_imshow_float128():
     fig, ax = plt.subplots()
     ax.imshow(np.zeros((3, 3), dtype=np.longdouble))
+
+
+def test_imshow_bool():
+    fig, ax = plt.subplots()
+    ax.imshow(np.array([[True, False], [False, True]], dtype=bool))
