Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/frame/__pycache__/test_frame_subplots.cpython-37.pyc
Назад
B �5�g*j � @ s� d Z ddlZddlZddlZddlZddlm mZ ddl Z ddl m Z m Z mZ ddlmZ ddlmZ ddlmZ ejjZe jG dd� de��ZdS )z Test cases for DataFrame.plot � N)� DataFrame�Series� date_range)� TestPlotBase)� pprint_thingc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zejj d d �d d � �Z d d� Z ej� di dddfddidddfg�dd� �Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zej� d-d.d/d0d1d2d3dd4gfg�ej� d5d6d7d8g�d9d:� ��Zej� d;d8d<d=�d8d<d>d?�d@d<d=�d@d<d>d?�d8dAd=�d8dAd>d?�d@dAd=�d@dAd>d?�d8d<dB�d8d<d>dC�d@d<dB�d@d<d>dC�d8d<dDdE�d8d<d>dDdF�d@d<dDdE�d@d<d>dDdF�d8dAdDdE�d8dAd>dDdF�d@dAdDdE�d@dAd>dDdF�d8d<dDdG�d8d<d>dDdH�d@d<dDdG�d@d<d>dDdH�g�dIdJ� �Zej� d;d8dAd=�d8d<d=�d@dAd=�d@d<d=�d8d<dB�d@d<dB�g�dKdL� �Zej� d;d8dAd=�d8d<d=�d@dAd=�d@d<d=�d8d<dB�d@d<dB�g�dMdN� �Z dOdP� Z dUdSdT�Z d2S )V�TestDataFramePlotsSubplotsc C sd t �| |� dd l}|�� t�� | _ttj j dd�tj j dd�t� d�tj j dd� d��| _ d S )Nr � )�size)�A�B�C) r � setup_methodZ matplotlibZ rcdefaults�tmZmakeTimeDataFrameZtdfr �np�random�uniform�arangeZ hexbin_df)�self�methodZmpl� r �e/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/frame/test_frame_subplots.pyr s z'TestDataFramePlotsSubplots.setup_methodc C s t tj�dd�ttjd d� �d�}�x�dD �]�}|j|dddd�}| j|ddd� |j d ksbt �x,t ||j �D ] \}}| j |t|�gd � qpW xx|d d � D ]h}| �|j� | j|�� d d � |dkr�| js�| j|jdd�d d � | j|j�� d d � | �|�� � q�W | �|d j� | �|d �� � | �|d jdd�� | �|d j�� � | �|d �� � |j|dd d�}xX|D ]P}| �|j� | �|�� � | �|jdd�� | �|j�� � | �|�� � �q~W |j|dd d�}x |D ]}|�� d k�s�t ��q�W q,W d S )N� � )�index)�bar�barh�line�areaT)�kind�subplots�sharex�legend)r � )�axes_num�layout)r )�labels���F)�visibler )�minor���)r r r )r r r! )r r r �rand�list�string� ascii_letters�plot�_check_axes_shape�shape�AssertionError�zip�columns�_check_legend_labelsr �_check_visible�xaxis�get_xticklabelsZ mpl_ge_3_1_0� get_label�get_yticklabelsZ get_legend)r �dfr �axes�ax�columnr r r � test_subplots* s: $ z(TestDataFramePlotsSubplots.test_subplotsc C s� t dddd�}ttj�dd�|d�}�x�dD �]|}|j|ddd �}| j|dd d � xj|d d � D ]Z}| �|j� | j|� � d d� | j|j dd�d d� | j|j� � d d� | �|� � � q`W | �|d j� | �|d � � � | �|d j dd�� | �|d j� � � | �|d � � � | j |dd� |j|dd ddd�}xj|D ]b}| �|j� | �|� � � | �|j dd�� | �|j� � � | �|� � � | j |dddd� �qDW q,W d S )Nz 2014-07-01�Mr )�start�freq�periodsr )r )r r T)r r r )r r" )r# r$ r& F)r' )r( r) r )�xrot�- � )r r r Zrot�fontsize)Z xlabelsizerC Z ylabelsize) r r r r r* r. r/ r5 r6 r7 r8 r9 Z_check_ticks_props)r �idxr: r r; r< r r r �test_subplots_timeseriesP s2 z3TestDataFramePlotsSubplots.test_subplots_timeseriesc C s t �dddg�tjddd�tjddd�tjdd d�gt�d �t�d �t�d �gtjd d d�tjd d d�tjd d d�gdddgd�}t|�}ddddg}xF|D ]>}|j|d�}|�� d �� d }|| j }||k� � s�t �q�W d}t j t|d�� |jdd� W d Q R X d S ) Nr" � � i�����s)�unitr �m�hz2017-08-01 00:00:00z2017-08-01 02:00:00z2017-08-02 00:00:00T)�utcZThisZshouldZfail)�numeric� timedelta�datetime_no_tz�datetime_all_tz�textrP rQ rR rS )�yr zno numeric data to plot)�matchrT )r �array�pdZ Timedelta� to_datetimer r. � get_lines�get_data�values�allr1 �pytest�raises� TypeError) r �data�testdataZy_cols�colr< �result�expected�msgr r r � test_subplots_timeseries_y_axisp s* z:TestDataFramePlotsSubplots.test_subplots_timeseries_y_axisz6not support for period, categorical, datetime_mixed_tz)�reasonc C s t �dddg�tjddd�tjddd�tjddd�gtjd d d gd d d gd d �tjddd�t�d�t�d�gd�}t|�}|jddd�}|�� d � � d |d j k� � s�t �|jddd�}|�� d � � d |d j k� � s�t �|jddd�}|�� d � � d |d j k� � �st �dS )a� This test will fail for: period: since period isn't yet implemented in ``select_dtypes`` and because it will need a custom value converter + tick formatter (as was done for x-axis plots) categorical: because it will need a custom value converter + tick formatter (also doesn't work for x-axis, as of now) datetime_mixed_tz: because of the way how pandas handles ``Series`` of ``datetime`` objects with different timezone, generally converting ``datetime`` objects in a tz-aware form could help with this problem r" rI rJ z2017-08-01 00:00:00�H)rA z2017-08-01 02:00z2017-08-02 00:00:00�c�b�aF)� categoriesZorderedT)rO z2017-08-01 02:00:00)rP �period� categorical�datetime_mixed_tzrP rn )�xrU r ro rp N) r rW rX ZPeriodZ CategoricalrY r r. rZ r[ r\ r] r1 )r ra rb Z ax_periodZax_categoricalZax_datetime_mixed_tzr r r �-test_subplots_timeseries_y_axis_not_supported� s$ &zHTestDataFramePlotsSubplots.test_subplots_timeseries_y_axis_not_supportedc C s� t tj�dd�ttjd d� �d�}|jddd�}| j|ddd� |j dksPt �|jddd�}| j|ddd� |j dks|t �|jdd d�}| j|ddd� |j dks�t �|jdd d�}| j|dd d� |j d ks�t �|jdd d�}| j|dd d� |j d k�st �|jdd d�}| j|dd d� |j d k�s0t �d}t j t |d�� |jddd� W d Q R X d}t j t |d�� |jddd� W d Q R X d S )Nr r )r T)rI rI )r r$ )r# r$ )r) rI )rI r) )r" � )r) rs )rs r) )rs r" z1Layout of 1x1 must be larger than required size 3)rV )r" r" z1At least one dimension of layout must be positive)r) r) )r r r r* r+ r, r- r. r/ r0 r1 r^ r_ � ValueError)r r: r; rf r r r �!test_subplots_layout_multi_column� s2 $z<TestDataFramePlotsSubplots.test_subplots_layout_multi_columnz:kwargs, expected_axes_num, expected_layout, expected_shaper" )r" r" )r" r$ )r r c C sZ t tj�dd�ttjd d� �d�}|jf ddi|��}| j|||d� |j |ksVt �d S )Nr r" )r r T)r# r$ ) r r r r* r+ r, r- r. r/ r0 r1 )r �kwargsZexpected_axes_numZexpected_layoutZexpected_shaper: r; r r r �"test_subplots_layout_single_column� s $z=TestDataFramePlotsSubplots.test_subplots_layout_single_columnc C sf t �d ��R ttj�dd��}|jddd� ttj�dd�tddd�d�}|jddd� W d Q R X d S ) N�d rs T)r rI )r r$ z1/1/2000)rB )r )r �assert_produces_warningr r r �randnr. r )r r: r r r �test_subplots_warnings� s z1TestDataFramePlotsSubplots.test_subplots_warningsc C sx | j �dd�\}}ttj�dd�ttjd d� �d�}|j d|d ddd�}| j |dd d � |j d ksjt �|d j |ks|t �|j d|d ddd�}| j |dd d � |j d ks�t �|d j |ks�t �| j |d dd � t�� d}tjt|d��& | j �dd�\}}|j d|d� W d Q R X | j �dd�\}}t�� �� t�dt� ttj�dd�ttjd d� �d�}|j d|dddd�}| j |ddd � |j dk�s�t �|j d|dddd�}| j |ddd � |j dk�s�t �|j d|dddd�}W d Q R X | j |ddd � |j dk�s t �| j �d d �\}}ttj�dd �ttjd d� �d�}|j d|gddd�}| j |d dd � |j dk�stt �d S ) NrI r r )r Tr F)r r< r �sharey)r" r )r# r$ )r r" � )rI r z@The number of passed axes must be 3, the same as the output plot)rV )r r< �ignorers )rI r" )r r< r$ r r| )rI rI )rs )rI r) )r) rI )r" r" )r" )�pltr r r r r* r+ r, r- r. r/ r0 r1 Zfigurer �closer^ r_ rt �warnings�catch_warnings� simplefilter� UserWarning)r �figr; r: �returnedrf r r r �test_subplots_multiple_axes sJ $ $$z6TestDataFramePlotsSubplots.test_subplots_multiple_axesc C sp | j jddddd�\}}| j jddddd� ttj�dd �td d dd �d �}x*t|� � �D ]\}}|| j |dd� qZW x*|dd� � � D ]}| j |� � dd� q�W x&|d � � D ]}| j |� � dd� q�W x2|dddgdgf � � D ]}| j |� � dd� q�W x4|dddgdgf � � D ]}| j |� � dd� �qW x4|dddgdgf � � D ]}| j |� � dd� �qPW d S )Nr T)r r| g�������?gffffff�?g333333�?)�left�rightZhspaceZwspacer � z 2014-07-01r? )r@ rA rB )r rJ )r< rF r r) F)r' r" rI )r r Zsubplots_adjustr r r rz r � enumerate�ravelr. r5 r7 r9 )r r� r; r: �ir< r r r �test_subplots_ts_share_axesE s" z6TestDataFramePlotsSubplots.test_subplots_ts_share_axesc C s� ddddgddddgddddgd �}t |td d �d �}|d dg jdd�}|d j|d dd� | j|d �� dd� | j|d �� dd� x"|�� D ]}| j|�� dd� q�W d S )Ng �?g @g @g @rJ r" r rs )r r r z 2014 10 11z 2014 10 14)r r r T)r r r )r< � secondary_yF)r' )r r r. r5 r7 r� r9 )r �dr: r; r< r r r �'test_subplots_sharex_axes_existing_axesb s $zBTestDataFramePlotsSubplots.test_subplots_sharex_axes_existing_axesc C s� t tj�dd�td�d�}|jdd�}x.|D ]&}| j|dgd� t|j�dks,t �q,W t � � |jddd �}x.|D ]&}| j|dgd� t|j�dksrt �qrW t � � |jdd �}| j|d gd d� t|j�d ks�t �t|j j�dks�t �d S ) NrJ Zaaaaa)r3 T)r rl )r% r" )r r� )r� z a (right)r ) r r r r* r+ r. r4 �len�linesr1 r r� Zright_ax)r r: r; r< r r r �test_subplots_dup_columnso s z4TestDataFramePlotsSubplots.test_subplots_dup_columnsc C s^ t �ddddg�}tdgd ttdd��d �td�d �}|jjd d d �}t�|j � � |� d S ) Ng�������?g �?g $@rx r rJ r" r} )r r )r T)�grid�log) r rW r r+ �ranger. r r �assert_numpy_array_equal�yaxis� get_ticklocs)r re r: r< r r r �test_bar_log_no_subplots� s &z3TestDataFramePlotsSubplots.test_bar_log_no_subplotsc C sn t �ddddddg�}ttddg�tdd g�g�jjd d d �}t�|d j� � |� t�|d j� � |� d S )Ng�������?g �?g $@g Y@g @�@g ��@�� i, i� T)r� r r r" ) r rW r r r. r r r� r� r� )r re r< r r r �test_bar_log_subplots� s z0TestDataFramePlotsSubplots.test_bar_log_subplotsc C sr | j }|jjdd�}t|t�s"t�| j|d dddgd� x4dD ],}|jj|dd�}| j||dddgd d � q>W d S ) NT)r �height�weight�category)� expected_keys)�dictr; Zboth)� return_typer F)r� Zcheck_ax_title)Zhist_dfr. �box� isinstancer r1 Z_check_box_return_type)r r: rd �tr� r r r �!test_boxplot_subplots_return_type� s z<TestDataFramePlotsSubplots.test_boxplot_subplots_return_typec C sH dd l m} ttj�dd�tddd�td�d�}|jddd d �\}}|j d |d �}x0|D ](}t |j �dksnt �| j |�� d d � qXW | j |d �� d d � | j |d jd d�d d � | j |d �� d d � | j |d jd d�d d � t�� |�dd�\}}t�t�� |j d |d d�}W d Q R X x4|D ],}t |j �dk�s@t �| j |�� d d � �q(W | j |d �� d d � | j |d jd d�d d � | j |d �� d d � | j |d jd d�d d � t�� |�dd�\}}|j d |d �}x\|D ]T}t |j �dk�s�t �| j |�� d d � | j |�� d d � | j |jd d�d d � �q�W t�� d S )Nr r rI z1/1/2000)rB ZAB)r r3 r" T)r )r r< )r' F)r( )r r< r )Zmatplotlib.pyplotZpyplotr r r rz r r+ r r. r� r� r1 r5 r9 r7 r r� ry r� )r r r: r� r; r< r r r �$test_df_subplots_patterns_minorticks� sD z?TestDataFramePlotsSubplots.test_df_subplots_patterns_minorticksc C s� t tj�dd��}tj|jdd �df< tj|jd d�df< | j�dd�\}}|jj |ddd� t� d dd �}t� d dd �}t � |d � � |� t � |d � � |� d S ) Nr rI rJ r" r TF)r< r r g @g �?g �)r r r r* �nanZilocr r r. r r r r� Z get_xticks)r r: ZfigsZaxsZ expected_ax1Z expected_ax2r r r �test_subplots_sharex_false� s z5TestDataFramePlotsSubplots.test_subplots_sharex_falsec C s� t ddd�}ttj�dd�|d�}i }t| jjd�r<d|d< | jjd |�\}}t � d ��: |j |d d � t j dd ��}| j� |� W d Q R X W d Q R X d S )N�nowr )r@ rB r )r Zget_constrained_layoutTZconstrained_layoutrI r )r< )Zreturn_filelike)rI )r r r r r* �hasattrr ZFigurer r ry r. Z ensure_cleanZsavefig)r rG r: rv r� r; �pathr r r � test_subplots_constrained_layout� s z;TestDataFramePlotsSubplots.test_subplots_constrained_layoutz index_name, old_label, new_label)N� �new)�oldr� r� )Nr� r� )Nr� r" Nr� rI r r r r c s� t ddgddggddgd�}||j_|j|dd�}td d � |D ��sHt�t�fd d �|D ��sbt�|j|� � dd �}t� fd d �|D ��s�t�t� fdd �|D ��s�t�d S )Nr" rI rJ zType AzType B)r3 T)r r c s s | ]}|� � d kV qdS )r� N)� get_ylabel)�.0r< r r r � <genexpr> s zSTestDataFramePlotsSubplots.test_xlabel_ylabel_dataframe_subplots.<locals>.<genexpr>c 3 s | ]}|� � � kV qd S )N)� get_xlabel)r� r< )� old_labelr r r� s )r ZylabelZxlabelr c 3 s | ]}|� � t� �kV qd S )N)r� �str)r� r< )� new_labelr r r� s c 3 s | ]}|� � t� �kV qd S )N)r� r� )r� r< )r� r r r� s )r r �namer. r] r1 )r r Z index_namer� r� r: r; r )r� r� r �%test_xlabel_ylabel_dataframe_subplots� s z@TestDataFramePlotsSubplots.test_xlabel_ylabel_dataframe_subplotsrv T)r �stackedg�������?)r r� �widthr F)r r )r r r� �edge)r r� �align)r r� r� r� )r r r� )r r r� r� c C s6 t dgd ttd��d�td�d�}| j|f|� d S )Nr rJ )r r )r )r r+ r� �_check_bar_alignment)r rv r: r r r � test_bar_align_multiple_columns s #$z:TestDataFramePlotsSubplots.test_bar_align_multiple_columnsc C s" t tj�d��}| j|f|� d S )NrJ )r r r rz r� )r rv r: r r r � test_bar_align_single_column8 s z7TestDataFramePlotsSubplots.test_bar_align_single_columnc C s. t tj�dd��}| j|fddd�|�� d S )NrJ g�������?g�������?)r� �position)r r r rz r� )r rv r: r r r �test_bar_barwidth_positionG s z5TestDataFramePlotsSubplots.test_bar_barwidth_positionc C s� t tj�dd��}x~dD ]v}|jjd|d�}|j�� }t� |t� ddddd g�� |� � d ksbt �|j d �� d ksxt �|j d �� d kst �qW | j|dddd� | j|dddd� | j|dddd� | j|dddd� | j|dddd� d S )NrJ )r" g �?T)r� r� r r" rI r rs )g �g @g �r) g @r )r r� r� r F)r r r� )r r r rz r. r r6 r� r r� rW �get_xlimr1 �patches�get_xr� )r r: �wr< Zticksr r r � test_bar_barwidth_position_intV s z9TestDataFramePlotsSubplots.test_bar_barwidth_position_int�center� �?c C sB |j ||||||dd�}| �|�}�x|D �]} |dkrt| j} | �� \} } tdd� | jD ��} tdd� | jD ��}nH|dkr�| j} | �� \} } tdd� | jD ��} td d� | jD ��}nt �t � | | d � t � | |d � | jd }|dk�r |dk�s|dk�r |� � }||� � | }n�|dk�rZ|d k�rZ|� � |� � t|j� | }|� � }n||dk�r�|dk�sx|dk�r�|�� |�� | }|�� }n@|dk�r�|d k�r�|�� |�� t|j� | }|�� }nt �| �� t�t|��k�� �s�t�|d k�rt � | �� d |� q*|dk�r6t � | �� d |� q*t �q*W |S )NT)r r� r r� r� r� r� r c s s | ]}|� � V qd S )N)r� )r� �pr r r r� � s zBTestDataFramePlotsSubplots._check_bar_alignment.<locals>.<genexpr>c s s | ]}|� � |�� V qd S )N)r� � get_width)r� r� r r r r� � s r c s s | ]}|� � V qd S )N)�get_y)r� r� r r r r� � s c s s | ]}|� � |�� V qd S )N)r� � get_height)r� r� r r r r� � s g �?r Fr� r� )r. Z_flatten_visibler6 r� �minr� �maxr� Zget_ylimrt r Zassert_almost_equalr� r� r� r3 r� r� r� r r r] r1 )r r: r r� r r� r� r� r; r< ZaxisZax_minZax_maxZmin_edgeZmax_edger� r� r� r r r r� i sX z/TestDataFramePlotsSubplots._check_bar_alignment)r FFr� r� r� ) �__name__� __module__� __qualname__r r> rH rg r^ �markZxfailrr ru Z parametrizerw r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s� & "3% ; . r )�__doc__r, r� �numpyr r^ Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasrX r r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr Zpandas.io.formats.printingr r� ZslowZ pytestmarkZskip_if_no_mplr r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка