OwlCyberSecurity - MANAGER
Edit File: test_agg.cpython-38.pyc
U ��]i � @ s� d dl Z d dlZd dlmZ d dlZd dlmZmZm Z m Z m Z d dlmZ d dlmZ d dlmZ d dlmZ dd � Zd d � Zd d � Zdd� Zedgdgdd�dd� �Zdd� Z dd� Z ej � d�dd � �Z d d � Z!d d � Z"dS )!� N)�assert_array_almost_equal)� collections�path�pyplot� transforms�rcParams)�imread)�FigureCanvasAgg)�Figure)�image_comparisonc C s� t ddg�} t| �}| �d� | j�d� t�� }| j|| �� dd� |� d� | j|| �� dd� |� d� t t t |�d �d d d � d S ) N� 皙�����?)r r r � �?Znone)Z facecolorZ edgecolorr �r r )� � �?r r � )�decimal) r � FigureCanvasZ set_facecolorZpatchZ set_alpha�io�BytesIO�savefigZ get_facecolor�seekr �tupler )�fig�canvas�buf� r �;/usr/lib/python3/dist-packages/matplotlib/tests/test_agg.py� test_repeated_save_with_alpha s& � � �r c C sh t �� } t�� \}}t�t�ddgddgddgddgddgg�g�}|�|� |� dd� t� | � d S )Ni����� � ���g����MbP?r ) r r �plt�subplotsr ZPathCollectionr �PathZ add_artist�set_xlimr )�buff�f�axZ collectionr r r �!test_large_single_path_collection/ s (� r* c C sV t �d�\} }d}t�|�}|�|� |�||d |d � t�� }| j|dd� d S )Nr �� g�������?g333333�?�png��format) r# r$ �npZarangeZsemilogxZ fill_betweenr r r )r r) Zsteps�datar r r r �test_marker_with_nan= s r1 c C sH t �� } t�� \}}tj�d� tj�d�}|�|� |j | dd� d S )Nr ip r, r- ) r r r# r$ r/ ZrandomZseedZrand�plotr )r' r r) �pointsr r r �test_long_pathI s r4 Z agg_filterr, T)Zbaseline_images� extensionsZ remove_textc s� dd� �d&�fdd� �G dd� dt �} G dd � d | ��G �fd d �d | �� G � �fd d �d | �}t�� }|�d�}|jdddgdddgdddddddd� \}|jdddgdddgdd dddddd� \}|d �}||fD ]v}|�� }|�� } |�|| �\} | �|� tj |� � |jd d d d!�} | � | � | � |� � d � | �|� | �d"� q�|�d#d$� |�d#d$� |j�d%� |j�d%� d S )'Nc S sv t jd| d | |dd� | d| d | d| d� f }t �|�}t j||�� |dd�}||d | d � S )N� r r ���Zsame)�mode)r/ Zr_ZhanningZconvolve�sum)�x� window_len�s�w�yr r r �smooth1dV s � z!test_agg_filter.<locals>.smooth1dr c sh t t|�d�d d � t��� fdd�t�| �D ��}t�|�}t��� fdd�|D ��}t�|�}|S )Nr r6 r c s g | ]}� |���qS r r ��.0r: �r? r; r r � <listcomp>` s z5test_agg_filter.<locals>.smooth2d.<locals>.<listcomp>c s g | ]}� |���qS r r r@ rB r r rC b s )�max�intr/ ZarrayZasarrayZ transpose)�A�sigmaZA1ZA2ZA3ZA4)r? )r; r �smooth2d^ s z!test_agg_filter.<locals>.smooth2dc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z#test_agg_filter.<locals>.BaseFilterc S sh |j \}}}tj|d | |d | |gdd�}|d d �d d �d d �f ||| �|| �d d �f< |S )Nr6 �d)Zdtype)�shaper/ Zzeros)�selfZ src_image�dpi�padZnyZnxZdepth� padded_srcr r r � prepare_imageh s $4z1test_agg_filter.<locals>.BaseFilter.prepare_imagec S s dS )Nr r �rK rL r r r �get_pado s z+test_agg_filter.<locals>.BaseFilter.get_padc S s2 | � |�}| �|||�}| �||�}|| | fS �N)rQ rO � process_image)rK �imrL rM rN � tgt_imager r r �__call__r s z,test_agg_filter.<locals>.BaseFilter.__call__N)�__name__� __module__� __qualname__rO rQ rV r r r r � BaseFilterg s rZ c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) z%test_agg_filter.<locals>.OffsetFilterNc S s |d krd| _ n|| _ d S )Nr )�offsets)rK r[ r r r �__init__y s z.test_agg_filter.<locals>.OffsetFilter.__init__c S s t t| j� d | �S )N� R@)rE rD r[ rP r r r rQ s z-test_agg_filter.<locals>.OffsetFilter.get_padc S sH | j \}}tj|t|d | �dd�}tj|t|d | � dd�}|S )Nr] r )Zaxisr )r[ r/ ZrollrE )rK rN rL ZoxZoyZa1Za2r r r rS � s z3test_agg_filter.<locals>.OffsetFilter.process_image)N�rW rX rY r\ rQ rS r r r r � OffsetFilterx s r_ c s. e Zd ZdZd dd�Zdd� Z� fdd �ZdS ) z'test_agg_filter.<locals>.GaussianFilterzsimple gauss filter� �?Nc S s& || _ || _|d kr d| _n|| _d S )N)r r r )rG �alpha�color)rK rG ra rb r r r r\ � s z0test_agg_filter.<locals>.GaussianFilter.__init__c S s t | jd d | �S )Nr r] )rE rG rP r r r rQ � s z/test_agg_filter.<locals>.GaussianFilter.get_padc sl t �|�}� |d d �d d �df | j | jd | �}||d d �d d �df< | j|d d �d d �d d�f< |S )Nr7 r] )r/ Z zeros_likera rG rb )rK rN rL rU Zaa�rH r r rS � s � z5test_agg_filter.<locals>.GaussianFilter.process_image)r` N)rW rX rY �__doc__r\ rQ rS r rc r r �GaussianFilter� s re c s, e Zd Zd � �fdd� Zdd� Zdd� ZdS ) z)test_agg_filter.<locals>.DropShadowFilter�333333�?Nc s � |||�| _ �|�| _d S rR )� gauss_filter� offset_filter)rK rG ra rb r[ �re r_ r r r\ � s z2test_agg_filter.<locals>.DropShadowFilter.__init__c S s t | j�|�| j�|��S rR )rD rg rQ rh rP r r r rQ � s �z1test_agg_filter.<locals>.DropShadowFilter.get_padc S s | j �||�}| j�||�}|S rR )rg rS rh )rK rN rL �t1�t2r r r rS � s z7test_agg_filter.<locals>.DropShadowFilter.process_image)rf NNr^ r ri r r �DropShadowFilter� s rl �o g�������?r` g�������?zbo-�br= r r! zLine 1)ZmecZmfcZlwZmewZmsZlabelg�������?gffffff�?zro-�r� g @g �r3 )r: r> ZunitsTr r F)r )�objectr# �figureZ add_subplotr2 Z get_xdataZ get_ydataZ update_from� mtransformsZ offset_copyZ get_transformZ set_transformZ set_zorderZ get_zorderZset_agg_filterZset_rasterizedr& Zset_ylimZxaxisZ set_visibleZyaxis) rZ rl r r) �l1�l2Zgauss�lZxxZyyZshadowZotr )re r_ r? rH r �test_agg_filterS sV � � � rw c C s8 t jdd�} t�� }t�t�� | �|� W 5 Q R X d S )N)i, r+ )Zfigsize)r# rr r r �pytestZraises� ValueErrorr )r r' r r r �test_too_large_image� s rz c C sd t d�} t�� \}}|�| t�| �� |j�� t�� \}}dtd< |�| t�| �� |j�� d S )N�� �i zagg.path.chunksize) �ranger# r$ r2 r/ Zsinr Zdrawr )r: r r) r r r �test_chunksize� s r~ ZAggc C sZ t �d�} t�dddgdddg� t�� }tj|ddd� | �|�}|jd d ksVt �d S ) N� PIL.Imager r r6 Zjpgr{ )r. rL rL )r{ r{ ) rx � importorskipr# r2 r r r �open�info�AssertionError)�Imager rT r r r � test_jpeg_dpi� s r� c C sh t �d�} ddlm} t�� }|� }|�dd� t�� j |dd|id� | � |�}|j d dksdt �d S ) Nr r )�PngInfoZSoftwareZtestr, �pnginfo�r. � pil_kwargs) rx r� ZPIL.PngImagePluginr� r r Zadd_textr# rr r r� r� r� )r� r� r r� rT r r r �test_pil_kwargs_png� s r� c sp t �d�} ddlm� t�� }ddi}t�� j|d|d� | � |�}� fdd �|j � � D �}|d dkslt �d S ) Nr r )�TAGS_V2Z descriptionz test imageZtiffr� c s i | ]\}}� | j |�qS r )�name)rA �k�v�ZTAGSr r � <dictcomp> s z(test_pil_kwargs_tiff.<locals>.<dictcomp>ZImageDescription) rx r� Z PIL.TiffTagsr� r r r# rr r r� Ztag_v2�itemsr� )r� r r� rT Ztagsr r� r �test_pil_kwargs_tiff� s r� )#r Znumpyr/ Z numpy.testingr rx Z matplotlibr r r r# r rs r Zmatplotlib.imager Z matplotlib.backends.backend_aggr r Zmatplotlib.figurer Z matplotlib.testing.decoratorsr r r* r1 r4 rw rz r~ ZmarkZbackendr� r� r� r r r r �<module> s. � ~