Ng FdZddlmZddlmZdZdZd dZd dZ d d Z dS) z| ======================== Widget testing utilities ======================== See also :mod:`matplotlib.tests.test_widgets`. )mockNctjdd\}}|ddgddg|d|jj|S)z"Create a plot and return its Axes.rg?)pltsubplotsplot set_aspectfigurecanvasdraw)figaxs V/var/www/html/ai-engine/env/lib/python3.11/site-packages/matplotlib/testing/widgets.pyget_axrsal1a  GCGGQHq#hMM#I IcdSN)argskwargss rnooprsDrrc&tj}||_|j||f||fgd\|_|_||c|_|_||_ |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 : 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. rNCustom)rMockbutton transData transformxyxdataydatainaxesr r keystepguiEventname)rrr!r"r$r%events r mock_eventr)s6 IKKEEL|--u~05u~/?@@@ACEGUW$eEKEL9#ELEIEJENEJ Lrcnt|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) tooletyperr!r"r$r%r(funcs rdo_eventr/Ds=( twuc4 @ @E 4  DDKKKKKrcV|!t|d|d|dd|t|d|d|ddt|d|d|ddt|d |d|dd|#t|d |d|dd|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`). N on_key_pressrr)r!r"rr$press)r!r"ronmovereleaseon_key_release)r/)r,startendr$s rclick_and_dragr8]s ~U1XU1Xs $ $ $ $ T7%(%(1EEEE T83q6QBBBB T9CF#a&CCCC 's1vSVs $ $ $ $ $ $r)rrrNrr) __doc__unittestrmatplotlib.pyplotpyplotrrrr)r/r8rrrr=s   &&&&R2$$$$$$r