matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_image/mask_image_over_under.png
26.6 KB b/‎lib/matplotlib/tests/baseline_images/test_image/mask_image_over_under.png
26.6 KB
diff --git a/‎lib/matplotlib/tests/test_image.py
Lines changed: 40 additions & 2 deletions b/‎lib/matplotlib/tests/test_image.py
Lines changed: 40 additions & 2 deletions
@@ -24,8 +24,12 @@
 from numpy.testing import (
     assert_array_equal, assert_array_almost_equal, assert_allclose)
 from matplotlib.testing.noseclasses import KnownFailureTest
-
-
+from copy import copy
+from numpy import ma
+import matplotlib.colors as colors
+import matplotlib.pyplot as plt
+import matplotlib.mlab as mlab
+import numpy as np
 
 import nose
 
@@ -658,6 +662,40 @@ def test_image_preserve_size2():
                        np.identity(n, bool)[::-1])
 
 
+@image_comparison(baseline_images=['mask_image_over_under'],
+                  remove_text=True, extensions=['png'])
+def test_mask_image_over_under():
+    delta = 0.025
+    x = y = np.arange(-3.0, 3.0, delta)
+    X, Y = np.meshgrid(x, y)
+    Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
+    Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
+    Z = 10*(Z2 - Z1)  # difference of Gaussians
+
+    palette = copy(plt.cm.gray)
+    palette.set_over('r', 1.0)
+    palette.set_under('g', 1.0)
+    palette.set_bad('b', 1.0)
+    Zm = ma.masked_where(Z > 1.2, Z)
+    fig, (ax1, ax2) = plt.subplots(1, 2)
+    im = ax1.imshow(Zm, interpolation='bilinear',
+                    cmap=palette,
+                    norm=colors.Normalize(vmin=-1.0, vmax=1.0, clip=False),
+                    origin='lower', extent=[-3, 3, -3, 3])
+    ax1.set_title('Green=low, Red=high, Blue=bad')
+    fig.colorbar(im, extend='both', orientation='horizontal',
+                 ax=ax1, aspect=10)
+
+    im = ax2.imshow(Zm, interpolation='nearest',
+                    cmap=palette,
+                    norm=colors.BoundaryNorm([-1, -0.5, -0.2, 0, 0.2, 0.5, 1],
+                                             ncolors=256, clip=False),
+                    origin='lower', extent=[-3, 3, -3, 3])
+    ax2.set_title('With BoundaryNorm')
+    fig.colorbar(im, extend='both', spacing='proportional',
+                 orientation='horizontal', ax=ax2, aspect=10)
+
+
 @image_comparison(baseline_images=['mask_image'],
                   remove_text=True)
 def test_mask_image():