OwlCyberSecurity - MANAGER
Edit File: test_widgets.cpython-38.pyc
U ��]�D � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dl m Z d dl Z dd� Z d5dd �Z d d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zed gd gd d d �d!d"� �Zed#gd d gd d$�d%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z dS )6� )�MockN)�image_comparison)�assert_allclosec C s> t �dd�\} }|�ddgddg� |�d� |jj�� |S )N� r �� � �?)�plt�subplotsZplotZ set_aspect�figure�canvasZdraw��fig�ax� r �?/usr/lib/python3/dist-packages/matplotlib/tests/test_widgets.py�get_ax s r r c C s� t � }||_| j}|j�||f||fg�d \|_|_|| |_|_||_ |j j |_ ||_ ||_ d|_d|_t| |�} | |� dS )a *name* the event name *canvas* the FigureCanvas instance generating the event *guiEvent* the GUI event that triggered the matplotlib event *x* x position - pixels from left of canvas *y* y position - pixels from bottom of canvas *inaxes* the :class:`~matplotlib.axes.Axes` instance if mouse is over axes *xdata* x coord of mouse in data coords *ydata* y coord of mouse in data coords *button* button pressed None, 1, 2, 3, 'up', 'down' (up and down are used for scroll events) *key* the key depressed when the mouse event triggered (see :class:`KeyEvent`) *step* number of scroll steps (positive for 'up', negative for 'down') r NZCustom)r �buttonr Z transDataZ transform�x�y�xdata�ydataZinaxesr r �key�stepZguiEvent�name�getattr) �tool�etyper r r r r Zeventr �funcr r r �do_event s" % �� r c s� t � � � fdd�}tj� |f| �}t|ddddd� t|ddddd� t|d d d dd� | �d d �d kr�t|jd ddddgdddddgg|jd� � js�t�d S )Nc sB d� _ | jdkst�| jdks"t�|jdks0t�|jdks>t�d S )NT�d �� )� _got_onselectr �AssertionErrorr �ZepressZerelease�r r r �onselectM s z!check_rectangle.<locals>.onselect�pressr r �r r r �onmover �release� �drawtype)�line�noneg Y@)�err_msg) r �widgets�RectangleSelectorr �getr �geometryr! r" ��kwargsr% r r r$ r �check_rectangleJ s ��r5 c C sN t � t ddd� t ddd� t dddd � t ddd d � t tdd �d � d S )Nr, F)r+ �useblitTr �r6 r r- � )r+ �minspanx�minspanyZpixels)r9 r: Z spancoords�Zfill�Z rectprops)r5 �dictr r r r �test_rectangle_selectord s r> c C s t � } dd� }tj| |ddd�}d|_t|ddddd d � t|d d d dd � t|dd d dd � |jdkslt�t|dddddd � t|ddddd � t|d dddd � t|ddddd � t|dddddd � |jdks�t�t|dddddd � t|ddddd � t|d dd dd � t|ddd dd � t|dddddd � dd� |jD �}|ddddgk�s\t�t|dddddd � t|ddddd � t|d dd dd � t|ddd dd � t|dddddd � d d� |jD �}|d d d d gk�s�t�|jjd k�s�t�t|jd!d!�d"f d#dg� d!S )$z'For ellipse, test out the key modifiersc S s d S �Nr r# r r r r% q s z test_ellipse.<locals>.onselectr8 T�r% Zmaxdist� interactive�r � r rC r& r � )r r r r r( � r' r) )�x � rF rG � on_key_pressr �control�} �on_key_release)�K rJ rL rJ �shift�# c S s g | ] }t |��qS r ��int��.0�er r r � <listcomp>� s z test_ellipse.<locals>.<listcomp>�" z ctrl+shift� c S s g | ] }t |��qS r rO rQ r r r rT � s �F � )� �I Nr g �Q@) r r/ ZEllipseSelector�extentsr r" r2 �shaper )r r% r r[ r r r � test_ellipsem s^ � � � � � � � �r] c C s t � } dd� }tj| |ddd�}d|_|jdks4t�|jdksBt�|jdksPt�|jdks^t�t|d d d d � t|d d d d � t|dd d d � |jdks�t�t|d ddd � t|d d d d � t|dd d d � |jdks�t�t|d ddd � t|d d d d � t|dd d d � |jdk�st�d S )Nc S s d S r? r r# r r r r% � s z(test_rectangle_handles.<locals>.onselectr8 Tr@ rB ))r rC rC r )r r rC rC ))r � @_@rC r^ )r^ r r^ rC r& r �r r r( rF r) )rF rC rF rC � )�l � ra rb )r8 r r8 r )r r/ r0 r[ Zcornersr" Z edge_centersr )r r% r r r r �test_rectangle_handles� s. �rc c s� t � � � fdd�}� fdd�}d|kr.||d< tj� |f| �|�}t|ddddd � t|d d d dd � t|d d d dd � � js�t�d|kr�� js�t�d S )Nc s" d� _ | dkst�|dks t�d S )NTr rC �r! r" �ZvminZvmaxr$ r r r% � s z check_span.<locals>.onselectc s" | dks t �|dkst �d� _d S )Nr rJ T)r" � _got_on_movere r$ r r r( � s zcheck_span.<locals>.onmove�onmove_callbackr& r r r' r( rJ r) rC )r r/ Z SpanSelectorr r! r" rf )�argsr4 r% r( r r r$ r � check_span� s ri c C s2 t dddd� t dddd� t dtdd�d � d S ) N� horizontalr8 T)Zminspanr6 �verticalr )rg r r; r< )ri r= r r r r �test_span_selector� s rl c sf t � � � fdd�}tj� |f| �}t|ddddd� t|ddddd� t|d d d dd� � jsbt�d S ) Nc s d� _ | dddgkst�d S )NT)r r )rJ rJ )rC rC rd )Zvertsr$ r r r% � s z&check_lasso_selector.<locals>.onselectr& r r r' r( rJ r) rC )r r/ Z LassoSelectorr r! r" r3 r r$ r �check_lasso_selector� s rm c C s( t � t dtdd�d� t ddd� d S )NFZred)Zcolor)r6 Z linepropsTr r7 )rm r= r r r r �test_lasso_selector� s rn c C sf t � } t�| dd�}|�� dddgks*t�|�d� |�� dddgksJt�|�dd� �}|�|� d S )N)�a�b�c)TFTTFr c S s d S r? r r r r r �<lambda> � z#test_CheckButtons.<locals>.<lambda>)r r/ � CheckButtonsZ get_statusr" Z set_activeZ on_clickedZ disconnect)r ZcheckZcidr r r �test_CheckButtons� s ru Zcheck_radio_buttonsZpngZmpl20T)�baseline_images� extensions�style� remove_textc C sT t � tjdd� t�ddddg�} t�ddddg�}t�| d� t�|dd � d S ) Ng333333�?)�left皙�����?�ffffff�?�333333�?g�������?)zRadio 1zRadio 2zRadio 3)zCheck 1zCheck 2zCheck 3)FTT)r r Zsubplots_adjust�axesr/ � RadioButtonsrt )Zrax1Zrax2r r r � test_check_radio_buttons_image s �r� Z check_bunch_of_radio_buttons)rv rx rw ry c C s" t �ddddg�} t�| d� d S )Nr{ g�������?r} r| )ZB1ZB2ZB3ZB4ZB5ZB6ZB7ZB8ZB9ZB10ZB11ZB12ZB13ZB14ZB15)r r~ r/ r )Zraxr r r �!test_check_bunch_of_radio_buttons s r� c C sd t �� \} }t�t�� tj|ddddd� W 5 Q R X t�t�� tj|ddddd� W 5 Q R X d S )N� � � 8@� $@)r �label�valmin�valmax� slidermin)r r� r� r� � slidermax)r r �pytestZraises� ValueErrorr/ �Sliderr r r r �'test_slider_slidermin_slidermax_invalid s � �r� c C sp t �� \} }tj|ddddd�}tj|dddd|d�}|j|jksFt�tj|dddd|d �}|j|jkslt�d S ) Nr� r� r� g @�r r� r� r� �valinitr )r r� r� r� r� r� r� )r r� r� r� r� r� )r r r/ r� �valr" )r r Zslider_�sliderr r r � test_slider_slidermin_slidermax) s � � �r� c C sX t �� \} }tj|ddddd�}|j|jks0t�tj|ddddd�}|j|jksTt�d S )Nr� r� r� g $�r� g 9@)r r r/ r� r� r� r" r� )r r r� r r r �test_slider_valmin_valmax7 s � �r� c C s� t �� \} }tj|dddddd�}|�d� |jdks:t�|j�� � |j � � �}t |j dddd g� t �� \} }tj|ddddd d�}|�d� |jdks�t�|j�� � |j � � �}t |j ddd dg� d S ) Nr� r � � rj )r r� r� r� r� Z orientationr8 g�������?r rk )r r r/ r� Zset_valr� r" ZpolyZ get_extentsZ transformedZ transAxes�invertedr Zbounds)r r r� Zboxr r r � test_slider_horizontal_verticalB s$ � � r� c s` t � � d� _� fdd�}t�� |�}| D ]\}}t||f|� q(� j|ksNt�� j|ks\t�dS )a� Helper function to test Polygon Selector Parameters ---------- event_sequence : list of tuples (etype, dict()) A sequence of events to perform. The sequence is a list of tuples where the first element of the tuple is an etype (e.g., 'onmove', 'press', etc.), and the second element of the tuple is a dictionary of the arguments for the event (e.g., xdata=5, key='shift', etc.). expected_result : list of vertices (xdata, ydata) The list of vertices that are expected to result from the event sequence. selections_count : int Wait for the tool to call its `onselect` function `selections_count` times, before comparing the result to the `expected_result` r c s � j d7 _ | � _d S )Nr )�_selections_count�_current_result)Zverticesr$ r r r% k s z(check_polygon_selector.<locals>.onselectN)r r� r/ ZPolygonSelectorr r" r� )�event_sequence�expected_resultZselections_countr% r r Z event_argsr r$ r �check_polygon_selectorV s r� c C s. dt | |d�fdt | |d�fdt | |d�fgS )Nr( r_ r& r) )r= r_ r r r �polygon_place_vertexx s �r� c C s� dddg} t dd�t dd� t dd� t dd� }t|| d� dddg} t dd�t dd� dtd d �fd tddd �fd tddd �fd tddd �fdtddd �fdtd d �fg t dd� t dd� }t|| d� dddg} t dd�t dd� dtdd �fd tddd �fd tddd �fd tddd �fdtddd �fdtdd �fg t dd� t dd� }t|| d� dddg} t dd�t dd� t dd� t dd� d tddd �fd tddd �fd tddd �fdtddd �fg }t|| d� dddg} t dd�t dd� t dd� t dd� dtdd �fd tddd �fd tddd �fd tddd �fdtddd �fdtdd �fg }t|| d� dddg} dtd d �fd tddd �fd tddd �fd tddd �fdtddd �fdtd d �fdtdd �fd tddd �fd tddd �fd tddd �fdtddd �fdtdd �fg t dd� t dd� t dd� t dd� }t|| d� dddg} t dd�t dd� dtdd �fdtdd �fg t dd� t dd� t dd� t dd� }t|| d� d S ) N)�2 r� )rC r� )r� rC r� rC r )rL r� rH rI )r r( r_ r& rL r) rK )r� rL )rC rL rM r rJ rY )rL rL )� rL )rL r� r* �escape)r� r� r= )r� r� r r r �test_polygon_selector~ s� ��� � ��� � � ��� � ����� ��� �� � � ��� � ������r� )r r r Nr ) Z unittest.mockr Zmatplotlib.widgetsr/ Zmatplotlib.pyplotZpyplotr Z matplotlib.testing.decoratorsr Z numpy.testingr r� r r r5 r> r] rc ri rl rm rn ru r� r� r� r� r� r� r� r� r� r r r r �<module> sD 6 7$ � � "