OwlCyberSecurity - MANAGER
Edit File: test_cbook.cpython-38.pyc
U ��]wI � @ s~ d dl Z d dlZd dlmZ d dlZd dlmZmZ d dlmZ d dl Z d dl m Z m Z mZ d dlZd dlmZ d dlmZ d dlmZmZ dd� ZG d d � d e�ZG d d � d e�ZG d d� de�Z dd� Z e dd� �Z dd� Z ddiddifddiddifddd�dd d �ffZ ddd�ddid ddgiidfddd�ddiddgid d �dfddd�ddid dddgiidfddd!d"�dd!d#�ddgid d$�dffZ!ddd�ddd�i fddiddid dddgiifddiddiddgid%d&�fdd!d#�dd!d#�d d'd �fdd!d#�dd!d#�d#d'd �fdd!d#�dd!d#�d#d#d �fdd!d#�dd!d#�d#d d �fdd!d#�dd!d#�dd#ifdd!d#�dd!d#�d(d#iff Z"ej#�$d)e �d*d+� �Z%ej#�$d,e!�d-d.� �Z&ej#�$d/e"�d0d1� �Z'd2d3� Z(d4d5� Z)d6d7� Z*d8d9� Z+d:d;� Z,d<d=� Z-d>d?� Z.ej#�$d@e �/dA��0d!dB�dfe �/dA�dfe �/dA�e �/d!�fg�dCdD� �Z1dEdF� Z2dGdH� Z3dIdJ� Z4dKdL� Z5dMdN� Z6dOdP� Z7dQdR� Z8dS )S� N)�ref)�patch�Mock)�datetime)�assert_array_equal�assert_approx_equal�assert_array_almost_equal)� MatplotlibDeprecationWarning�delete_masked_pointsc C s. d} t �| �st�dddg}t �|�r*t�d S )N�string�listZofZstings)�cbookZ is_hashable�AssertionError)�sZlst� r �=/usr/lib/python3/dist-packages/matplotlib/tests/test_cbook.py�test_is_hashable s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �Test_delete_masked_pointsc C s� ddddddg| _ t�dd�| _dddtjtjdg| _t�| j�| _tjj| j| j d �| _ d d d d ddg| _ t�| j �| _ t ddd�t ddd�t ddd�t ddd�t ddd�t ddd�g| _ t�| j �| _ddd d ddg| _t�| j�| _d S )NFTg �?g @� � � � )�mask�a�b�c�d�e�fi� � � �r�g�m�y)Zmask1�np�arange�arr0�nan�arr1�array�arr2Zma�arr3�arr_s�arr_s2r �arr_dt�arr_dt2Z arr_colors�mcolorsZ to_rgba_array�arr_rgba��selfr r r � setup_method s �z&Test_delete_masked_points.setup_methodc C s& t �t�� td| j� W 5 Q R X d S )Nza string)�pytest�raises� ValueError�dmpr' r3 r r r �test_bad_first_arg, s z,Test_delete_masked_points.test_bad_first_argc C sV t | j| j�}ddddg}| j| | j| f}t|d |d � t|d |d � d S )Nr r r r )r9 r- r) r. r+ r �r4 �actualZind�expectedr r r �test_string_seq0 s z)Test_delete_masked_points.test_string_seqc C sX t | j| j�}dddg}| j| | j| �� f}t|d |d � t|d |d � d S �Nr r r )r9 r/ r, r0 � compressedr r; r r r � test_datetime7 s z'Test_delete_masked_points.test_datetimec C sX t | j| j�}dddg}| j| �� | j| f}t|d |d � t|d |d � d S r? )r9 r, r2 r@ r r; r r r � test_rgba> s z#Test_delete_masked_points.test_rgbaN)�__name__� __module__� __qualname__r5 r: r> rA rB r r r r r s r c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) �Test_boxplot_statsc C s� t j�d� d| _d| _t jj| j| jfddd�| _tddd d d d d ddddg �| _t � | j�| _ dddddt � ddddg�dd d d d � | _ d d!d"�| _d#d t � ddg�d$�| _d%d#d&�| _d d'd&�| _d S )(Ni� �% r � �?g �?)�size�meanZsigmarJ �med�q1�q3�iqr�cilo�cihi�whislo�whishi�fliers�labelgP��$�C@g@� ��ÿg>z�zD�*@gjt�xJ*@g�Dm�(� @gӾ�#W@g7�9�q�U@g��Eɳ E@g�qמ�C@g��]Y���?g�6 v�-@g I��Q�;@g �inޓ�?) rP rO rN rJ rK rS rL rM rR rQ g� N�!@g� ��?)rP rO g =ɳ E@)rR rQ rS g�eLM��?)rQ rR g�,��#W@)r% �randomZseedZnrowsZncolsZ lognormal�data�sorted� known_keysr � boxplot_stats� std_resultsr* �known_nonbootstrapped_res�known_bootstrapped_ci�known_whis3_res�known_res_percentiles�known_res_ranger3 r r r �setupG sd � � ��� ���zTest_boxplot_stats.setupc C s t | jt�st�d S �N)� isinstancerZ r r r3 r r r �test_form_main_listy s z&Test_boxplot_stats.test_form_main_listc C s | j D ]}t|t�st�qd S ra )rZ rb �dictr �r4 �resr r r �test_form_each_dict| s z&Test_boxplot_stats.test_form_each_dictc C s& | j D ]}t|�t| j�kst�qd S ra )rZ �setrX r re r r r �test_form_dict_keys� s z&Test_boxplot_stats.test_form_dict_keysc C s0 | j d }| j�� D ]\}}t|| |� qd S )Nr )rZ r[ �itemsr )r4 rf �key�valuer r r �test_results_baseline� s z(Test_boxplot_stats.test_results_baselinec C s> t j| jdd�}|d }| j�� D ]\}}t|| |� q"d S )Ni' )Z bootstrapr )r rY rV r\ rj r �r4 �resultsrf rk rl r r r �test_results_bootstrapped� s z,Test_boxplot_stats.test_results_bootstrappedc C s> t j| jdd�}|d }| j�� D ]\}}t|| |� q"d S )Nr �Zwhisr )r rY rV r] rj r rn r r r �test_results_whiskers_float� s z.Test_boxplot_stats.test_results_whiskers_floatc C s> t j| jdd�}|d }| j�� D ]\}}t|| |� q"d S )N�rangerq r )r rY rV r_ rj r rn r r r �test_results_whiskers_range� s z.Test_boxplot_stats.test_results_whiskers_rangec C sB t j| jddgd�}|d }| j�� D ]\}}t|| |� q&d S )Nr �_ rq r )r rY rV r^ rj r rn r r r �!test_results_whiskers_percentiles� s z4Test_boxplot_stats.test_results_whiskers_percentilesc C sn ddddg}t j| j|d�}|d }t||�D ]\}}|d |ks.t�q.t �| j�}|D ]}d|ksXt�qXd S )NZTest1r Zardvarkr ��labelsr rT )r rY rV �zipr )r4 rx ro rf Zlabr r r �test_results_withlabels� s z*Test_boxplot_stats.test_results_withlabelsc C s2 ddg}t �t�� tj| j|d�}W 5 Q R X d S )Nr r rw )r6 r7 r8 r rY rV )r4 rx ro r r r �test_label_error� s z#Test_boxplot_stats.test_label_errorc C s2 t jjdd�}t�t�� t�|�}W 5 Q R X d S )N)�" r| r| )rI )r% rU Znormalr6 r7 r8 r rY )r4 rV ro r r r � test_bad_dims� s z Test_boxplot_stats.test_bad_dimsc C s� t jdd�}t �d|dg�}tj|dd�}tj|dd�}|d d dksLt�|d d dks`t�t|d d ddg� |d d dks�t�|d d dks�t�t|d d g � d S ) N� )�shapei��� F)Z autorangeTr rQ rR rS )r% ZzerosZhstackr rY r r )r4 �xZ bstats_falseZ bstats_truer r r �"test_boxplot_stats_autorange_false� s z5Test_boxplot_stats.test_boxplot_stats_autorange_falseN)rC rD rE r` rc rg ri rm rp rr rt rv rz r{ r} r� r r r r rF F s 2 rF c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�Test_callback_registryc C s d| _ t�� | _d S )NZtest)�signalr �CallbackRegistry� callbacksr3 r r r r` � s z Test_callback_registry.setupc C s | j �||�S ra )r� �connect)r4 r �funcr r r r� � s z Test_callback_registry.connectc C s$ | j ji kst�| j j i ks t�d S ra �r� � _func_cid_mapr r3 r r r �is_empty� s z Test_callback_registry.is_emptyc C s$ | j ji kst�| j j i ks t�d S ra r� r3 r r r � is_not_empty� s z#Test_callback_registry.is_not_emptyc C s� | � � t� }| �| j|j�}t|�tks.t�| �� | �| j|j�}||ksRt�| �� t | j j �dksnt�t | j j �dks�t�~| � � d S )Nr ) r� r� r� r� �dummy�type�intr r� �lenr� r� )r4 Zmini_meZcid1Zcid2r r r �test_callback_complete� s z-Test_callback_registry.test_callback_completec C s d S ra r r3 r r r r� � s z Test_callback_registry.dummyc C s" t t�t�t�� ��d�s t�d S )Nr� )�hasattr�pickle�loads�dumpsr r� r r3 r r r � test_pickling� s �z$Test_callback_registry.test_picklingN) rC rD rE r` r� r� r� r� r� r� r r r r r� � s r� c s� G dd� dt �� dd� }� fdd�}t�� }|�d|� tjd d�}|�d|� tj|d�}|�d|� tj�d |d g|tg|� gg�| �S ) Nc @ s e Zd ZdS )z%raising_cb_reg.<locals>.TestExceptionN�rC rD rE r r r r � TestException� s r� c S s t �d S ra )� RuntimeErrorr r r r �raising_function� s z(raising_cb_reg.<locals>.raising_functionc s t | t�r� �| �d S ra )rb r� )�excp�r� r r � transformer� s z#raising_cb_reg.<locals>.transformer�foo)Zexception_handlerzcb, excp)� Exceptionr r� r� r6 �mark� parametrizer� )r� r� r� �cbZcb_oldZcb_filtr r� r �raising_cb_reg� s" ���r� c C s8 |d k r*t �|�� | �d� W 5 Q R X n | �d� d S )Nr� )r6 r7 Zprocess)r� r� r r r �'test_callbackregistry_process_exception s r� c C s� dddd�} dddg}dddg}dd d g}|t t�| �� ��ksDt�|t t�| �� ��ks^t�|t t�| �� ��ksxt�|t�|�ks�t�|t�|�ks�t�d S ) Nr r r �r r r r r r )r r )r r )r r )rW r Zsanitize_sequence�keysr �valuesrj )r �k�v�ir r r �test_sanitize_sequence s r� r r � forbidden�requiredr r )r r r )r� �allowed� alias_mapping)r )r� r� r r� )r r )r� r� )r )r� r� )r r� zinp, kwargs_to_normc C s( t �t�� tj| f|� W 5 Q R X d S ra )r6 r7 � TypeErrorr �normalize_kwargs)�inp�kwargs_to_normr r r �test_normalize_kwargs_failI s r� z)inp, expected, kwargs_to_norm, warn_countc C sL t jdd��6}t �d� |tj| f|�ks.t�t|�|ks>t�W 5 Q R X d S )NT��record�always��warnings�catch_warnings� simplefilterr r� r r� )r� r= r� Z warn_count�wr r r �test_normalize_kwargs_warnO s r� z inp, expected, kwargs_to_normc C sL t jdd��6}t �d� |tj| f|�ks.t�t|�dks>t�W 5 Q R X d S )NTr� r� r r� )r� r= r� r� r r r �test_normalize_kwargs_passX s r� c C sp t �td��4} td d�| _tjdd��}t�d� W 5 Q R X W 5 Q R X t|�dksVt �t |d j �dkslt �d S )N�sys)Z return_valueTr� r� r r ) r �objectr r � _getframer� r� Z_warn_externalr� r �str�message)Zmock_sysr� r r r �(test_warn_external_frame_embedded_pythona s r� c C s� t �d�} t �d�}t �d�d d d� }t�| ||�\}}}t jdddddddgdd�}t jdddddddgdd�}t jdddddddgdd�}t||� t||� t||� t�| |�\}}t||� t||� d S � Nr ���r r r r �float��dtype)r% r& r �pts_to_prestep�asarrayr � r� Zy1Zy2ZxsZy1sZy2sZx_targetZ y1_targetZ y2_targetr r r �test_to_prestepj s r� c C s t �g g �} | jdkst�d S �N)r r )r r� r r �Zstepsr r r �test_to_prestep_empty~ s r� c C s� t �d�} t �d�}t �d�d d d� }t�| ||�\}}}t jdddddddgdd�}t jdddddddgdd�}t jdddddddgdd�}t||� t||� t||� t�| |�\}}t||� t||� d S r� )r% r& r �pts_to_poststepr� r r� r r r �test_to_poststep� s r� c C s t �g g �} | jdkst�d S r� )r r� r r r� r r r �test_to_poststep_empty� s r� c C s� t �d�} t �d�}t �d�d d d� }t�| ||�\}}}t jddddddddgdd �}t jddd d d d ddgdd �}t jddd d d d ddgdd �}t||� t||� t||� t�| |�\}}t||� t||� d S ) Nr r� r g �?rH g @r r� r� r r )r% r& r �pts_to_midstepr� r r� r r r �test_to_midstep� s r� c C s t �g g �} | jdkst�d S r� )r r� r r r� r r r �test_to_midstep_empty� s r� �args� r c C s$ t �t�� tj| � W 5 Q R X d S ra )r6 r7 r8 r r� )r� r r r �test_step_fails� s r� c s G dd� d�� � fdd�t d�D � \} }}}}}t�� }|j|� tt|�d �t|�ks`t�t|�| ��t|�kszt�|dd � D ]}|�| |�s�t�q�|� | � |dd � D ]}|�| |�r�t�q�t � |dd � |dd � �D ]\}} |�|| �s�t�q�d S )Nc @ s e Zd ZdS )ztest_grouper.<locals>.dummyNr� r r r r r� � s r� c s g | ] }� � �qS r r ��.0�j�r� r r � <listcomp>� s z test_grouper.<locals>.<listcomp>r r r ) rs r �Grouper�joinrh r r Z get_siblingsZjoined�remove� itertools�product) r r r r r �objsr" �other�A�Br r� r � test_grouper� s $ $r� c s� G dd� d�� � fdd�t d�D �} t�� }|j| � |j}| D ]}t|�|ks@t�q@|t| d � }| dd � D ]}|t|� |ksrt�qrd S )Nc @ s e Zd ZdS )z#test_grouper_private.<locals>.dummyNr� r r r r r� � s r� c s g | ] }� � �qS r r r� r� r r r� � s z(test_grouper_private.<locals>.<listcomp>r r r )rs r r� r� �_mappingr r )r� r" �mapping�oZbase_setr r� r �test_grouper_private� s r� c C sj t �d�} | j}dt|�ks t�dt|�ks0t�t�|�}|dksFt�dt|�ksVt�dt|�ksft�d S )Nr r r )r% r& Zflat�nextr r �safe_first_element)r� �itZretr r r � test_flatiter� s r� c s� G dd� d�� t �g d�} t�| �dks,t�� fdd�td�D �}t �|d�} t�| �dks`t�t�d�}t �|d�} t�| �dks�t�� fd d�td �D �}t �|d�} t�| �d ks�t�tj�d d�}t �|d�} t�| �d ks�t�d dd gd dgdgg}t �|d�} t | t ��st�t | d tj ��r<| d jdk�s@t�t | d tj ��rb| d jdk�sft�t | d tj ��r�| d jdk�s�t�G dd� dtj �}tjdt d�}|dt |j d�}t �|d�} t| �d k�s�t�t | d |��s�t�d S )Nc @ s e Zd ZdS )z test_reshape2d.<locals>.dummyNr� r r r r r� � s r� r� )r r c s g | ] }� � �qS r r r� r� r r r� � s z"test_reshape2d.<locals>.<listcomp>r )r r c s" g | ]}� fd d�t d�D ��qS )c s g | ] }� � �qS r r r� r� r r r� s z-test_reshape2d.<locals>.<listcomp>.<listcomp>r )rs )r� r� r� r r r� s r )r r )r r r r r r )r )r )r c s( e Zd Z� fdd�Z� fdd�Z� ZS )z%test_reshape2d.<locals>.ArraySubclassc 3 s t � �� D ]}t�|�V q d S ra )�super�__iter__r% r* )r4 rl �� __class__r r r� s z.test_reshape2d.<locals>.ArraySubclass.__iter__c s t �t� �|��S ra )r% r* r� � __getitem__)r4 �itemr� r r r� s z1test_reshape2d.<locals>.ArraySubclass.__getitem__)rC rD rE r� r� � __classcell__r r r� r � ArraySubclass s r� � r� )r� )r� �buffer)r Z _reshape_2Dr% r r rs r& rU Zrandrb r Zndarrayr� rV r� )Zxnewr� r� r� r r� r �test_reshape2d� s6 &&& r� c C s" d\} }}dg| dg| dg| }d| f| | | | | fg}t �|�|ksVt�d\}}|dg| }t �|�|ks~t�dg| |d | � }||| f|| | || | | fg}t �|�|ks�t�|dg| }t �|�|ks�t�t �dgd �g k�s t�t �g �g k�s t�d S )N)r r r TFr )r � r )r Zcontiguous_regionsr )r r r r r= r r r r r �test_contiguous_regions. s (r� c C s2 | j td�tdd�d�}t�|�}|dks.t�d S )Nr r� � )�indexr )ZSeriesrs r r� r )Zpdr r< r r r �%test_safe_first_element_pandas_seriesE s r c C sx t �dd�d dd��}|ddd� t| �dks2t�t�t�� |dd� W 5 Q R X t�t�� |ddd � W 5 Q R X d S ) Nz3.0�argc S s d S ra r )Zprer Zpostr r r r� M s z$test_make_keyword_only.<locals>.funcr r )r r r )N)r Z_make_keyword_onlyr� r r6 Zwarnsr )Zrecwarnr� r r r �test_make_keyword_onlyL s r )9r� r� �weakrefr r� Z unittest.mockr r r Znumpyr% Z numpy.testingr r r r6 Zmatplotlib.cbookr Zmatplotlib.colorsZcolorsr1 r r r9 r r� r rF r� r� r� r� Z fail_mappingZwarn_passing_mappingZ pass_mappingr� r� r� r� r� r� r� r� r� r� r� r� r& Zreshaper� r� r� r� r� r� r r r r r r �<module> s� *0 � � �� ������ � � �� >