TST: add a test for re-scale and re-lim

jklymak · jklymak · commit dfbbcfc8dade · 2020-12-08T09:41:21.000-08:00
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -227,7 +227,7 @@ def _set_ticks_on_axis_warn(*args, **kw):
 class ColorbarAxes(Axes):
     """
     Colorbar Axes are actually two axes, a parent axes that takes care of
-    positioning the axes and an inset_axes that takes care of the drawing,
+    positioning the axes, and an inset_axes that takes care of the drawing,
     labels, ticks, etc. The inset axes is used as a way to properly
     position the triangles (or rectangles) that are used to indicate
     over/under colors.
diff --git a/lib/matplotlib/tests/baseline_images/test_colorbar/colorbar_change_lim_scale.png b/lib/matplotlib/tests/baseline_images/test_colorbar/colorbar_change_lim_scale.png
diff --git a/lib/matplotlib/tests/test_colorbar.py b/lib/matplotlib/tests/test_colorbar.py
@@ -709,3 +709,16 @@ def test_anchored_cbar_position_using_specgrid():
     np.testing.assert_allclose(
             [cx1, cx0],
             [x1 * shrink + (1 - shrink) * p0, p0 * (1 - shrink) + x0 * shrink])
+
+
+@image_comparison(['colorbar_change_lim_scale.png'], remove_text=True,
+                  style='mpl20')
+def test_colorbar_change_lim_scale():
+    fig, ax = plt.subplots(1, 2, constrained_layout=True)
+    pc = ax[0].pcolormesh(np.arange(100).reshape(10, 10)+1)
+    cb = fig.colorbar(pc, ax=ax[0], extend='both')
+    cb.ax.set_yscale('log')
+
+    pc = ax[1].pcolormesh(np.arange(100).reshape(10, 10)+1)
+    cb = fig.colorbar(pc, ax=ax[1], extend='both')
+    cb.ax.set_ylim([20, 90])
