Make sure pcolormesh log limits are good

Author: dstansby

lib/matplotlib/axes/_axes.py

@@ -6481,8 +6481,7 @@ def pcolormesh(self, *args, alpha=None, norm=None, cmap=None, vmin=None,
         maxx, maxy = np.max(coords, axis=0)
         collection.sticky_edges.x[:] = [minx, maxx]
         collection.sticky_edges.y[:] = [miny, maxy]
-        corners = (minx, miny), (maxx, maxy)
-        self.update_datalim(corners)
+        self.update_datalim(coords)
         self._request_autoscale_view()
         return collection
 

lib/matplotlib/tests/test_axes.py

@@ -1497,6 +1497,27 @@ def test_pcolormesh_nearest_noargs(fig_test, fig_ref):
     ax.pcolormesh(x, y, C, shading="nearest")
 
 
+@check_figures_equal(extensions=["png"])
+def test_pcolormesh_log_scale(fig_test, fig_ref):
+    """
+    Check that setting a log scale sets good default axis limits
+    when using pcolormesh.
+    """
+    x = np.linspace(0, 1, 11)
+    y = np.linspace(1, 2, 5)
+    X, Y = np.meshgrid(x, y)
+    C = X + Y
+
+    ax = fig_test.subplots()
+    ax.pcolormesh(X, Y, C)
+    ax.set_xscale('log')
+
+    ax = fig_ref.subplots()
+    ax.pcolormesh(X, Y, C)
+    ax.set_xlim(1e-2, 1e1)
+    ax.set_xscale('log')
+
+
 @image_comparison(['pcolormesh_datetime_axis.png'], style='mpl20')
 def test_pcolormesh_datetime_axis():
     # Remove this line when this test image is regenerated.
@@ -1550,6 +1571,8 @@ def test_pcolor_datetime_axis():
             label.set_rotation(30)
 
 
+
+
 def test_pcolorargs():
     n = 12
     x = np.linspace(-1.5, 1.5, n)