U md @sNdZddlmZddlmZddZddZdd d Zdd d Z dd dZ dS)z| ======================== Widget testing utilities ======================== See also :mod:`matplotlib.tests.test_widgets`. N)mockcCs>tdd\}}|ddgddg|d|jj|S)z"Create a plot and return its axes.rg?)pltZsubplotsZplotZ set_aspectfigurecanvasZdraw)Zfigaxr S/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/testing/widgets.pyget_ax s   r cOsdS)Nr )argskwargsr r r noopsrrcCslt}||_|j||f||fgd\|_|_|||_|_||_ |j j |_ ||_ ||_ 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 : int x coord of mouse in data coords. ydata : int 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. rNZCustom)rZMockbuttonZ transDataZ transformxyxdataydataZinaxesrrkeystepZguiEventname)rrrrrreventr r r mock_events   rc Cs*t|j|||||}t||}||dS)aW Trigger an event on the given tool. Parameters ---------- tool : matplotlib.widgets.RectangleSelector etype : str The event to trigger. xdata : int x coord of mouse in data coords. ydata : int 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)rrgetattr) tooletyperrrrrrfuncr r r do_eventCs rcCs|dk r$t|d|d|dd|dt|d|d|dddt|d|d|dddt|d |d|ddd|dk rt|d |d|dd|ddS) 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`). NZ on_key_pressrr)rrrrZpress)rrrZonmovereleaseZon_key_release)r)rstartendrr r r click_and_drag\sr!)rrrNr)rrrNr)N) __doc__Zmatplotlib.pyplotZpyplotrZunittestrr rrrr!r r r r s    )