OwlCyberSecurity - MANAGER
Edit File: test_simplification.cpython-38.pyc
U ��]�* � @ s� d dl Z d dlZd dlZd dlmZmZ d dlZd dlm Z d dl m Z d dl mZmZ d dlmZ e dgdd�d d � �Ze d gdd�d d � �Ze dgdd�dd� �Zdd� Zdd� Zej�dd ejd ejd ejd g�ej�dd dg�dd � ��Zd d � Ze d gdd�d d!� �Z e d"gdd�d#d$� �Z e d%gdd�d&d'� �Z d(d)� Z d*d+� Z e d,gdd�d-d.� �Z!e d/gdd�d0d1� �Z"e d2gd3�d4d5� �Z#d6d7� Z$dS )8� N)�assert_array_almost_equal�assert_array_equal)�image_comparison)�patches� transforms)�PathZclippingT)�baseline_imagesZ remove_textc C sL t �ddd�} t �dt j | �}t�� \}}|j| |dd� |�d� d S )N� � @�{�G�z�?� � �?)� linewidth)g������ɿg�Q� ��ѿ)�np�arange�sin�pi�plt�subplots�plot�set_ylim)�t�s�fig�ax� r �F/usr/lib/python3/dist-packages/matplotlib/tests/test_simplification.py� test_clipping s r Zoverflowc C sH t �ddddg�} t �t| ��}t�� \}}|�| |� |�dd� d S )Nr r � @g jAr � )r �arrayr �lenr r r �set_xlim��x�yr r r r r � test_overflow s r& Zclipping_diamondc C s\ t �dddddg�} t �dddddg�}t�� \}}|�| |� |�dd� |�dd� d S )Nr r � �g333333�g333333�?)r r r r r r" r r# r r r � test_diamond' s r( c C s| t j�d� t jjdd�d } t�� \}}|j| ddd�}|d �� }|d �� }|� |�}|j dd �}|j j d ksxt �d S ) Nr �P� ��size�2 �roundr �Zsolid_joinstyler T�Zsimplifyi�c )r �random�seed�uniformr r r �get_path� get_transform�transform_path�cleaned�verticesr+ �AssertionError�r$ r r �p1�path� transform� simplifiedr r r � test_noise2 s r> c C s� dd� } ddddddg}ddddddg}| ||�}t ddgdd gdd gd dgg|jd d �d d �f � ddddddg}ddddd dg}| ||�}t ddgdd gdd gd dgg|jd d �d d �f � ddddddg}ddddddg}| ||�}t ddgdd gdd gddgd dgg|jd d �d d �f � ddddddg}ddddddg}| ||�}t ddgddgd dgg|jd d �d d �f � ddddddg}ddddddg}| ||�}t ddgdd gdd gd dgg|jd d �d d �f � d S )Nc S sX t �� \}}|�| |�}|d �� }|d �� }|�|�}|jdd�}|�� �|�}|S )Nr Tr/ )r r r r3 r4 r5 r6 �inverted)r$ r% r r r: r; r< r= r r r �_get_simplifiedB s z9test_antiparallel_simplification.<locals>._get_simplifiedr � � �?���r r r' r r ���g �� r )r r7 )r@ r$ r% r= r r r � test_antiparallel_simplificationA sf �� �� �� �� ��rF �angle� rE r �offsetrB c C sJ d}t j�d� t j�d�| | }d|d< d|d< t �| �| }t �| �| }|dd � �� }|dd � �� }|dd � �� }|dd � �� } |dkr�tddg||g|| g|d |d gddggdddddgd�} n4tddg||g|d |d gddggddddgd�} tt � ||g�j �} | j d d �} t | j | j � t| j| j� d S ) N� i!N,� r rA rC r )�codesTr/ )r r0 r1 �randr Zcos�max�minr Zvstack�Tr6 r r7 r rL ) rG rI ZscaleZ vert_offsetsr$ r% Zx_maxZx_minZy_maxZy_minZ p_expected�pZp2r r r �test_angled_antiparallel� sB � �� � �rR c C s� t j�d� t �t �dt jd d��t jjdd�d } t�� \}}|j | ddd�}|d � � }|d � � }|� |�}|j dd �}|jjd ks�t�d S ) Nr r r) r* r r- r. Tr/ i�b )r r0 r1 r �linspacer r2 r r r r3 r4 r5 r6 r7 r+ r8 r9 r r r �test_sine_plus_noise� s � rT Zsimplify_curvec C sx t jtdddtjdfddddgtjtjtjtjtjtjtjtjg�dd�} t� � \}}|� | � |� d � |� d � d S ) N�r r )rA r )rA rA rA )r r )r r Znone)Zfc)r r ) r Z PathPatchr r �nan�MOVETOZCURVE3Z CLOSEPOLYr r � add_patchr" r )Zpp1r r r r r �test_simplify_curve� s � ��� rY Zhatch_simplifyc C s> t �� \} }|�t jdddddd�� |�d� |�d� d S )NrU rA F�/)ZfillZhatch)g�������?g�������?)r r rX Z Rectangler" r )r r r r r � test_hatch� s r[ Z fft_peaksc C s� t �� \} }t�d�}|�ttj�t�dtj d | �t� t |�� ���}|d � � }|d � � }|� |�}|jdd�}|jjdks�t�d S )Ni r r r Tr/ �$ )r r r r r �absZfftr r Zblackmanr! r3 r4 r5 r6 r7 r+ r8 )r r r r: r; r< r= r r r �test_fft_peaks� s 8 r^ c C sz d} t jt�| �dd�}|�t|�d df�}t|�}|jt� � dd�}t |�}t|�dks`t �|d d tj ksvt �d S ) Ns� ZwAAAAku+v9UAQAA+Tj6/z8CAADpQ/r/KAMAANlO+v8QBAAAyVn6//UEAAC6ZPr/2gUAAKpv+v+8 BgAAm3r6/50HAACLhfr/ewgAAHyQ+v9ZCQAAbZv6/zQKAABepvr/DgsAAE+x+v/lCwAAQLz6/7wM AAAxx/r/kA0AACPS+v9jDgAAFN36/zQPAAAF6Pr/AxAAAPfy+v/QEAAA6f36/5wRAADbCPv/ZhIA AMwT+/8uEwAAvh77//UTAACwKfv/uRQAAKM0+/98FQAAlT/7/z0WAACHSvv//RYAAHlV+/+7FwAA bGD7/3cYAABea/v/MRkAAFF2+//pGQAARIH7/6AaAAA3jPv/VRsAACmX+/8JHAAAHKL7/7ocAAAP rfv/ah0AAAO4+/8YHgAA9sL7/8QeAADpzfv/bx8AANzY+/8YIAAA0OP7/78gAADD7vv/ZCEAALf5 +/8IIgAAqwT8/6kiAACeD/z/SiMAAJIa/P/oIwAAhiX8/4QkAAB6MPz/HyUAAG47/P+4JQAAYkb8 /1AmAABWUfz/5SYAAEpc/P95JwAAPmf8/wsoAAAzcvz/nCgAACd9/P8qKQAAHIj8/7cpAAAQk/z/ QyoAAAWe/P/MKgAA+aj8/1QrAADus/z/2isAAOO+/P9eLAAA2Mn8/+AsAADM1Pz/YS0AAMHf/P/g LQAAtur8/10uAACr9fz/2C4AAKEA/f9SLwAAlgv9/8ovAACLFv3/QDAAAIAh/f+1MAAAdSz9/ycx AABrN/3/mDEAAGBC/f8IMgAAVk39/3UyAABLWP3/4TIAAEFj/f9LMwAANm79/7MzAAAsef3/GjQA ACKE/f9+NAAAF4/9/+E0AAANmv3/QzUAAAOl/f+iNQAA+a/9/wA2AADvuv3/XDYAAOXF/f+2NgAA 29D9/w83AADR2/3/ZjcAAMfm/f+7NwAAvfH9/w44AACz/P3/XzgAAKkH/v+vOAAAnxL+//04AACW Hf7/SjkAAIwo/v+UOQAAgjP+/905AAB5Pv7/JDoAAG9J/v9pOgAAZVT+/606AABcX/7/7zoAAFJq /v8vOwAASXX+/207AAA/gP7/qjsAADaL/v/lOwAALZb+/x48AAAjof7/VTwAABqs/v+LPAAAELf+ /788AAAHwv7/8TwAAP7M/v8hPQAA9df+/1A9AADr4v7/fT0AAOLt/v+oPQAA2fj+/9E9AADQA/// +T0AAMYO//8fPgAAvRn//0M+AAC0JP//ZT4AAKsv//+GPgAAojr//6U+AACZRf//wj4AAJBQ///d PgAAh1v///c+AAB+Zv//Dz8AAHRx//8lPwAAa3z//zk/AABih///TD8AAFmS//9dPwAAUJ3//2w/ AABHqP//ej8AAD6z//+FPwAANb7//48/AAAsyf//lz8AACPU//+ePwAAGt///6M/AAAR6v//pj8A AAj1//+nPwAA/////w==z<i4)Zdtyper )r r � Y@r_ �ZcliprA r ) r Z frombuffer�base64Z decodebytesZreshaper! r � iter_segmentsr ZIdentityTransform�listr8 rW )�dataZvertsr; Zsegsr r r �test_start_with_moveto� s �re c C sj dt jd< t�d�} tj�d�}tj|d< t �� \}}|�| |� t � t �� |� t �� � W 5 Q R X d S )NFz path.simplifyi@ i� )r ZrcParamsr r r0 rM rV r r �pytestZraises� OverflowErrorZsavefig�io�BytesIO)ZxxZyyr r r r r �(test_throw_rendering_complexity_exceeded s rj Z clipper_edgec C s� d} t jdd�}|jddddd� |jdddd �}|�| � |j�t �d �� |j�t �d �� |j� d � |j� d � |� d d� d S )N) r rA r r r rE r rH r rJ )r rA )Zfigsizer )�left�bottomZwspaceZhspace)r r r r )r rJ F)ZylimZ autoscale_onrA rl rk rJ � ) r ZfigureZsubplots_adjustZadd_axesr ZxaxisZset_major_locatorZMultipleLocatorZyaxisZset_ticks_positionr" )Zdatr r r r r � test_clipper! s rn Zpara_equal_perpc C s� t �ddddddddgdgd �} t �ddddddddgdgd �}t�� \}}|�| d |d � |�| d |d d� d S )Nr rA r rC � Zro)r r r r r r# r r r �test_para_equal_perp1 s $$ rp Zclipping_with_nans)r c C sP t �ddd�} t �| �}t j| d d d�< t�� \}}|�| |� |�dd� d S )Nr g ��Q� @i� �d g пg �?)r rS r rV r r r r r# r r r �test_clipping_with_nans; s rr c C s� t ddggd �} t| jddddgd��}|g ks6t�t ddgdd ggd d g�} t| jddddgd��}d d � |D �ddgd fdd gd fgks�t�t ddggd g�} t| jddddgd��}dd � |D �ddgd fgks�t�d S )Ngꌠ9Y>)FrJ r rq r` r, �( �K �A rA r c S s g | ]\}}t |�|f�qS r �rc ��.0r$ r% r r r � <listcomp>M s z&test_clipping_full.<locals>.<listcomp>c S s g | ]\}}t |�|f�qS r rv rw r r r ry R s )r rc rb r8 )rQ r= r r r �test_clipping_fullF s � �rz )%ra rh Znumpyr Z numpy.testingr r rf Z matplotlib.testing.decoratorsr Zmatplotlib.pyplotZpyplotr Z matplotlibr r Zmatplotlib.pathr r r& r( r> rF ZmarkZ parametrizer rR rT rY r[ r^ re rj rn rp rr rz r r r r �<module> sF O& , "