Add test for pre-locked zoom/pan

kianelbo · kianelbo · commit fd99fb85c050 · 2022-07-22T11:10:38.000+04:30
diff --git a/lib/matplotlib/tests/test_backend_bases.py b/lib/matplotlib/tests/test_backend_bases.py
@@ -175,6 +175,17 @@ def test_interactive_zoom():
     assert not ax.get_autoscalex_on() and not ax.get_autoscaley_on()
 
 
+def test_widgetlock_zoompan():
+    fig, ax = plt.subplots()
+    ax.plot([0, 1], [0, 1])
+    fig.canvas.widgetlock(ax)
+    tb = NavigationToolbar2(fig.canvas)
+    tb.zoom()
+    assert ax.get_navigate_mode() is None
+    tb.pan()
+    assert ax.get_navigate_mode() is None
+
+
 @pytest.mark.parametrize("plot_func", ["imshow", "contourf"])
 @pytest.mark.parametrize("orientation", ["vertical", "horizontal"])
 @pytest.mark.parametrize("tool,button,expected",
