Improve test for correct clipping

Zac-HD · Zac-HD · commit e05a632d46f0 · 2018-01-18T21:12:08.000+11:00
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -801,16 +801,18 @@ def test_imshow_clips_rgb_to_valid_range(dtype):
     arr = np.arange(300, dtype=dtype).reshape((10, 10, 3))
     if dtype.kind != 'u':
         arr -= 10
+    too_low = arr < 0
+    too_high = arr > 255
     if dtype.kind == 'f':
         arr = arr / 255
     _, ax = plt.subplots()
     out = ax.imshow(arr).get_array()
-    assert out.min() == 0
+    assert (out[too_low] == 0).all()
     if dtype.kind == 'f':
-        assert out.max() == 1
+        assert (out[too_high] == 1).all()
         assert out.dtype.kind == 'f'
     else:
-        assert out.max() == 255
+        assert (out[too_high] == 255).all()
         assert out.dtype == np.uint8
 
 
