improve tests

johan12345 · johan12345 · commit 45e5bc77005f · 2020-10-03T19:21:16.000+02:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4073,16 +4073,38 @@ def test_axline(fig_test, fig_ref):
 
 
 @check_figures_equal()
-def test_axline_transform(fig_test, fig_ref):
+def test_axline_transaxes(fig_test, fig_ref):
     ax = fig_test.subplots()
     ax.set(xlim=(-1, 1), ylim=(-1, 1))
     ax.axline_transaxes((0, 0), slope=1)
     ax.axline_transaxes((1, 0.5), slope=1, color='C1')
+    ax.axline_transaxes((0.5, 0.5), slope=0, color='C2')
 
     ax = fig_ref.subplots()
     ax.set(xlim=(-1, 1), ylim=(-1, 1))
     ax.plot([-1, 1], [-1, 1])
     ax.plot([0, 1], [-1, 0], color='C1')
+    ax.plot([-1, 1], [0, 0], color='C2')
+
+
+@check_figures_equal()
+def test_axline_transaxes_panzoom(fig_test, fig_ref):
+    # test that it is robust against pan/zoom and
+    # figure resize after plotting
+    ax = fig_test.subplots()
+    ax.set(xlim=(-1, 1), ylim=(-1, 1))
+    ax.axline_transaxes((0, 0), slope=1)
+    ax.axline_transaxes((0.5, 0.5), slope=2, color='C1')
+    ax.axline_transaxes((0.5, 0.5), slope=0, color='C2')
+    ax.set(xlim=(0, 5), ylim=(0, 10))
+    fig_test.set_size_inches(3, 3)
+
+    ax = fig_ref.subplots()
+    ax.set(xlim=(0, 5), ylim=(0, 10))
+    fig_ref.set_size_inches(3, 3)
+    ax.plot([0, 5], [0, 5])
+    ax.plot([0, 5], [0, 10], color='C1')
+    ax.plot([0, 5], [5, 5], color='C2')
 
 
 def test_axline_args():
