Add tests for ToolManager

oscargus · oscargus · commit b2d78d146f74 · 2022-10-17T16:27:45.000+02:00
diff --git a/lib/matplotlib/backend_managers.py b/lib/matplotlib/backend_managers.py
@@ -185,7 +185,7 @@ def update_keymap(self, name, key):
             Keys to associate with the tool.
         """
         if name not in self._tools:
-            raise KeyError(f'{name} not in Tools')
+            raise KeyError(f'{name!r} not in Tools')
         self._remove_keys(name)
         if isinstance(key, str):
             key = [key]
@@ -404,6 +404,7 @@ def get_tool(self, name, warn=True):
             return name
         if name not in self._tools:
             if warn:
-                _api.warn_external(f"ToolManager does not control tool {name}")
+                _api.warn_external(
+                    f"ToolManager does not control tool {name!r}")
             return None
         return self._tools[name]
diff --git a/lib/matplotlib/tests/test_backend_bases.py b/lib/matplotlib/tests/test_backend_bases.py
@@ -4,6 +4,7 @@
 from matplotlib.backend_bases import (
     FigureCanvasBase, LocationEvent, MouseButton, MouseEvent,
     NavigationToolbar2, RendererBase)
+from matplotlib.backend_tools import RubberbandBase
 from matplotlib.figure import Figure
 from matplotlib.testing._markers import needs_pgf_xelatex
 import matplotlib.pyplot as plt
@@ -349,3 +350,56 @@ def test_interactive_pan(key, mouseend, expectedxlim, expectedylim):
     # Should be close, but won't be exact due to screen integer resolution
     assert tuple(ax.get_xlim()) == pytest.approx(expectedxlim, abs=0.02)
     assert tuple(ax.get_ylim()) == pytest.approx(expectedylim, abs=0.02)
+
+
+def test_toolmanager_remove():
+    expected_warning_regex = (
+        r"Treat the new Tool classes introduced in "
+        r"v[0-9]*.[0-9]* as experimental for now; "
+        "the API and rcParam may change in future versions.")
+    with pytest.warns(UserWarning, match=expected_warning_regex):
+        plt.rcParams['toolbar'] = 'toolmanager'
+    fig = plt.gcf()
+    initial_len = len(fig.canvas.manager.toolmanager.tools)
+    assert 'forward' in fig.canvas.manager.toolmanager.tools.keys()
+    fig.canvas.manager.toolmanager.remove_tool('forward')
+    assert len(fig.canvas.manager.toolmanager.tools) == initial_len - 1
+    assert 'forward' not in fig.canvas.manager.toolmanager.tools.keys()
+
+
+def test_toolmanager_get_tool():
+    expected_warning_regex = (
+        r"Treat the new Tool classes introduced in "
+        r"v[0-9]*.[0-9]* as experimental for now; "
+        "the API and rcParam may change in future versions.")
+    with pytest.warns(UserWarning, match=expected_warning_regex):
+        plt.rcParams['toolbar'] = 'toolmanager'
+    fig = plt.gcf()
+    rubberband = fig.canvas.manager.toolmanager.get_tool('rubberband')
+    assert isinstance(rubberband, RubberbandBase)
+    assert fig.canvas.manager.toolmanager.get_tool(rubberband) is rubberband
+    with pytest.warns(UserWarning,
+                      match="ToolManager does not control tool 'foo'"):
+        assert fig.canvas.manager.toolmanager.get_tool('foo') is None
+    assert fig.canvas.manager.toolmanager.get_tool('foo', warn=False) is None
+
+    with pytest.warns(UserWarning,
+                      match="ToolManager does not control tool 'foo'"):
+        assert fig.canvas.manager.toolmanager.trigger_tool('foo') is None
+
+
+def test_toolmanager_update_keymap():
+    expected_warning_regex = (
+        r"Treat the new Tool classes introduced in "
+        r"v[0-9]*.[0-9]* as experimental for now; "
+        "the API and rcParam may change in future versions.")
+    with pytest.warns(UserWarning, match=expected_warning_regex):
+        plt.rcParams['toolbar'] = 'toolmanager'
+    fig = plt.gcf()
+    assert 'v' in fig.canvas.manager.toolmanager.get_tool_keymap('forward')
+    with pytest.warns(UserWarning,
+                      match="Key c changed from back to forward"):
+        fig.canvas.manager.toolmanager.update_keymap('forward', 'c')
+    assert fig.canvas.manager.toolmanager.get_tool_keymap('forward') == ['c']
+    with pytest.raises(KeyError, match="'foo' not in Tools"):
+        fig.canvas.manager.toolmanager.update_keymap('foo', 'c')
