OwlCyberSecurity - MANAGER
Edit File: test_artist.cpython-38.pyc
U ��]�$ � @ s> d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dl m Z d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z dd� Z dd� Z e dgd d �d d � �Z d d� Z!e dgd dd�dd� �Z"dd� Z#e dgd dgdd�dd� �Z$dd � Z%d d � Z&d d � Z'ej(�)d!d"d#d$d%d&d'd(g�d)d*� �Z*d+d,� Z+dS )-� N)�chain)�image_comparisonc C s< t �� } | �ddg� | �ddg� d}| j�|�}tj|ddddd�}| �|� |j | jksbt �tj|dddd dd �}|� � s�t �| �|� t |j t j�s�t �tj|d d t �� dd �}| �|� t |j t j�s�t �tj|ddddd�}|�� }|� � r�t �| �|� |�� |k�st �|� � �s&t �|j | jk�s8t �d S ) N� � �� r �yellow� �?)�width�height�fc�alpha�x �coral)r r r � transformr �d )r r r r )�plt�axes�set_xlim�set_ylim� transData�transform_point�mpatches�Ellipse� add_patchZ _transform�AssertionErrorZis_transform_set� isinstance� mtransforms�IdentityTransform� get_transform)�ax�xy_data�xy_pix�eZintermediate_transform� r$ �>/usr/lib/python3/dist-packages/matplotlib/tests/test_artist.py� test_patch_transform_of_none s< � � � r& c C s t �� } | �ddg� | �ddg� d}| j�|�}tj|ddd�}tj |gddd�}| � |� |� � |� � | jkszt �tj|ddd�}tj |gd dd�}|�d � | � |� t|� � tj�s�t �tj|d d d�}tj |gt�� dd �}| � |� t|jtj��st �d S ) Nr r r )r r r r )� facecolorr r r r )r r )r r r r r r r r � mcollectionsZPatchCollection�add_collectionZget_offset_transformr r Z set_transformr r r Z _transOffset)r r! r" r# �cr$ r$ r% �!test_collection_transform_of_none= s. � � r+ Zclip_path_clippingT)�baseline_images� remove_textc C s* t j�� �� } | jd9 _| jd8 _t j�� �� }|jd d d� |_t jt�| j|jg�t�| j|jg�d�}t j� d��� }| jd9 _t � d�}t j |gdd d d d d �}|�||j� |�|� t j d||d�}tj|dd d d d d �}|�||j� |�|� |�ddg� |�ddg� d S )N� r ���)�vertices�codes� g������@�y � ZblueZredgffffff�?�*)�lw� edgecolorr' r �hatch�z )ZsharexZsharey���r )�mpath�PathZunit_rectangle�deepcopyr0 Z unit_circle�npZ concatenater1 Zunit_regular_starr Zsubplotr( ZPathCollectionZ set_clip_pathr r) r � PathPatchr r r )ZexteriorZinteriorZ clip_pathZstar�ax1�col�ax2Zpatchr$ r$ r% � test_clippinga s> ��� � � rC c C s� t j�d�} t j�d�}t�� \}}|�| |d� |�dd� t�� }|j|dd� t |� � �dk sht �t�� }|j|dd� t |� � �dk s�t �d S ) Ni N zk.r r �pdf)�formati@ �svg) r> Zrandomr �subplots�plotr �io�BytesIOZsavefig�len�getvaluer )�x�y�figr rD rF r$ r$ r% �test_cull_markers� s rP Zhatching�default)r, r- �stylec C s� t �dd�\} }tjddddd�}|�|� tjddgdg|jdd �}|�|� tjd dddd d �}|�|� tjddgd g|jdd d�}|�|� |� dd� |� dd� d S )Nr �r r r r. �/)r8 i�> )g �?� @)�sizes�offsets� transOffsetr8 )r. r ZC1)r8 r7 )g @rU )rV rW rX r8 r7 r � � ) r rG r Z Rectangler r( ZRegularPolyCollectionr r) r r )rO r Zrect1Zrect2Zrect3Zrect4r$ r$ r% � test_hatching� s( � � r[ c C s� t �� \} }|�t�d��dd��}|�td��\}| js>t �|jsHt �| j � � | jr\t �|jrft �|jrpt �||j ks~t �||j ks�t �|j |ks�t �|�� |�� ||fD ] }|j d ks�t �|jd ks�t �q�||j ks�t �| js�t �|js�t �d S )N�$ r2 r4 )r rG Zimshowr> �arangeZreshaperH �rangeZstaler ZcanvasZdrawZ_mouseover_setr �removeZfigure)rO r Zim�lnZartr$ r$ r% � test_remove� s( ra Z default_edgesZpng)r, r- � extensionsrR c C s� t �dd�\} \\}}\}}|�t�d�t�d�dt�d�d t�d�d� |jt�d�t�d�dd� |jddd d td d �d � |�d� |� d� t j t � ddddgt j jt j jt j jt j jg�d|jd�}|�|� d S )Nr � rM r �oZedge)Zalignr ZBOX� Zsawtooth)Zboxstyle)�sizeZbbox)r/ r rS )r r )r r Znone)r r )r rG rH r> r] Zbar�text�dictr r r r? r; r< ZMOVETOZCURVE3Z CLOSEPOLYr r )rO r@ rB Zax3Zax4Zpp1r$ r$ r% �test_default_edges� s* � �� �ri c C sJ t �g g �} tjdd��(}t�d� | �� t|�dks<t�W 5 Q R X d S )NT)�record�alwaysr )�mlines�Line2D�warnings�catch_warnings� simplefilterZ propertiesrK r )r` �wr$ r$ r% �test_properties� s rr c C s� t �g � t �g g� t �� \} }|�td��}|�td��}t�t||�dd� t j|j�� dd� t � � }t j|d|d� |� � dks�t �d S ) Nr r6 r4 Zgreen)Zcolor�zorder)�filez zorder: float ) r ZsetprG rH r^ �martistr Zspines�valuesrI �StringIOrL r )rO r Zlines1Zlines2Zsior$ r$ r% � test_setp� s rx c C sp t �� \} }|jtd�d d�\}|�� tjjks4t�|� d� |�� dksNt�|� d � |�� tjjkslt�d S )Nr4 )rs i@� ) r rG rH r^ Z get_zorderrl rm rs r Z set_zorder)rO r r` r$ r$ r% �test_None_zorder� s ry zaccept_clause, expected)� �unknown)z ACCEPTS: [ '-' | '--' | '-.' ]z[ '-' | '--' | '-.' ])zACCEPTS: Some description.�Some description.)z .. ACCEPTS: Some description.r| )z arg : int�int)z *arg : intr} )z"arg : int ACCEPTS: Something else.zSomething else. c C s> G dd� dt j�}d| |j_t �|��d�}||ks:t�d S )Nc @ s e Zd Zdd� ZdS )z:test_artist_inspector_get_valid_values.<locals>.TestArtistc S s d S )Nr$ )�self�argr$ r$ r% �set_f s z@test_artist_inspector_get_valid_values.<locals>.TestArtist.set_fN)�__name__� __module__� __qualname__r� r$ r$ r$ r% � TestArtist s r� z Some text. %s �f)ru ZArtistr� �__doc__�ArtistInspectorZget_valid_valuesr )Z accept_clauseZexpectedr� Z valid_valuesr$ r$ r% �&test_artist_inspector_get_valid_values s �r� c C s* t �tj�} | �� }|d dhks&t�d S )NZ linewidthr6 )ru r� rl rm Z get_aliasesr )Zai�aliasesr$ r$ r% �!test_artist_inspector_get_aliases s r� ),rI � itertoolsr rn Znumpyr> ZpytestZmatplotlib.pyplotZpyplotr Zmatplotlib.patchesZpatchesr Zmatplotlib.lines�linesrl Zmatplotlib.path�pathr; Zmatplotlib.transformsZ transformsr Zmatplotlib.collections� collectionsr( Zmatplotlib.artistZartistru Z matplotlib.testing.decoratorsr r& r+ rC rP r[ ra ri rr rx ry ZmarkZ parametrizer� r� r$ r$ r$ r% �<module> sR *$ � � �