Merge pull request #24556 from leejjoon/fix-axesimage-window-extent

tacaswell · web-flow · commit 9b7c902fd72e · 2022-11-30T09:19:02.000-05:00
Fix incorrect window_extent of AxesImage
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -935,7 +935,7 @@ def __init__(self, ax,
     def get_window_extent(self, renderer=None):
         x0, x1, y0, y1 = self._extent
         bbox = Bbox.from_extents([x0, y0, x1, y1])
-        return bbox.transformed(self.axes.transData)
+        return bbox.transformed(self.get_transform())
 
     def make_image(self, renderer, magnification=1.0, unsampled=False):
         # docstring inherited
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -589,6 +589,20 @@ def test_get_window_extent_for_AxisImage():
 
     assert_array_equal(im_bbox.get_points(), [[400, 200], [700, 900]])
 
+    fig, ax = plt.subplots(figsize=(10, 10), dpi=100)
+    ax.set_position([0, 0, 1, 1])
+    ax.set_xlim(1, 2)
+    ax.set_ylim(0, 1)
+    im_obj = ax.imshow(
+        im, extent=[0.4, 0.7, 0.2, 0.9], interpolation='nearest',
+        transform=ax.transAxes)
+
+    fig.canvas.draw()
+    renderer = fig.canvas.renderer
+    im_bbox = im_obj.get_window_extent(renderer)
+
+    assert_array_equal(im_bbox.get_points(), [[400, 200], [700, 900]])
+
 
 @image_comparison(['zoom_and_clip_upper_origin.png'],
                   remove_text=True, style='mpl20')
