OwlCyberSecurity - MANAGER
Edit File: test_units.cpython-38.pyc
U ��]� � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dl Z d dl Z d dl Z G dd� de�Ze jdd� �Zedgd d i�e �� d �d gd dd�dd� �Zedgd d i�e �� d �d gddd�dd� �Zdd� Zedgd gddidd�d d � �Zed gd gddidd�d d � �Zd!d"� Zd#d$� ZdS )%� )� MagicMockN)�image_comparisonc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �Quantityc C s || _ || _d S �N)� magnitude�units)�self�datar � r �=/usr/lib/python3/dist-packages/matplotlib/tests/test_units.py�__init__ s zQuantity.__init__c C sL ddddddd�}| j |kr:|| j |f }t|| j |�S t| j| j �S d S )Ni g�?�< g�0��(?� i�� ))�hours�seconds)�minutesr )r r )�feet�miles)r �inches)r r )r r r )r Z new_unitsZfactorsZmultr r r �to s � z Quantity.toc C s t | j|�S r )�getattrr )r �attrr r r � __getattr__ s zQuantity.__getattr__c C s0 t �| j�r t| j| | j�S t| j| j�S d S r )�np�iterabler r r )r �itemr r r � __getitem__ s zQuantity.__getitem__c C s t �| j�S r )r Zasarrayr )r r r r � __array__$ s zQuantity.__array__N)�__name__� __module__� __qualname__r r r r r r r r r r s r c C sD t �� } dd� }dd� }t|d�| _tdd� d�| _t|d�| _| S )Nc sz t | d�r| ���jS t�| �r`z�fdd�| D �W S tk r\ � �fdd�| D � Y S X nt| � �� ����jS d S )Nr c s g | ]}|� � �j�qS r )r r ��.0�v)�unitr r � <listcomp>3 s z7quantity_converter.<locals>.convert.<locals>.<listcomp>c s" g | ]}t |� �� ����j�qS r )r � get_unitsr r r! ��axisr$ r r r% 5 s �)�hasattrr r r r �AttributeErrorr r& )�valuer$ r( r r' r �convert. s �z#quantity_converter.<locals>.convertc S s@ t | d�r| jS t�| �r<| D ]}t |d�r |j S q d S d S )Nr )r) r r r )r+ r( r# r r r � default_units: s z)quantity_converter.<locals>.default_units)Z side_effectc S s t j| d�S )N)Zlabel)�munitsZAxisInfo)�u�ar r r �<lambda>D � z$quantity_converter.<locals>.<lambda>)r. ZConversionInterfacer r, �axisinfor- )Zqcr, r- r r r �quantity_converter( s r4 Z plot_pintZaarch64g{�G�z�?g ZpngFZmpl20)�baseline_imagesZtol� extensionsZ remove_text�stylec C s� | t jt< tt�dd�d�}tt�dd�d�}t�� \}}|jdd� |�||d� |j td d �d d � |j td d�dd � |j � d� |j � d� | jjs�t�| jjs�t�| jjs�t�d S )Nr � r � r g333333�?��leftztab:bluei g r ztab:red)Zcolor�x r z tab:greenr r )r. �registryr r �linspace�plt�subplotsZsubplots_adjust�plotZaxhlineZaxvlineZyaxisZ set_unitsZxaxisr, Zcalled�AssertionErrorr3 r- )r4 �y�x�fig�axr r r �test_numpy_facadeK s rG Zplot_masked_unitsTc C sL t �dd�} t jj| | dk| dk @ d�}t|d�}t�� \}}|�|� d S )N���r9 ���� )�mask�meters)r r> Zma�arrayr r? r@ rA )r Z data_maskedZdata_masked_unitsrE rF r r r �test_plot_masked_unitsd s rN c C sJ | t jt< t�� \}}|�tdd�tdd�� |�tdd�tdd�� d S )N���rL � r � )r. r= r r? r@ �set_xlim�set_ylim)r4 rE rF r r r � test_empty_set_limits_with_unitsp s rT Z jpl_bar_unitsZdpir< )r5 r6 Z savefig_kwargr7 c C s� ddl m } dd lm m} |�� |�dd�}d|j d|j d|j g}d| d| d| g}|jd| dd d �d �}t� � \}}|j |||d � |� |d| ||d d| g� d S )Nr ��datetime�ET� �@� rJ � �� � � �Zdt)ZbottomrO ) rV � matplotlib.testing.jpl_units�testing� jpl_units�register�Duration�km�Epochr? r@ ZbarrS �rV r ZdayrD �w�brE rF r r r �test_jpl_bar_unitsy s ri Zjpl_barh_unitsc C s� ddl m } dd lm m} |�� |�dd�}d|j d|j d|j g}d| d| d| g}|jd| dd d �d �}t� � \}}|j |||d � |� |d| ||d d| g� d S )Nr rU rW rX rY rJ rZ r[ r\ r] r^ r: rO ) rV r_ r` ra rb rc rd re r? r@ ZbarhrR rf r r r �test_jpl_barh_units� s rj c C s t �tjg dd�t�g �� d S )Nzdatetime64[ns]�Zdtype)r? �scatterr rM r r r r �test_empty_arrays� s rm c C sR t jdddd�} t jt| �dd�}t j|d< t�� \}}|�| |� |j�� d S )Nz2005-02z2005-03z datetime64[D]rk �floatr ) r Zarange�len�nanr? r@ rl ZcanvasZdraw)�timesrC rE rF r r r � test_scatter_element0_masked� s rr )Z unittest.mockr Zmatplotlib.pyplotZpyplotr? Z matplotlib.testing.decoratorsr Zmatplotlib.unitsr r. Znumpyr �platformZpytest�objectr Zfixturer4 �get�machinerG rN rT ri rj rm rr r r r r �<module> sH " � � � �