def test_interactive_colorbar(plot_func, orientation, tool, button, expected): fig, ax = plt.subplots() data = np.arange(12).reshape((4, 3)) vmin0, vmax0 = 0, 10 coll = getattr(ax, plot_func)(data, vmin=vmin0, vmax=vmax0) cb = fig.colorbar(coll, ax=ax, orientation=orientation) if plot_func == "contourf": # Just determine we can't navigate and exit out of the test assert not cb.ax.get_navigate() return assert cb.ax.get_navigate() # Mouse from 4 to 6 (data coordinates, "d"). vmin, vmax = 4, 6 # The y coordinate doesn't matter, it just needs to be between 0 and 1 # However, we will set d0/d1 to the same y coordinate to test that small # pixel changes in that coordinate doesn't cancel the zoom like a normal # axes would. d0 = (vmin, 0.5) d1 = (vmax, 0.5) # Swap them if the orientation is vertical if orientation == "vertical": d0 = d0[::-1] d1 = d1[::-1] # Convert to screen coordinates ("s"). Events are defined only with pixel # precision, so round the pixel values, and below, check against the # corresponding xdata/ydata, which are close but not equal to d0/d1. s0 = cb.ax.transData.transform(d0).astype(int) s1 = cb.ax.transData.transform(d1).astype(int) # Set up the mouse movements start_event = MouseEvent("button_press_event", fig.canvas, *s0, button) stop_event = MouseEvent("button_release_event", fig.canvas, *s1, button) tb = NavigationToolbar2(fig.canvas) if tool == "zoom": tb.zoom() tb.press_zoom(start_event) tb.drag_zoom(stop_event) tb.release_zoom(stop_event) else: tb.pan() tb.press_pan(start_event) tb.drag_pan(stop_event) tb.release_pan(stop_event) # Should be close, but won't be exact due to screen integer resolution assert (cb.vmin, cb.vmax) == pytest.approx(expected, abs=0.15)
def test_interactive_zoom(): fig, ax = plt.subplots() ax.set(xscale="logit") assert ax.get_navigate_mode() is None tb = NavigationToolbar2(fig.canvas) tb.zoom() assert ax.get_navigate_mode() == 'ZOOM' xlim0 = ax.get_xlim() ylim0 = ax.get_ylim() # Zoom from x=1e-6, y=0.1 to x=1-1e-5, 0.8 (data coordinates, "d"). d0 = (1e-6, 0.1) d1 = (1 - 1e-5, 0.8) # Convert to screen coordinates ("s"). Events are defined only with pixel # precision, so round the pixel values, and below, check against the # corresponding xdata/ydata, which are close but not equal to d0/d1. s0 = ax.transData.transform(d0).astype(int) s1 = ax.transData.transform(d1).astype(int) # Zoom in. start_event = MouseEvent("button_press_event", fig.canvas, *s0, MouseButton.LEFT) fig.canvas.callbacks.process(start_event.name, start_event) stop_event = MouseEvent("button_release_event", fig.canvas, *s1, MouseButton.LEFT) fig.canvas.callbacks.process(stop_event.name, stop_event) assert ax.get_xlim() == (start_event.xdata, stop_event.xdata) assert ax.get_ylim() == (start_event.ydata, stop_event.ydata) # Zoom out. start_event = MouseEvent("button_press_event", fig.canvas, *s1, MouseButton.RIGHT) fig.canvas.callbacks.process(start_event.name, start_event) stop_event = MouseEvent("button_release_event", fig.canvas, *s0, MouseButton.RIGHT) fig.canvas.callbacks.process(stop_event.name, stop_event) # Absolute tolerance much less than original xmin (1e-7). assert ax.get_xlim() == pytest.approx(xlim0, rel=0, abs=1e-10) assert ax.get_ylim() == pytest.approx(ylim0, rel=0, abs=1e-10) tb.zoom() assert ax.get_navigate_mode() is None assert not ax.get_autoscalex_on() and not ax.get_autoscaley_on()