OwlCyberSecurity - MANAGER
Edit File: test_pickle.cpython-38.pyc
U ��]i � @ s d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dl m Z d dl m Z d dl mZ d dlmZ dd� Ze dgd gd d d i�e�� d �dd�dd� �Zdd� Zdd� Zdd� Zdd� ZG dd� de�Z d d � Z d d � Z d d!� Z ej �!d"ej"�#� �d#d$� �Z$dS )%� )�BytesION)�cm)�image_comparison)� rrulewrapperc C s� t �� } t�| t� tj� t �d�}t�|t� tj� t jdd�}t jt � d�dd� t � � t�|t� tj� t �� t j t � d�t � d�d� t�t � � t� tj� t �� } t �� }t �t � d�� |�d� t�| t� tj� d S ) N�y �polar)Z projection� Zfoobar�Zlabel)�xZheight�log)�plt�figure�pickle�dumpr �HIGHEST_PROTOCOL�subplot�axes�plot�np�arange�legendZbarZgcaZ set_yscale)�fig�ax� r �>/usr/lib/python3/dist-packages/matplotlib/tests/test_pickle.py� test_simple s r Z multi_pickleZpngTZaarch64g{�G�z�?g Zmpl20)Zbaseline_images� extensionsZ remove_textZtolZstylec C s t jddd�} t �d� t�d�t�d� }}t�ddd��dd� } }}t�|d �}t �d d d � t � t t d��� t �d d d � t j |d dgd� t � � t �d d d � t �|� t �d d d� t �|� t �d d d� t �|� t �d d d�}|�dd� |�dd� t �||||� t �d d d�}|�dd� |�dd� t �||||� t �d d d� t j||d dd� t jdd� t �d d d� t j||d ddd � t� }t�| |tj� t � d � t j j j i k�s�t �|�!d� t�"|�} t j j j i k�st �| �#� dk�st �d S ) NzFigure with a label?)r � )Zfigsizez!Can you fit any more in a figure?� r r �P g333333�� � � z//Zooo)Zhatches� � r � � z$x^2$r z upper left)Zlocg �g�������?g�������?)ZxerrZyerr�all)$r r Zsuptitler r Zlinspace�reshapeZsinr r �list�rangeZcontourfZcolorbarZ pcolormesh�imshowZpcolor�set_xlimZset_ylimZ streamplotZquiverZscatterr Zerrorbarr r r r �closeZ_pylab_helpersZGcfZfigs�AssertionError�seek�loadZ get_label)r r �y�data�u�vr Z result_fhr r r � test_complete* sJ r5 c C sb ddl m} ddlm} |� }| |�}|�ddd�}|�dddgdddg� t�|t� tj � d S )Nr )�FigureCanvasPdf)�Figurer! r" r ) Z matplotlib.backends.backend_pdfr6 Zmatplotlib.figurer7 � add_subplotr r r r r )r6 r7 r �_r r r r �test_no_pyploto s r: c C s* ddl m} | ddd�}t�|t� � d S )Nr )� RendererAggr � � )� matplotlib.backends.backend_aggr; r r r )r; Zrendererr r r � test_renderer{ s r? c C s^ ddl m} | d�}|jj}|�ddd�}|�t�d��dd�� |j� � t � |t � � d S )Nr )�new_figure_manageri� r! � r r# ) r> r@ Zcanvasr r8 r+ r r r( �drawr r r )r@ Zmanagerr r r r r � test_image� s rC c C s6 t jddd�} t �� }t�|�}t�|� t �� d S )N�o T)r )r r Zgcfr �dumps�loadsrB )r r �pfr r r � test_polar� s rH c @ s e Zd Zdd� ZdS )� TransformBlobc C sJ t �� | _t �� | _t �| j| j�| _t �| j�| _t �| j| j�| _d S �N) � mtransformsZIdentityTransformZidentityZ identity2ZCompositeGenericTransform� compositeZTransformWrapper�wrapper� composite2)�selfr r r �__init__� s ��zTransformBlob.__init__N)�__name__� __module__� __qualname__rP r r r r rI � s rI c C s| t � } t�| �}~ t�|�} | jj| jks.t�dd� | jj� � D �| j gksPt�| jj | jj ksdt�| jj | jj ksxt�d S )Nc S s g | ] }|� �qS r r )�.0r4 r r r � <listcomp>� s z"test_transform.<locals>.<listcomp>) rI r rE rF rM Z_childrL r. Z_parents�valuesrN Z input_dimsZ output_dims)�objrG r r r �test_transform� s "rX c C s@ t d�} zt�t�| �� W n tk r: td� � Y nX d S )Nr" z!rrulewrapper pickling test failed)r r rF rE �RecursionError�print)�rr r r �test_rrulewrapper� s r\ c C sN t jddd�\} }t�t�| ��} | jd �dd� | jd �� dksJt�d S ) Nr" T)Zsharexr r r< r! )r r< ) r Zsubplotsr rF rE r r, Zget_xlimr. )r Zaxsr r r � test_shared� s r] �cmapc C s t �| � d S rJ )r rE )r^ r r r � test_cmap� s r_ )%�ior r �platformZnumpyr ZpytestZ matplotlibr Z matplotlib.testing.decoratorsr Zmatplotlib.datesr Zmatplotlib.pyplotZpyplotr Zmatplotlib.transformsZ transformsrK r �get�machiner5 r: r? rC rH �objectrI rX r\ r] ZmarkZ parametrizeZcmap_drV r_ r r r r �<module> s6 � A