Make relim() take images into account too.

anntzer · anntzer · commit 06f896a6314e · 2018-01-05T02:30:45.000-08:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -1860,6 +1860,10 @@ def add_image(self, image):
         self.stale = True
         return image
 
+    def _update_image_limits(self, image):
+        xmin, xmax, ymin, ymax = image.get_extent()
+        self.axes.update_datalim(((xmin, ymin), (xmax, ymax)))
+
     def add_line(self, line):
         """
         Add a :class:`~matplotlib.lines.Line2D` to the list of plot
@@ -2036,6 +2040,10 @@ def relim(self, visible_only=False):
             if not visible_only or p.get_visible():
                 self._update_patch_limits(p)
 
+        for image in self.images:
+            if not visible_only or image.get_visible():
+                self._update_image_limits(image)
+
     def update_datalim(self, xys, updatex=True, updatey=True):
         """
         Update the data lim bbox with seq of xy tups or equiv. 2-D array
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -868,3 +868,11 @@ def test_composite(fmt, counted, composite_image, count):
     buf = io.BytesIO()
     fig.savefig(buf, format=fmt)
     assert buf.getvalue().count(counted) == count
+
+
+def test_relim():
+    fig, ax = plt.subplots()
+    ax.imshow([[0]], extent=(0, 1, 0, 1))
+    ax.relim()
+    ax.autoscale()
+    assert ax.get_xlim() == ax.get_ylim() == (0, 1)
