matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_colors/test_offset_norm.p 8000 ng
81.7 KB b/‎lib/matplotlib/tests/baseline_images/test_colors/test_offset_norm.p 8000 ng
81.7 KB
diff --git a/‎lib/matplotlib/tests/test_colors.py
Lines changed: 17 additions & 0 deletions b/‎lib/matplotlib/tests/test_colors.py
Lines changed: 17 additions & 0 deletions
@@ -275,6 +275,23 @@ def test_premature_scaling(self):
         norm.inverse(np.array([0.1, 0.5, 0.9]))
 
 
+@image_comparison(baseline_images=['test_offset_norm'], extensions=['png'])
+def test_offset_norm_img():
+    x = np.linspace(-2, 7)
+    y = np.linspace(-1*np.pi, np.pi)
+    X, Y = np.meshgrid(x, y)
+    Z = x * np.sin(Y)**2
+
+    fig, (ax1, ax2) = plt.subplots(ncols=2)
+    cmap = plt.cm.coolwarm
+    norm = mcolors.OffsetNorm(vmin=-2, vcenter=0, vmax=7)
+
+    img1 = ax1.imshow(Z, cmap=cmap, norm=None)
+    cbar1 = fig.colorbar(img1, ax=ax1)
+
+    img2 = ax2.imshow(Z, cmap=cmap, norm=norm)
+    cbar2 = fig.colorbar(img2, ax=ax2)
+
 def test_SymLogNorm():
     """
     Test SymLogNorm behavior