OwlCyberSecurity - MANAGER
Edit File: test_collections.cpython-38.pyc
U ��]L^ � @ s2 d Z ddlZddlZddlZddlmZmZ ddlZddl m Z ddl m Z ddlmZ ddl mZmZmZ ddlmZ dd� Zedgd �d d � �Zd d � Zdd� Zdd� Zdd� Z dd� Z dd� Z dd� Z edgd �dd � �Z ed gd �d d � �Z!ed gd �d!d"� �Z"ed#gd �d$d%� �Z#ed&gd �d'd(� �Z$ed)gd �d*d+� �Z%ed,gd �d-d.� �Z&ed/gd �d0d1� �Z'ed2gd �d3d4� �Z(ed5gd �d6d7� �Z)ed8gd9d:�d;d<� �Z*ed=gd �d>d?� �Z+ed@gd �dAdB� �Z,dCdD� Z-dEdF� Z.dGdH� Z/dIdJ� Z0dKdL� Z1dMdN� Z2edOgdPgdQdRi�3e�4� dS�d9dT�dUdV� �Z5edWgdPgd9dX�dYdZ� �Z6ed[gdPgd9dX�d\d]� �Z7ed^gdPgd9dX�d_d`� �Z8dadb� Z9dcdd� Z:edegd9dPgdf�dgdh� �Z;didj� Z<ej=�>dk�dldm� �Z?ej=�>dk�dndo� �Z@ej=�>dk�dpdq� �ZAedrgdPgds�dtdu� �ZBedvgdPgd9dkdw�dxdy� �ZCdzd{� ZDd|d}� ZEdS )~z+ Tests specific to the collections module. � N)�assert_array_equal�assert_array_almost_equal)� Collection�LineCollection�EventCollection)�image_comparisonc C s� t �ddddddddg�} t �d d d g�}d }d }d}d}d ddd g}d}d}t| |||||||d�} t�� } | �d d d �} | �| � | �d� | ||||||||d� } | �dd� | � dd� | | | fS )z5 generate the initial collection and plot it � � �?� @� @g @g @g *@g 5@g A@g �K@g @V@� horizontal� � �?� r �solidT)� orientation� lineoffset� linelength� linewidth�color� linestyle� antialiasedzEventCollection: default) � positions�extra_positionsr r r r r r r ���� ) �np�arrayr �plt�figure� add_subplot�add_collection� set_title�set_xlim�set_ylim) r r r r r r r r r �coll�fig�splt�props� r) �C/usr/lib/python3/dist-packages/matplotlib/tests/test_collections.py� generate_EventCollection_plot sF � � r+ Z EventCollection_plot__default)�baseline_imagesc C s0 t � \} }}t||d |d |d |d � dS )zN check to make sure the default segments have the correct coordinates r r r r N)r+ �check_segments��_r% r( r) r) r* �#test__EventCollection__get_segments= s �r0 c C s& t � \} }}tj�|d |�� � dS )zL check to make sure the default positions match the input positions r N)r+ r �testingr � get_positionsr. r) r) r* �$test__EventCollection__get_positionsJ s r3 c C s$ t � \} }}|d |�� ks t�dS )�V check to make sure the default orientation matches the input orientation r N)r+ �get_orientation�AssertionErrorr. r) r) r* �&test__EventCollection__get_orientationR s r7 c C s t � \} }} |�� st�dS )r4 N)r+ � is_horizontalr6 �r/ r% r) r) r* �$test__EventCollection__is_horizontal[ s r: c C s$ t � \} }}|d |�� ks t�dS )zP check to make sure the default linelength matches the input linelength r N)r+ �get_linelengthr6 r. r) r) r* �%test__EventCollection__get_linelengthd s r< c C s$ t � \} }}|d |�� ks t�dS )zP check to make sure the default lineoffset matches the input lineoffset r N)r+ �get_lineoffsetr6 r. r) r) r* �%test__EventCollection__get_lineoffsetl s r> c C s" t � \} }} |�� dgks t�dS )zN check to make sure the default linestyle matches the input linestyle �NNN)r+ � get_linestyler6 r9 r) r) r* �$test__EventCollection__get_linestylet s rA c C s8 t � \} }}tj�|d |�� � t|�� |d � dS )zF check to make sure the default color matches the input color r N)r+ r r1 r � get_color�check_allprop_array� get_colorsr. r) r) r* � test__EventCollection__get_color| s rE Z#EventCollection_plot__set_positionsc C st t � \} }}t�|d |d g�}|�|� tj�||�� � t|||d |d |d � | �d� | � dd� d S ) z9 check to make sure set_positions works properly r r r r r z EventCollection: set_positionsr �Z N) r+ r �hstackZ set_positionsr1 r r2 r- r" r# �r' r% r( � new_positionsr) r) r* �$test__EventCollection__set_positions� s � rJ Z#EventCollection_plot__add_positionsc C s� t � \} }}t�|d |d d g�}|�|d d � tj�||�� � t|||d |d |d � | �d� | � dd � d S ) z9 check to make sure add_positions works properly r r r r r r z EventCollection: add_positionsr �# N) r+ r rG Z add_positionsr1 r r2 r- r" r# rH r) r) r* �$test__EventCollection__add_positions� s �� rL Z&EventCollection_plot__append_positionsc C s� t � \} }}t�|d |d d g�}|�|d d � tj�||�� � t|||d |d |d � | �d� | � dd � d S ) z< check to make sure append_positions works properly r r r r r r z!EventCollection: append_positionsr rF N) r+ r rG Zappend_positionsr1 r r2 r- r" r# rH r) r) r* �'test__EventCollection__append_positions� s �� rM Z&EventCollection_plot__extend_positionsc C s� t � \} }}t�|d |d dd� g�}|�|d dd� � tj�||�� � t|||d |d |d � | �d� | � d d � dS ) z< check to make sure extend_positions works properly r r r Nr r r z!EventCollection: extend_positionsr rF ) r+ r rG Zextend_positionsr1 r r2 r- r" r# rH r) r) r* �'test__EventCollection__extend_positions� s �� rN Z(EventCollection_plot__switch_orientationc C sz t � \} }}d}|�� ||�� ks(t�|�� r4t�|�� }t|||d |d |� | �d� | �dd� | � dd� d S ) z> check to make sure switch_orientation works properly �verticalr r z#EventCollection: switch_orientationr r r r N) r+ �switch_orientationr5 r6 r8 r2 r- r" r$ r# )r' r% r( �new_orientationrI r) r) r* �)test__EventCollection__switch_orientation� s � rR Z,EventCollection_plot__switch_orientation__2xc C s� t � \} }}|�� |�� |�� }|d |�� ks8t�|�� sDt�tj�|d |� t |||d |d |d � | � d� dS )zj check to make sure calling switch_orientation twice sets the orientation back to the default r r r r z&EventCollection: switch_orientation 2xN) r+ rP r2 r5 r6 r8 r r1 r r- r" rH r) r) r* �,test__EventCollection__switch_orientation_2x� s �rS Z%EventCollection_plot__set_orientationc C sx t � \} }}d}|�|� ||�� ks*t�|�� r6t�t||d |d |d |� | �d� | �dd� | �dd � d S ) z; check to make sure set_orientation works properly rO r r r z EventCollection: set_orientationr r r r N) r+ Zset_orientationr5 r6 r8 r- r" r$ r# )r' r% r( rQ r) r) r* �&test__EventCollection__set_orientation� s � rT Z$EventCollection_plot__set_linelengthc C s` t � \} }}d}|�|� ||�� ks*t�t||d ||d |d � | �d� | �dd� dS ) z: check to make sure set_linelength works properly � r r r z EventCollection: set_linelengthi��� N)r+ Zset_linelengthr; r6 r- r" r$ )r' r% r( Znew_linelengthr) r) r* �%test__EventCollection__set_linelength s � rW Z$EventCollection_plot__set_lineoffsetc C s` t � \} }}d}|�|� ||�� ks*t�t||d |d ||d � | �d� | �dd� dS ) z: check to make sure set_lineoffset works properly g �r r r z EventCollection: set_lineoffseti�������N)r+ Zset_lineoffsetr= r6 r- r" r$ )r' r% r( Znew_lineoffsetr) r) r* �%test__EventCollection__set_lineoffset s � rY Z#EventCollection_plot__set_linestylec C s: t � \} }}d}|�|� |�� dgks,t�| �d� dS )�9 check to make sure set_linestyle works properly �dashed�r )� @r] � EventCollection: set_linestyleN�r+ Z set_linestyler@ r6 r" �r' r% r/ Z new_linestyler) r) r* �$test__EventCollection__set_linestyle1 s ra Z!EventCollection_plot__set_ls_dashT)r, � remove_textc C s: t � \} }}d}|�|� |�� dgks,t�| �d� dS )zH check to make sure set_linestyle accepts a single dash pattern r\ r^ Nr_ r` r) r) r* �0test__EventCollection__set_linestyle_single_dash= s rc Z#EventCollection_plot__set_linewidthc C s8 t � \} }}d}|�|� |�� |ks*t�| �d� dS )rZ � z EventCollection: set_linewidthN)r+ Z set_linewidthZ get_linewidthr6 r" )r' r% r/ Z new_linewidthr) r) r* �$test__EventCollection__set_linewidthJ s re Z EventCollection_plot__set_colorc C sV t � \} }}t�ddddg�}|�|� tj�||�� � t|�� |� | � d� dS )z5 check to make sure set_color works properly r r zEventCollection: set_colorN) r+ r r � set_colorr1 r rB rC rD r" )r' r% r/ Z new_colorr) r) r* � test__EventCollection__set_colorV s rg c C s� | � � }|�� dks(|�� dks(|dkr2d}d}n |�� dkrHd}d}ntd��t|�D ]p\}} | d|f ||d ks|t�| d|f ||d ks�t�| d|f || ks�t�| d|f || ksXt�qXdS ) z� check to make sure all values in the segment are correct, given a particular set of inputs note: this is not a test, it is used by tests r �noneNr r rO z.orientation must be 'horizontal' or 'vertical'r )Z get_segments�lower� ValueError� enumerater6 ) r% r r r r ZsegmentsZpos1Zpos2�iZsegmentr) r) r* r- c s" �� r- c C s | D ]}t j�||� qdS )z{ check to make sure all values match the given target if arrays note: this is not a test, it is used by tests N)r r1 r )�values�target�valuer) r) r* rC s rC c C s4 t �g �} | �t�� �}t|�� tj�� �� � d S )N) � mcollections�PathCollection� get_datalim� mtransformsZIdentityTransformr Z get_pointsZBboxZnull)�colZ col_data_limr) r) r* � test_null_collection_datalim� s �ru c C sV t �� t �� } | �ddgddg�}| �|� | jj}| �g g �}| jj|ksRt�d S )Nr r )r r �axes�scatterr! �dataLim�boundsr6 )�axr% ry r) r) r* �test_add_collection� s r{ c C s� t �� } t�d�t�d� }}t�ddd��dd� }}t �||||�}|�| j�j dks^t �t � � t �� } t�ddd�}t�dd d�}t� ||�\}}t �� �d d �| j }t j||t�|�t�|�|d � | jj d ks�t �d S )N� � r �P )r r g @g "@���rV ���� � � �Z transform)g 4@g >@g .@r] )r rv r �arange�linspaceZreshapeZquiverrr � transDatary r6 r �meshgridrs �Affine2D� translate�sin�cosrx )rz �x�y�u�v�q�transr) r) r* �test_quiver_limits� s r� c C s� t �� } t�ddd�}t�ddd�}t�||�\}}t�� �dd�| j }t j ||t� |�t� |�|d� t | j jd d d � d S ) Nr r} rV r� r� r� r� r� )rV � rU � r )�decimal)r rv r r� r� rs r� r� r� Zbarbsr� r� r rx ry )rz r� r� r� r) r) r* �test_barb_limits� s �r� Z EllipseCollection_test_imageZpngZaarch64g{�G�z�?r )r, � extensionsZtolrb c C s� t �� \} }t�d�}t�d�}t�||�\}}t�|�� |�� f�j}||d }||d }t�|�d } t j ||| d||j dd�} |� | � |� � d S )Nr� � r rV r� rh )�units�offsets� transOffset� facecolors)r �subplotsr r� r� ZvstackZravel�TZ ones_likerp �EllipseCollectionr� r! �autoscale_view) r& rz r� r� �X�YZXYZwwZhh�aa�ecr) r) r* �test_EllipseCollection� s � r� Zpolycollection_close)r, r� rb c C sH ddl m} ddgddgddgddggddgddgddgddggddgddgddgddggddgddgddgddggg}t�� }| |�}dd d d d g}ttd ��}tj|t|� dd�}|� d� g }g }t ||�D ]0\} } |� | gt|� � |� | gt|� � q�|� |� |j ||d d� |�dd� |�dd� |�dd� d S )Nr )�Axes3Dr r r r � @�r�g�br� �krd g �?�r gffffff�?)�zsZzdirr� r� )Zmpl_toolkits.mplot3dr� r r �list�rangerp ZPolyCollection�len� set_alpha�zip�extendrf Zadd_collection3dZ set_xlim3dZ set_zlim3dZ set_ylim3d) r� Z vertsQuadr& rz �colorsZzposZpolyr� Zcs�z�cr) r) r* �test_polycollection_close� s2 � � r� Z regularpolycollection_rotatec C s� t jd d�d d�f \} }t �| �� |�� g�}t �ddt j t|��}t�� \}}t ||�D ].\}}t j dd||g|j d�}|j |dd� q^|�� d S ) Nr} r r r� )�d )�sizes�rotationr� r� T�Zautolim)r ZmgridZ transposeZflattenr� �pir� r r� r� rp �RegularPolyCollectionr� r! r� ) ZxxZyyZ xy_pointsZ rotationsr& rz �xy�alphart r) r) r* �!test_regularpolycollection_rotate� s �r� Zregularpolycollection_scalec C sd G dd� dt j�} t�� \}}dg}tjd g}| |||jd�}|j|dd� |�dd dd g� d S ) Nc s$ e Zd Z� fdd�Zdd� Z� ZS )z:test_regularpolycollection_scale.<locals>.SquareCollectionc s t � jddtjd i|�� d S )Nr� r� r� )r� )�super�__init__r r� )�self�kwargs�� __class__r) r* r� s zCtest_regularpolycollection_scale.<locals>.SquareCollection.__init__c S sJ | j }d|jj }||jj |jj }||jj |jj }t�� � ||�S )z4Return transform scaling circle areas to data space.g R@) rv r �dpiZbbox�widthZviewLimZheightrs r� Zscale)r� rz Z pts2pixelsZscale_xZscale_yr) r) r* � get_transform s zHtest_regularpolycollection_scale.<locals>.SquareCollection.get_transform)�__name__� __module__� __qualname__r� r� � __classcell__r) r) r� r* �SquareCollection s r� )r r r )r� r� r� Tr� r r ) rp r� r r� r r� r� r! Zaxis)r� r& rz r� Z circle_areasZsquaresr) r) r* � test_regularpolycollection_scale s �r� c C s� t �� \} }|jdgdgdgdd�}| jt�� | jd� G dd� dt�}|� }d|_d |_ |� |�\}}|spt �t |d dg� d S ) Nr i� T)Zpicker)r� c @ s e Zd ZdS )z test_picking.<locals>.MouseEventN)r� r� r� r) r) r) r* � MouseEvent. s r� iE �� Zind) r r� rw Zsavefig�io�BytesIOr� �objectr� r� �containsr6 r )r& rz rt r� Zevent�found�indicesr) r) r* � test_picking) s r� c C s0 t jdddgdddgdddgfd� t �� d S )Nr r r r r )r )r rw �drawr) r) r) r* � test_linestyle_single_dashes9 s $r� Z size_in_xy)r, rb r� c C s` t �� \} }d\}}}d}ddg}tj|||d||jd�}|�|� |�dd� |�dd� d S )N)�r} r} r} r r� )rU rU r� )r� r� r� r r� )r r� rp r� r� r! r# r$ )r& rz ZwidthsZheightsZanglesZcoords�er) r) r* �test_size_in_xy> s � r� c C s� dddg}| j dddg|d� }}| j dd d g|d�}| j d d d g|d�}| j dddg|d�}t|d� t|d� t|d� t|d� t|d� d S )N� � � �redZblueZgreen)�indexr r r� r r[ ZdashdotTF)Z edgecolors)r� )� linewidths)Z linestyles)Z antialiaseds)ZSeriesr )Zpdr� r� ZfcZlwZlsr� r) r) r* �test_pandas_indexingR s r� �defaultc C s� t �g �} | �ddg� | �dddg� | �� dgd ks>t�| �� dddgd ksXt�| �dddg� | �� dgd ks~t�| �� dddgk�� s�t�d S )N�-r r r� r? r� )rp rq Zset_linestylesZset_linewidthsZget_linestylesr6 Zget_linewidths�all�rt r) r) r* �test_lslw_bcastc s r� c C s< t jg dd�} | �� dks t�| �d� | �� dks8t�d S )N�round)ZcapstyleZbutt)rp rq Z get_capstyler6 � set_capstyler� r) r) r* � test_capstyleq s r� c C s< t jg dd�} | �� dks t�| �d� | �� dks8t�d S )Nr� )Z joinstyle�miter)rp rq Z get_joinstyler6 � set_joinstyler� r) r) r* �test_joinstyley s r� Zcap_and_joinstyle)r, r� c C s� t �� } | �ddd�}|�ddg� |�ddg� t�dddg�}t�dgdgdgg�t�dddgg� }t�d�}||d d �d d �d f< ||d d �d d �df< t|d d d gd �}|� d� |� d� |� |� |� d� d S )Nr g �g �?� @r r r )r� r� r r r} rU rV r� r� r� z2Line collection with customized caps and joinstyle) r r r r# r$ r r Zzerosr r� r� r! r" )r& rz r� ZysZsegsZ line_segmentsr) r) r* � test_cap_and_joinstyle_image� s ( r� Zscatter_post_alpha)r, r� rb �stylec C s@ t �� \} }|jtd�td�td�d�}| j�� |�d� d S )Nrd )r� g�������?)r r� rw r� �canvasr� r� )r& rz �scr) r) r* �test_scatter_post_alpha� s r� c C s� t j�d� t j�dd�\} }t j�d�}t jjdddd�}t jjdddd�}t�� \}}|j| |||dd dd �}|jd d �\}}t |�dks�t �t t � |�� t�t �d�� t � d d� |D ��} |�t �d�d �} t | | � |j||dd�} |jdd�\} } t | �dk�st �|j| | dd�}|jdddd�\}}t � dd� |D ��}t |d� t � dd� |D ��}t |d� |j||dd�}|jddddd � d �\}}d d� |D �}t �t � |�� t�d �}t||� |j||d d�}dd lm} |jdd ddd!d dd"d dgd#�}|j|d�\}}t | �t |�k�s0t �d$dd%d&g}|j|dd d'�\}}t t � |�� t�|dd � � | |||fD ]}|�|� �qz|j�� d S )(Ni!N,r r} r rd )�sizei, Zjet�o)r� �s�cmapZmarkerr� z{x:g})�fmtc S s g | ] }|� � �qS r) )rB ��.0�liner) r) r* � <listcomp>� s z7test_pathcollection_legend_elements.<locals>.<listcomp>r� r )�loc� )�numr� r r� )�propr� r c S s g | ] }|� � �qS r) )Z get_alphar� r) r) r* r� � s c S s g | ] }|� � �qS r) )Zget_markerfacecolorr� r) r) r* r� � s z{x:.2f}c S s d| S )Nr r) )r� r) r) r* �<lambda>� � z5test_pathcollection_legend_elements.<locals>.<lambda>)r r r� �funcc S s g | ] }|� � �qS r) )Zget_markersizer� r) r) r* r� � s r� r| r� r� )ZnbinsZ min_n_ticksZstepsr g33333�K@i )r r r� )r ZrandomZseedZrandZrandintr r� rw Zlegend_elementsr� r6 r r Zastype�floatr� r� ZlegendZsqrtr Zmatplotlib.tickerZtickerZ MaxNLocatorZ add_artistr� r� ) r� r� r� r� r& rz r� �h�lr� Zcolors2�l1Zh2Zlab2�l2r� r �l3ZactsizesZ labeledsizes�l4Zmtickerr Zh5Zlab5ZlevelsZh6Zlab6r) r) r* �#test_pathcollection_legend_elements� sT � � r c C s: t �ddddg�} t| �}t j�| t �ddddg�� d S )Nr� r r r} )r r r r1 r )Zarrr% r) r) r* �test_EventCollection_nosort� s r )F�__doc__r� �platformZnumpyr Z numpy.testingr r ZpytestZmatplotlib.pyplotZpyplotr Zmatplotlib.collections� collectionsrp Zmatplotlib.transformsZ transformsrs r r r Z matplotlib.testing.decoratorsr r+ r0 r3 r7 r: r< r> rA rE rJ rL rM rN rR rS rT rW rY ra rc re rg r- rC ru r{ r� r� �get�machiner� r� r� r� r� r� r� r� Zmarkr� r� r� r� r� r� r r r) r) r) r* �<module> s� + � � � � $ � � � � � 4