Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/__pycache__/test_misc.cpython-37.pyc
Назад
B �5�g�N � @ s� d Z ddlZddlZddlm mZ ddlm Z m Z ddl m Z ddlmZmZ ddlmZ ejjZejdd� �Zdd� ZejG d d � d e��ZejG d d � d e��ZdS ) z$ Test cases for misc plot functions � N)� DataFrame�Series)� TestPlotBase�_check_plot_worksc C s6 t dddgi�} tjtdd�� | �� W d Q R X d S )N�A� � z#matplotlib is required for plotting)�match)r �pytest�raises� ImportError�plot)�df� r �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/test_misc.py�test_import_error_message s r c C s" t jjj} d}tjt|d�� | dg g i d� W d Q R X d}tjt|d�� | dttd�dd gi d� W d Q R X | dt � dgd d d d �d�\}}}}|dks�t �|d ks�t �|d ks�t �|d d iks�t �| dttd�g i d�\}}}}|d ks�t �|d ks�t �|dk�s t �t |�dk�s t �d S )Nz@Called plot accessor for type list, expected Series or DataFrame)r � )� backend_name�data�args�kwargsz.should not be called with positional arguments)Zdtype�line�x�y�barF)r �kind�gridr zpandas.plotting._matplotlib� ) �plottingZ_coreZ PlotAccessorZ_get_call_argsr r � TypeErrorr �objectr �AssertionError�len)�func�msgr r r r r r r �test_get_accessor_args s2 $ r% c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestSeriesPlotsc C s2 t �| |� dd l}|�� t�� | _d| j_d S )Nr �ts)r � setup_method� matplotlibZ rcdefaults�tmZmakeTimeSeriesr' �name)�self�methodZmplr r r r( D s z TestSeriesPlots.setup_methodc C sb ddl m} t�d ��2 t|| jd� t|| jjd� || jdd�}W d Q R X | j|dgd� d S )Nr )�autocorrelation_plot)�seriesZTest)�label)�labels)�pandas.plottingr. r* �assert_produces_warningr r' �valuesZ_check_legend_labels)r, r. �axr r r �test_autocorrelation_plotM s z)TestSeriesPlots.test_autocorrelation_plotc C s. ddl m} t|| jd� t|| jdd� d S )Nr )�lag_plot)r/ � )r/ Zlag)r2 r7 r r' )r, r7 r r r � test_lag_plotX s z TestSeriesPlots.test_lag_plotc C s ddl m} t|| jdd� d S )Nr )�bootstrap_plot� )r/ �size)r2 r: r r' )r, r: r r r �test_bootstrap_plot^ s z#TestSeriesPlots.test_bootstrap_plotN)�__name__� __module__� __qualname__r( r6 r9 r= r r r r r& B s r& c @ s� e Zd Zejej�dddg�dd� ��Zdd� Z dd � Z ej� d �d d � �Z d d� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�TestDataFramePlots� pass_axisFTc C s: ddl m} tj}d }|r,| j�dd�\}}t�d�� tt j � dd��}W d Q R X tj t |� d�� t|d|d|d �}W d Q R X |d d j�� }d d d g} | �|| � | j|d dd dd� |d d d |d< t� t �� t|d|d|d �}W d Q R X |d d j�� }dddg} | �|| � | j|d dd dd� d S )Nr )� mpl_ge_3_0_0� �* �d )Zraise_on_extra_warnings�alwaysg�������?)�filterwarnings�frameZ range_paddingr5 z-2�0�2� �Z )Z xlabelsizeZxrotZ ylabelsizeZyrotr z-1.0z-0.5z0.0)Z"pandas.plotting._matplotlib.compatrC r �scatter_matrix�plt�subplotsr* Z RNGContextr �np�random�randnr3 � UserWarningr ZyaxisZget_majorticklabelsZ_check_text_labelsZ_check_ticks_props) r, rB rC rN r5 �_r ZaxesZ axes0_labels�expectedr r r �test_scatter_matrix_axisf s> z+TestDataFramePlots.test_scatter_matrix_axisc s� ddl m� ddlm} |}t�d �� t||dd� W d Q R X d}t||d|d�}| j|�� d d� ||d d d� d � d d d g}t||d|d�}| j|�� d d� ||d d d� d � t||d� j d �}� fdd�t � dd|d � � �D �}| j|�� d d� ||d d d� d � d}t t j�|�t j�|�t j�|�dg| d��}t||dd� d}t||d|d�}| j|�� d d� ||d d d� d � d d d g}t||d|d�}| j|�� d d� ||d d d� d � t||d� j d �}� fdd�t � dd|d � � �D �}| j|�� d d� ||d d d� d � dddg} t dddgdddgdddg| d��}||d| d�}|�� \} } | j| | d� d S )Nr )�cm)�andrews_curves�Name)rI � class_column)z#556270z#4ECDC4z#C7F464)rI r[ �colorr; )� linecolors�mapping� dodgerblue� aquamarine�seagreen)rI r[ �colormapc s g | ]}� � |��qS r )�jet)�.0�n)rX r r � <listcomp>� s z:TestDataFramePlots.test_andrews_curves.<locals>.<listcomp>r r )r �B�CrZ c s g | ]}� � |��qS r )rc )rd re )rX r r rf � s �b�g�rr rD )r\ )r] )r) rX r2 rY r* r3 r � _check_colors� get_linesrc rQ �linspace�nuniquer rR �rand�get_legend_handles_labels) r, �irisrY r �rgbar5 �cnames�cmaps�length�colors�handlesr1 r )rX r �test_andrews_curves� s^ $ $$$ $ $$$ $ z&TestDataFramePlots.test_andrews_curvesc s� ddl m� ddlm} |}t||dd�}t|�� �}t|j�� �}d}t||d|d�}| j |�� d d� ||d d d� d � d d d g}t||d|d�}| j |�� d d� ||d d d� d � t||d� j d �}� fdd�t � dd|d � � �D �} | j |�� d d� | |d d d� d � t||ddd�}t|�� �|| k�sJt�dddg} tdddgdddgdddg| d��}||d| d�}|�� \} } | j | | d� d S )Nr )rX )�parallel_coordinatesrZ )rI r[ )z#556270z#4ECDC4z#C7F464)rI r[ r\ r; )r] r^ r_ r` ra )rI r[ rb c s g | ]}� � |��qS r )rc )rd re )rX r r rf s z@TestDataFramePlots.test_parallel_coordinates.<locals>.<listcomp>r F)rI r[ Zaxvlinesri rj rk r rD )r rg rh rZ )r\ )r] )r) rX r2 rz r r" rm �xaxisZget_ticklabelsrl rc rQ rn ro r! r rq ) r, rr rz r r5 �nlinesZnxticksrs rt ru rw rx r1 r )rX r �test_parallel_coordinates� s: $ $$$ $ z,TestDataFramePlots.test_parallel_coordinatesz$ignore:Attempting to set:UserWarningc C s� ddl m} tttd��dd� td�D �dd� td�D � dd� td�D � d ��}||d d d �}|�� \}}td d� |D �|�}t|dd� d�}tt|dd� �t|dd� ��}x4|D ],\} } | d | d k r�| d | d k s�t�q�W dS )z For #15908r )rz � c S s g | ]}d �qS )r r )rd rU r r r rf s zSTestDataFramePlots.test_parallel_coordinates_with_sorted_labels.<locals>.<listcomp>r; c S s g | ]}d �qS )rD r )rd rU r r r rf s c S s g | ]}d �qS )r r )rd rU r r r rf s )Zfeat�classr T)Z sort_labelsc S s g | ] }|� � �qS r )� get_color)rd Zpolyliner r r rf ! s c S s | d S )Nr r )r r r r �<lambda># � zQTestDataFramePlots.test_parallel_coordinates_with_sorted_labels.<locals>.<lambda>)�key���r N) r2 rz r �list�rangerq �zip�sortedr! ) r, rz r r5 Z polylinesr1 Zcolor_label_tuplesZordered_color_label_tuplesZprev_next_tupels�prevZnxtr r r �,test_parallel_coordinates_with_sorted_labels s < z?TestDataFramePlots.test_parallel_coordinates_with_sorted_labelsc s� ddl m� ddlm} |}t�d �� t||dd� W d Q R X d}t||d|d�}dd � |jd d � D �}| j|d d � ||d d d � d � d ddg}t||d|d� dd � |jd d � D �}| j|||d d d � d � t||d� j d� � fdd �t � dd|d � � �D �}dd � |jd d � D �}| j|||d d d � d � ddddgddddgddddgg} t dddgdddgdddgddd gd ��}||d| d �}|�� \} } | j| | d � d S ) Nr )rX )�radvizrZ )rI r[ )z#556270z#4ECDC4z#C7F464)rI r[ r\ c S s g | ]}|� � d kr|�qS )r )� get_label)rd �pr r r rf 8 s z2TestDataFramePlots.test_radviz.<locals>.<listcomp>� r; )� facecolorsr^ r_ r` ra c S s g | ]}|� � d kr|�qS )r )r� )rd r� r r r rf = s )rI r[ rb c s g | ]}� � |��qS r )rc )rd re )rX r r rf A s r c S s g | ]}|� � d kr|�qS )r )r� )rd r� r r r rf B s g g �?g �?r rD ri rj rk )r rg rh rZ )r\ )r� )r) rX r2 r� r* r3 r �patchesrl rc rQ rn ro r rq ) r, rr r� r rs r5 r� rt ru rw rx r1 r )rX r � test_radviz+ s. $ $ "( z TestDataFramePlots.test_radvizc C s |j ddd��� }t|j�}|jd|d�}dd� |D �|ks@t�d}tjt|d �� |jd|d g d� W d Q R X tjt|d �� |jd|d d � d� W d Q R X d }tjt|d �� |jd |d� W d Q R X |j ddd�jdd|d d� d�}dd� |D �}||d d� dg k�st�d S )NrZ r )ZaxisT)rP �titlec S s g | ] }|� � �qS r )� get_title)rd r� r r r rf T s z:TestDataFramePlots.test_subplot_titles.<locals>.<listcomp>zjThe length of `title` must equal the number of columns if using `title` of type `list` and `subplots=True`)r zkittens > puppiesr zNUsing `title` of type `list` is not supported unless `subplots=True` is passedFZ SepalWidth)r r r� )rP Zlayoutr� c S s g | ]}|D ] }|� � �q qS r )r� )rd Zsublistr5 r r r rf n s rD r ) Zdrop�headr� �columnsr r! r r � ValueError)r, rr r r� r r$ Z title_listr r r �test_subplot_titlesM s z&TestDataFramePlots.test_subplot_titlesc C sz t t�d��}t�|d� tj�� }t�|d� tj�� }||ksFt�ddlm} |ddd�}|ddd�}||ksvt�d S )N)r; r; r )�get_standard_colorsr rR )� color_type) r rQ Zzerosr rz rR r! �!pandas.plotting._matplotlib.styler� )r, r Zrand1Zrand2r� �color1�color2r r r �$test_get_standard_colors_random_seedq s z7TestDataFramePlots.test_get_standard_colors_random_seedc C sd ddl m} |ddd�}|ddd�}|ddd�}t|�dks@t�t|�dksPt�t|�dks`t�d S )Nr )r� r �default)r� � r� )r� r� r" r! )r, r� r� r� Zcolor3r r r �+test_get_standard_colors_default_num_colors� s z>TestDataFramePlots.test_get_standard_colors_default_num_colorsc s� t dddgdddgddd gd d d gd ddgdd dgd��}|j�� j�� }dd� |�� dd� D �� t� fdd�� D ��s~t�d S )Nz 2017-02-03z 2017-03-03z 2017-01-01z Alice Andersz Bob BakerzCharlie Chapling�z�Ga��g\��(�$@g L�@i� ix � i i� i� �4 � )z account-start�clientZbalancezdb-idzproxy-idZrankc S s g | ] }|� � �qS r )� get_facecolor)rd �rectr r r rf � s z=TestDataFramePlots.test_plot_single_color.<locals>.<listcomp>r rD c 3 s | ]}|� d kV qdS )r Nr )rd r\ )rw r r � <genexpr>� s z<TestDataFramePlots.test_plot_single_color.<locals>.<genexpr>)r r� Z value_countsr r � get_children�allr! )r, r r5 r )rw r �test_plot_single_color� s z)TestDataFramePlots.test_plot_single_colorc C s� ddl m} ddlm} |�td��}|d|d�}t|�t|�ksFt�tt j � dd�t d �d �}|�t � ddd ��}|jjjd |d �}|jd �� |jd �� ks�t�d S )Nr )rX )r� r8 r )r\ �0 � ZABCD)r� � )r� � )Zfigsizer\ � )r) rX r� r� �gnuplotr� r" r! r rQ rR rS r� rn r r r r� r� )r, rX r� Z color_beforeZ color_afterr Z color_listr� r r r �%test_get_standard_colors_no_appending� s z8TestDataFramePlots.test_get_standard_colors_no_appendingc s� ddg}ddg� t tj�dd�|d�}ddd�}|jd|d �}d d � |�� d d d� D �}t� fdd�t|�D ��sxt�|jd|d �}dd � |� � d d� D �}t� fdd�t|�D ��s�t�d S )N�ari )g �?g� ��Q��?g333333�?)g333333�?gffffff�?gffffff�?r )r� )ri r� r )r r\ c S s g | ]}|� � d d� �qS )r r� )r� )rd r� r r r rf � s z<TestDataFramePlots.test_dictionary_color.<locals>.<listcomp>r rD c 3 s | ]\}}|� | kV qd S )Nr )rd �indexr\ )rV r r r� � s z;TestDataFramePlots.test_dictionary_color.<locals>.<genexpr>r c S s g | ] }|� � �qS r )r� )rd r� r r r rf � s c 3 s | ]\}}|� | kV qd S )Nr )rd r� r\ )rV r r r� � s ) r rQ rR rp r r� r� � enumerater! rm )r, � data_filesZdf1Z dic_colorr5 rw r )rV r �test_dictionary_color� s z(TestDataFramePlots.test_dictionary_colorc C s� t jjj}| j�� }|�dd�}|jd|d d d�|d d< |jd|d d d�|d d< |d d �� |d d �� ||d d d�s�t �||d d d�r�t �||d d d�s�t �||d d d�r�t �d S ) Nr r� �� r r )�sharex�� r rD ) r � _matplotlib�tools�_has_externally_shared_axisrO �figurerP � add_subplot�twinxr! )r, r# �fig�plotsr r r �&test_has_externally_shared_axis_x_axis� s z9TestDataFramePlots.test_has_externally_shared_axis_x_axisc C s� t jjj}| j�� }|�dd�}|jd|d d d�|d d< |jd|d d d�|d d< |d d �� |d d �� ||d d d�s�t �||d d d�r�t �||d d d�s�t �||d d d�r�t �d S ) Nr� r iA r r )�shareyiE r rD ) r r� r� r� rO r� rP r� Ztwinyr! )r, r# r� r� r r r �&test_has_externally_shared_axis_y_axis� s z9TestDataFramePlots.test_has_externally_shared_axis_y_axisc C sr t jjj}| j�� }|�dd�}|jd|d d d�|d d< d}tj t |d�� ||d d d � W d Q R X d S ) Nr� r iA r r )r� z&needs 'x' or 'y' as a second parameter)r �z) r r� r� r� rO r� rP r� r r r� )r, r# r� r� r$ r r r �4test_has_externally_shared_axis_invalid_compare_axis� s zGTestDataFramePlots.test_has_externally_shared_axis_invalid_compare_axisc C s� t tj�d�tj�d�d��}| j�� }|�dd�}|jd|d d d�|d d< |jd |d d d�|d d< |d d �� }|d d �� }|d j |d d d d �� d � |d j |d d d� |d j |d d dd �� d� |d j |d d d� |d j |d d dd �� d � |d j |d d d� |d j |dd� |d j |dd� |d d j � � � � �rxt�|d d j � � � � �s�t�|d d j � � � � �r�t�d S )Ni� )r� ri r rD r� r r )r� r� r� zExternal share only)r5 r� z"this label should never be visible)r5 zInternal share (twin) onlyz#this label should always be visibleZBothri �green)r5 r\ �yellow)r rQ rR rS rO r� rP r� r� r Z set_xlabelr{ r� Z get_visibler! )r, r r� r� Ztwin_ax1Ztwin_ax2r r r �test_externally_shared_axes s* z.TestDataFramePlots.test_externally_shared_axesN)r> r? r@ �tdZskip_if_no_scipyr �markZ parametrizerW ry r} rH r� r� r� r� r� r� r� r� r� r� r� r� r r r r rA d s /L/"$ rA ) �__doc__�numpyrQ r Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r Zpandas._testingZ_testingr* Z pandas.tests.plotting.commonr r r2 r r� ZslowZ pytestmarkZ skip_if_mplr r% Zskip_if_no_mplr& rA r r r r �<module> s "!
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.07 |
proxy
|
phpinfo
|
Настройка