Add rubberband in toolmanager to remove warning

rohit-rawat-amdocs · rohit-rawat-amdocs · commit 3e3472482b6e · 2020-09-24T09:21:54.000+05:30
diff --git a/lib/matplotlib/tests/test_backend_bases.py b/lib/matplotlib/tests/test_backend_bases.py
@@ -3,7 +3,7 @@
 from matplotlib.backend_bases import (
     FigureCanvasBase, LocationEvent, MouseButton, MouseEvent,
     NavigationToolbar2, RendererBase)
-from matplotlib.backend_tools import (ToolZoom, ToolPan,
+from matplotlib.backend_tools import (ToolZoom, ToolPan, RubberbandBase,
                                       ToolViewsPositions, _views_positions)
 import matplotlib.pyplot as plt
 import matplotlib.transforms as transforms
@@ -172,13 +172,13 @@ def test_toolbar_zoompan():
                                                       tool=ToolPan)
         ax.figure.canvas.manager.toolmanager.add_tool(name=_views_positions,
                                                       tool=ToolViewsPositions)
+        ax.figure.canvas.manager.toolmanager.add_tool(name='rubberband',
+                                                      tool=RubberbandBase)
         ax.figure.canvas.manager.toolmanager.trigger_tool('zoom')
         assert ax.get_navigate_mode() == "ZOOM"
         ax.figure.canvas.manager.toolmanager.trigger_tool('pan')
         assert ax.get_navigate_mode() == "PAN"
         assert(len(rec) == 4)
-        for r in rec[:-1]:
+        for r in rec:
             assert("The new Tool classes introduced in v1.5 are experimental"
                    in str(r.message))
-        assert("ToolManager does not control tool rubberband"
-               in str(rec[-1].message))
