jWdZddlmZddlmZddlmZmZddlm Z dZ dZ ej dd dd Zej dd ddZddZdS)z| ======================== Widget testing utilities ======================== See also :mod:`matplotlib.tests.test_widgets`. )mock)_api) MouseEventKeyEventNctjdd\}}|ddgddg|d|j|S)z"Create a plot and return its Axes.rg?)pltsubplotsplot set_aspectcanvasdraw)figaxs /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib64/python3.11/site-packages/matplotlib/testing/widgets.pyget_axrs\l1a  GCGGQHq#hMM#JOO IcdSN)argskwargss rnooprsDrz3.11zMouseEvent or KeyEvent) alternativercDtj}||_|j||f||fgd\|_|_||c|_|_||_ | dj |_ ||_ ||_ d|_d|_|S)a Create a mock event that can stand in for `.Event` and its subclasses. This event is intended to be used in tests where it can be passed into event handling functions. Parameters ---------- ax : `~matplotlib.axes.Axes` The Axes the event will be in. xdata : float x coord of mouse in data coords. ydata : float y coord of mouse in data coords. button : None or `MouseButton` or {'up', 'down'} The mouse button pressed in this event (see also `.MouseEvent`). key : None or str The key pressed when the mouse event triggered (see also `.KeyEvent`). step : int Number of scroll steps (positive for 'up', negative for 'down'). Returns ------- event A `.Event`\-like Mock instance. rT)rootNCustom)rMockbutton transData transformxyxdataydatainaxes get_figurerkeystepguiEventname)rr r%r&r)r*events r mock_eventr.s8 IKKEEL|--u~05u~/?@@@ACEGUW$eEKEL==d=++2ELEIEJENEJ Lrzcallbacks.process(event)cnt|j|||||}t||}||dS)aT Trigger an event on the given tool. Parameters ---------- tool : matplotlib.widgets.AxesWidget etype : str The event to trigger. xdata : float x coord of mouse in data coords. ydata : float y coord of mouse in data coords. button : None or `MouseButton` or {'up', 'down'} The mouse button pressed in this event (see also `.MouseEvent`). key : None or str The key pressed when the mouse event triggered (see also `.KeyEvent`). step : int Number of scroll steps (positive for 'up', negative for 'down'). N)r.rgetattr) tooletyper r%r&r)r*r-funcs rdo_eventr4Gs=* twuc4 @ @E 4  DDKKKKKrc|j}|)tjd|||t jd||dt jd||dt jd||d|+tjd|||dSdS)a} Helper to simulate a mouse drag operation. Parameters ---------- tool : `~matplotlib.widgets.Widget` start : [float, float] Starting point in data coordinates. end : [float, float] End point in data coordinates. key : None or str An optional key that is pressed during the whole operation (see also `.KeyEvent`). Nkey_press_eventbutton_press_eventrmotion_notify_eventbutton_release_eventkey_release_event)rr_from_ax_coords_processr)r1startendr)rs rclick_and_dragr?as B  !2BsCCLLNNN3RBBKKMMM4b#qAAJJLLL5r3BBKKMMM  !4b#sCCLLNNNNNr)rrrNrr)__doc__unittestr matplotlibrmatplotlib.backend_basesrrmatplotlib.pyplotpyplotr rr deprecatedr.r4r?rrrrGs 99999999   %=>>>&&&?>&R%?@@@A@2OOOOOOr