Add more tests...

Author: isaacrobinson2000

.gitignore

@@ -1,3 +1,4 @@
 venv/
 .idea
-__pycache__
+__pycache__
+result_images

matplotview/__init__.py

@@ -67,5 +67,7 @@ def inset_zoom_axes(axes, bounds, *, image_interpolation="nearest", transform=No
     --------
     See `Axes.inset_axes` method for examples.
     """
-    inset_ax = axes.inset_axes(bounds, transform, zorder, **kwargs)
+    inset_ax = axes.inset_axes(
+        bounds, transform=transform, zorder=zorder, **kwargs
+    )
     return view(inset_ax, axes, image_interpolation)

matplotview/tests/test_inset_zoom.py

@@ -3,9 +3,39 @@
 from matplotlib.testing.decorators import check_figures_equal
 from matplotview import view, inset_zoom_axes
 
+@check_figures_equal(tol=3)
+def test_double_plot(fig_test, fig_ref):
+    np.random.seed(1)
+    im_data = np.random.rand(30, 30)
+
+    # Test case...
+    ax_test1, ax_test2 = fig_test.subplots(1, 2)
+
+    ax_test1.plot([i for i in range(10)], "r")
+    ax_test1.add_patch(plt.Circle((3, 3), 1, ec="black", fc="blue"))
+    ax_test1.imshow(im_data, origin="lower", cmap="Blues", alpha=0.5,
+                   interpolation="nearest")
+    ax_test2 = view(ax_test2, ax_test1)
+    ax_test2.set_aspect(ax_test1.get_aspect())
+    ax_test2.set_xlim(ax_test1.get_xlim())
+    ax_test2.set_ylim(ax_test1.get_ylim())
+
+    # Reference...
+    ax_ref1, ax_ref2 = fig_ref.subplots(1, 2)
+
+    ax_ref1.plot([i for i in range(10)], "r")
+    ax_ref1.add_patch(plt.Circle((3, 3), 1, ec="black", fc="blue"))
+    ax_ref1.imshow(im_data, origin="lower", cmap="Blues", alpha=0.5,
+                   interpolation="nearest")
+    ax_ref2.plot([i for i in range(10)], "r")
+    ax_ref2.add_patch(plt.Circle((3, 3), 1, ec="black", fc="blue"))
+    ax_ref2.imshow(im_data, origin="lower", cmap="Blues", alpha=0.5,
+                   interpolation="nearest")
+
+
 # Tolerance needed because the way the auto-zoom axes handles images is
 # entirely different, leading to a slightly different result.
-@check_figures_equal(tol=3)
+@check_figures_equal(tol=3.5)
 def test_auto_zoom_inset(fig_test, fig_ref):
     np.random.seed(1)
     im_data = np.random.rand(30, 30)
@@ -35,5 +65,4 @@ def test_auto_zoom_inset(fig_test, fig_ref):
     axins_ref.add_patch(plt.Circle((3, 3), 1, ec="black", fc="blue"))
     axins_ref.imshow(im_data, origin="lower", cmap="Blues", alpha=0.5,
                      interpolation="nearest")
-    ax_ref.indicate_inset_zoom(axins_ref, edgecolor="black")
-
+    ax_ref.indicate_inset_zoom(axins_ref, edgecolor="black")