TST: add inset axes test

jklymak · jklymak · commit 4c63f1ce4040 · 2018-04-16T15:30:09.000-07:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -77,6 +77,7 @@ def _plot_args_replacer(args, data):
                          "supported due to ambiguity of arguments.\nUse "
                          "multiple plotting calls instead.")
 
+
 class _inset_locator(object):
     """
     Helper class to locate inset axes, used in `.Axes.inset_axes`.
@@ -535,8 +536,8 @@ def inset_rectangle(self, rect, inset_ax=None, transform=None,
         if not inset_ax is None:
             # want to connect the indicator to the rect....
 
-            pos = inset_ax.get_position() # this is in fig-fraction.
-            coordsA='axes fraction'
+            pos = inset_ax.get_position()  # this is in fig-fraction.
+            coordsA = 'axes fraction'
             connects = []
             xr = [rect[0], rect[0]+rect[2]]
             yr = [rect[1], rect[1]+rect[3]]
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -5702,3 +5702,37 @@ def test_empty_errorbar_legend():
 def test_plot_columns_cycle_deprecation():
     with pytest.warns(MatplotlibDeprecationWarning):
         plt.plot(np.zeros((2, 2)), np.zeros((2, 3)))
+
+
+def test_zoom_inset():
+
+    dx, dy = 0.05, 0.05
+
+    # generate 2 2d grids for the x & y bounds
+    y, x = np.mgrid[slice(1, 5 + dy, dy),
+                    slice(1, 5 + dx, dx)]
+    z = np.sin(x)**10 + np.cos(10 + y*x) * np.cos(x)
+
+    fig, ax = plt.subplots()
+    ax.pcolormesh(x, y, z)
+    ax.set_aspect(1.)
+    ax.apply_aspect()
+    # we need to apply_aspect to make the drawing below work.
+
+    # Make the inset_axes...  Position axes co-ordinates...
+    axin1 = ax.inset_axes([0.7, 0.7, 0.35, 0.35])
+    # redraw the data in the inset axes...
+    axin1.pcolormesh(x, y, z)
+    axin1.set_xlim([1.5, 2.15])
+    axin1.set_ylim([2, 2.5])
+    axin1.set_aspect(ax.get_aspect())
+
+    rec, connectors = ax.zoom_inset_rectangle(axin1)
+    fig.canvas.draw()
+    xx = np.array([[1.5,  2.  ],
+                   [2.15, 2.5 ]])
+    assert(np.all(rec.get_bbox().get_points() == xx))
+    xx = np.array([[0.6325  , 0.692308],
+                   [0.8425  , 0.907692]])
+    np.testing.assert_allclose(axin1.get_position().get_points(),
+            xx, rtol=1e-4)
