OwlCyberSecurity - MANAGER
Edit File: test_scale.cpython-38.pyc
U ��] � @ s d dl mZ d dlmZmZmZ d dlmZm Z d dl Z d dl m Z d dlZd dlZd dlZe� dd� �Zdd� Ze d gd d gd �d d� �Zdd� Zdd� Ze dgd d gd �dd� �Zdd� Zdd� Zdd� Ze d gd d gd d d �d d!� �Zd"d#� Ze d$gd d gd d%�d&d'� �Z dS )(� N)�Log10Transform�InvertedLog10Transform�SymmetricalLogTransform)�check_figures_equal�image_comparison)�assert_allclosec C sz | j dddd�}|�d� |�d� |�� }|�� }|j dddd�}|j||d� |�ddg|d� |�|ddgd� d S )N�z �logZsymlog)�yscaleZxscaleg�����8@)�xlim�ylim�b)Z add_subplotZaxvlineZaxhline�get_xlim�get_ylim�set�plot)Zfig_testZfig_refZax_testr r Zax_ref� r �=/usr/lib/python3/dist-packages/matplotlib/tests/test_scale.py�test_log_scales s r c C s t ddd�} | �� }t�ddd�}|�| �|��}t||� t|�t|�ksPt�tj|d< |�| �|��}t||� t|�t|�ks�t�tj � |�}|�| �|��}t||� t|�t|�ks�t�tj j |d< |�| �|��}t||� t|�t|�ks�t�d S ) N� � � g ��� � �?� � ) r �inverted�np�arangeZtransform_non_affiner �type�AssertionError�nanZma�arrayZmasked)ZsltZslti�x�outr r r �test_symlog_mask_nan s$ r% Z logit_scalesTZpng)�baseline_images� remove_text� extensionsc C s� t �� \} }t�ddddddddd d d d d ddddg�}d| }|�||� |�d� |�d� |�| j� � �}t� |j �s�t �t� |j �s�t �d S )Ng����MbP?g�~j�t�h?�{�G�z�?g� ��Q��?g�������?皙�����?g333333�?g�������?r g333333�?gffffff�?g�������?g�������?g ףp= �?g�G�z��?g��C�l��?g+����?g �?ZlogitT)�plt�subplotsr r" r � set_xscaleZgridZ get_tightbbox�canvasZ get_rendererZisfiniteZx0r Zy0)�fig�axr# �yZbboxr r r �test_logit_scales5 s$ � r2 c C sx t �d�\} }t�d�}t�d�d }|�||� t�� }| j|dd� t�� }| j|dd� t�� }| j|dd� dS )z Issue #1799r r Zpdf)�formatZepsZsvgN)r+ r, r r �scatter�io�BytesIOZsavefig)r/ r0 r# r1 Zbufr r r �test_log_scatterG s r7 c C s4 t �� \} }|jdt�dddg�d� | j�� d S )Nr r r r )Zsubsy)r+ r, � set_yscaler r" r. �draw�r/ r0 r r r �test_logscale_subsZ s r; Z logscale_maskc C sJ t �ddd�} t�� \}}|�t �| d �� |j�� |jdd� d S )Nr �2 i� r r )r ) r Zlinspacer+ r, r �expr. r9 r )�xsr/ r0 r r r �test_logscale_maska s r? c C s4 t �� \} }t�t�� |jddd� W 5 Q R X d S )Nr �mask�Znonpos)r+ r, �pytestZraises� ValueErrorr8 r: r r r �test_extra_kwargs_raisen s rD c C sB t �� \} }|�d� |j|j�� �� }tt� �� t�s>t �d S )Nr ) r+ r, r8 Z transAxes� transDatar � isinstancer r r )r/ r0 Ztformr r r � test_logscale_invert_transformt s rG c C s2 t �� \} }|�d� t|j�}ttdd��}d S )Nr ZcliprA )r+ r, r8 �reprrE r )r/ r0 �sr r r � test_logscale_transform_repr~ s rJ Zlogscale_nonpos_valuesg{�G�z�?Zmpl20)r& r' r( Ztol�stylec C s t j�d� t jjtd�d�} t�dd�\}\\}}\}}|j| ddd� |�d� |j| ddd� |jdd d � t � d dd �}t � | �}d d| t � d| � t � | � }|� ||| || � |�d� t � dd�} | d } | d } |j| | | d� |�d� |�d� d S )Ni!N,g @�@)�sizer )���r r )�rangeZbinsr r@ )Znonposyr r) r* r ���r r )�yerr)r ZrandomZseedZnormal�intr+ r, Zhistr8 r r= ZcosZ fill_betweenZlogspaceZerrorbarr- ) r> r/ Zax1Zax2Zax3Zax4ZxdataZydataZedatar# r1 rP r r r �test_logscale_nonpos_values� s$ & rR c C s t �� \} }|�tdd�tdd�� |�d� |�� }t�t�� |j dd� W 5 Q R X |�� |ksht �t�t�� |j dd� W 5 Q R X |�� |ks�t �|� d� |� � }t�t�� |j dd� W 5 Q R X |� � |ks�t �t�t�� |j dd� W 5 Q R X |� � |k�st �d S ) Nr r r )�leftrO )�right)Zbottom)�top)r+ r, r4 rN r- r rB Zwarns� UserWarning�set_xlimr r8 r Zset_ylim)r/ r0 Z original_xlimZ original_ylimr r r �test_invalid_log_lims� s$ rX Zfunction_scales)r& r' r( rK c C sV dd� } dd� }t �� \}}t�dd�}|�||� |jd|| fd� |�dd� d S ) Nc S s | d S )Nr r �r# r r r �inverse� s z$test_function_scale.<locals>.inversec S s | d S )Nr r rY r r r �forward� s z$test_function_scale.<locals>.forwardr i� Zfunction)Z functions)r+ r, r r r r- rW )rZ r[ r/ r0 r# r r r �test_function_scale� s r\ ) Zmatplotlib.pyplotZpyplotr+ Zmatplotlib.scaler r r Z matplotlib.testing.decoratorsr r Znumpyr Z numpy.testingr r5 �platformrB r r% r2 r7 r; r? rD rG rJ rR rX r\ r r r r �<module> sF � � � �