TST: add inset axes test

jklymak · jklymak · commit 082b8a6be437 · 2018-04-16T16:12:47.000-07:00
diff --git a/examples/mplot3d/lines3d.py b/examples/mplot3d/lines3d.py
@@ -23,7 +23,7 @@
 x = r * np.sin(theta)
 y = r * np.cos(theta)
 
-ax.plot(x, y, z, label='parametric curve')
+l = ax.plot(x, y, z, label='parametric curve')
 ax.legend()
 
 plt.show()
diff --git a/examples/subplots_axes_and_figures/zoom_inset_axes.py b/examples/subplots_axes_and_figures/zoom_inset_axes.py
@@ -10,6 +10,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
+
 def get_demo_image():
     from matplotlib.cbook import get_sample_data
     import numpy as np
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`.
@@ -532,11 +533,11 @@ def inset_rectangle(self, rect, inset_ax=None, transform=None,
                 zorder=zorder,  label=label, transform=transform, **kwargs)
         self.add_patch(rectpatch)
 
-        if not inset_ax is None:
+        if inset_ax is not 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)
