Add test for InsetPosition and correct example

oscargus · oscargus · commit 6dd30a23fee9 · 2022-09-15T22:27:46.000+02:00
diff --git a/lib/mpl_toolkits/axes_grid1/inset_locator.py b/lib/mpl_toolkits/axes_grid1/inset_locator.py
@@ -43,7 +43,7 @@ def __init__(self, parent, lbwh):
 
         >>> parent_axes = plt.gca()
         >>> ax_ins = plt.axes([0, 0, 1, 1])
-        >>> ip = InsetPosition(ax, [0.5, 0.1, 0.4, 0.2])
+        >>> ip = InsetPosition(parent_axes, [0.5, 0.1, 0.4, 0.2])
         >>> ax_ins.set_axes_locator(ip)
         """
         self.parent = parent
diff --git a/lib/mpl_toolkits/tests/baseline_images/test_axes_grid1/insetposition.png b/lib/mpl_toolkits/tests/baseline_images/test_axes_grid1/insetposition.png
diff --git a/lib/mpl_toolkits/tests/test_axes_grid1.py b/lib/mpl_toolkits/tests/test_axes_grid1.py
@@ -21,7 +21,8 @@
     Divider, HBoxDivider, make_axes_area_auto_adjustable)
 from mpl_toolkits.axes_grid1.axes_rgb import RGBAxes
 from mpl_toolkits.axes_grid1.inset_locator import (
-    zoomed_inset_axes, mark_inset, inset_axes, BboxConnectorPatch)
+    zoomed_inset_axes, mark_inset, inset_axes, BboxConnectorPatch,
+    InsetPosition)
 import mpl_toolkits.axes_grid1.mpl_axes
 
 import pytest
@@ -571,6 +572,14 @@ def test_rgb_axes():
     ax.imshow_rgb(r, g, b, interpolation='none')
 
 
+@image_comparison(['insetposition.png'], remove_text=True)
+def test_insetposition():
+    fig, ax = plt.subplots(figsize=(2, 2))
+    ax_ins = plt.axes([0, 0, 1, 1])
+    ip = InsetPosition(ax, [0.2, 0.25, 0.5, 0.4])
+    ax_ins.set_axes_locator(ip)
+
+
 # The original version of this test relied on mpl_toolkits's slightly different
 # colorbar implementation; moving to matplotlib's own colorbar implementation
 # caused the small image comparison error.
