OwlCyberSecurity - MANAGER
Edit File: test_lines.cpython-38.pyc
U ��]6 � @ s8 d Z ddlZddlZddlmZ ddlZddlZddlZddlm Z ddl m Z ddlmZmZ ejjdd�dd� �Zd d � Zed gd d �dd� �Zdd� Zdd� Zdd� Zedgd dgd�dd� �Zdd � Zd d � Zed gd d d!�d"d#� �Z ed$gd dgd�d%d&� �Z ed'gd(d)�d*d+� �Z d,d-� Z e� d.d/� �Z dS )0z% Tests specific to the lines module. � N)�cycler)�image_comparison�check_figures_equal� )Zrerunsc C s� d} t �dd| �}t jj| d�}t�� }t�d�}t�||�}|� d� t t j |j jddd��}|�|� t t j |j jddd��}|| }d } || k s�t�d S ) ac Github issue #1256 identified a bug in Line.draw method Despite visibility attribute set to False, the draw method was not returning early enough and some pre-rendering code was executed though not necessary. Consequence was an excessive draw time for invisible Line instances holding a large number of points (Npts> 10**6) i��� r � )�size�o Fr )Znumber�repeat� N)�np�linspace�randomZnormal�plt�figureZsubplot�mlines�Line2DZ set_visible�min�timeitr �canvas�drawZadd_line�AssertionError) �N�x�y�fig�ax�lZ t_no_lineZt_invisible_lineZslowdown_factorZslowdown_threshold� r �=/usr/lib/python3/dist-packages/matplotlib/tests/test_lines.py� test_invisible_Line_rendering s r c C sH t �� } | �ddd�}tj�d� |jtj�dd�dgd�}dsDt�d S )Nr r � � �r )r r �Z linestylesT) r r � add_subplotr r �seed�contour�randnr �r r Zcsr r r �test_set_line_coll_dash@ s r) Z line_dashesT)�baseline_images� remove_textc C s. t �� } | �ddd�}|jtd�ddd� d S )Nr � r" � �� linestyle�lw)r r r$ �plot�range�r r r r r �test_line_dashesL s r4 c C s� t �� } | �ddd�}|jtd�dd� |jtd�dd� |jtd�dd� |jtd�dd� |jtd�dd� | j�� d s�t�d S ) Nr r, Znone)�color�rz.3)r r r r )r r r T�r r r$ r1 r2 r r r r3 r r r �test_line_colorsT s r8 c C sH t �� } | �ddd�}dD ]}|jtd�|d� q| j�� dsDt�d S )Nr )�-Zsolidz--�dashedz-.�dashdot�:�dottedr, )r/ Tr7 )r r �lsr r r �test_linestyle_variants` s r? c C s0 t �g g �} t�t�� | �d� W 5 Q R X d S )NZaardvark)r r �pytest�raises� ValueErrorZ set_linestyle��liner r r �test_valid_linestylesk s rE Zdrawstyle_variantsZpng)r* r+ � extensionsc C sX t �d�\} }dddddd g}t|j|�D ](\}}|jtd�|d� |jd d d � q*d S ) N� �defaultz steps-mid� steps-prez steps-postZstepsi� )Z drawstyle)r r )ZxlimZylim)r �subplots�zipZflatr1 r2 �set)r ZaxsZdssr Zdsr r r �test_drawstyle_variantsq s rM c C s0 t �g g �} t�t�� | �d� W 5 Q R X d S )NZfoobar)r r r@ rA rB � set_drawstylerC r r r �test_valid_drawstyles s rO c C s� t �ddt j d�} t �| �}t�� \}}|�| |�\}|�d� t|� � j �dt| � d ksdt �|�d� t|� � j �t| �ks�t �d S )Nr r r, rI r rH ) r r ZpiZsinr rJ r1 rN �lenZget_pathZverticesr )r r r r rD r r r �test_set_drawstyle� s " rQ Zline_collection_dashesZmpl20)r* r+ �stylec C s@ t �� } | �ddd�}tj�d� |jtj�dd�dgd�}d S )Nr r r r! r" r# )r r r$ r r r% r& r' r( r r r � test_set_line_coll_dash_image� s rS Zmarker_fill_stylesc C s� t �dddgdddddt�dddg�g�} d}t�ddg�}t�dd g�}t�� \}}ttjj �D ]^\}}ttjj �D ]H\}} t | �} |j |d | || d |d |d || | d| | d d� qvqb|� ddg� |�ddg� d S )Nr r �gz#ff0000�c�mr Z lightgreen� r, � �?r r r- ) �markerZ markersizeZmarkerfacecoloraltZ fillstyleZlabelZ linewidthr5 ZmarkeredgecolorZmarkeredgewidthg @���� )� itertools�cycler �arrayr rJ � enumerater r Zfilled_markersZ fillStyles�nextr1 Zset_ylimZset_xlim) ZcolorsZaltcolorr r r r �jrY �iZfsr5 r r r �test_marker_fill_styles� s. � � rc Z scaled_linesrH )r* rR c C s� t �dd�} t�� \}}dddg}ttjd �}tt||��D ]L\}\}}t �ddd�D ]0}|j | |t � d � d | f||d �|�� qXq>d S ) Nr � r: r= r; zaxes.prop_cyclerX r, �2 g�������?r. ) r r r rJ r � matplotlibZrcParamsr_ rK r1 Zones) Zthr r Z lins_stylesZcyra r> Zstyr0 r r r �test_lw_scaling� s rg c C sl t �g g �} | �t�dddg��s&t�| �t�dtjdg��sBt�| �ddgtjgd ddg �rht�d S )Nr r r r- �d r )r r Z _is_sortedr r^ r �nanrC r r r �test_nan_is_sorted� s rj c C s> | � � �ddgd� |� � jdddgdddgdddgd� d S )Nr r z-or )Z markevery)rJ �stepr1 )Zfig_testZfig_refr r r �test_step_markers� s rl )!�__doc__r\ r r Znumpyr r@ rf Zmatplotlib.lines�linesr Zmatplotlib.pyplotZpyplotr Z matplotlib.testing.decoratorsr r ZmarkZflakyr r) r4 r8 r? rE rM rO rQ rS rc rg rj rl r r r r �<module> sJ + � � �