Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/__pycache__/test_hist_method.cpython-37.pyc
Назад
B �5�g�m � @ s� d Z ddlZddlZddlZddlm mZ ddl m Z m Z m Z m Z ddlmZ ddlmZmZ ejjZejG dd� de��ZejG dd� de��ZejG d d � d e��ZdS ) z Test cases for .hist method � N)� DataFrame�Index�Series� to_datetime)� TestPlotBase�_check_plot_worksc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e j � dddddg�dd� �Ze j � dddg�dd � �Ze j � d d d g�d d!� �Zd"d#� Zejd$d%� �Zejd&d'� �Zd S )(�TestSeriesPlotsc C s2 t �| |� dd l}|�� t�� | _d| j_d S )Nr �ts)r � setup_methodZ matplotlibZ rcdefaults�tmZmakeTimeSeriesr �name)�self�methodZmpl� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/test_hist_method.pyr s z TestSeriesPlots.setup_methodc C sL t | jj� t | jjdd� t | jjdd� t�t�� t | jj| jjjd� W d Q R X t�t�� t | jj| jjjdd� W d Q R X | j� dd�\}}t | jj|d d � t | jj||d d � t | jj|d d � t� � | j� dd �\}\}}t | jj||d d� t | jj||d d� d}t j t |d�� | jj| jj|d� W d Q R X d S )NF)�grid)� � )�figsize)�by� )r �bins� T)�ax� default_axes)r �figurer )r r � )r r r z`Cannot pass 'figure' when using the 'by' argument, since a new 'Figure' instance will be created)�match)r r )r r �histr �assert_produces_warning� UserWarning�index�month�plt�subplots�close�pytest�raises� ValueError)r �figr �ax1�ax2�msgr r r �test_hist_legacy# s$ "z TestSeriesPlots.test_hist_legacyc C s< t tj�dd��}|jdd�d d }t|j�dks8t�d S )Nr r )r r )r �np�random�randnr �len�patches�AssertionError)r �dfr r r r �test_hist_bins_legacy> s z%TestSeriesPlots.test_hist_bins_legacyc C sb | j }d}tjt|d�� |jjdd� W d Q R X tjt|d�� |jjddgd� W d Q R X d S )Nz7The 'layout' keyword is not supported when 'by' is None)r )r r )�layoutr )�hist_dfr& r' r( �heightr )r r4 r, r r r �test_hist_layoutC s z TestSeriesPlots.test_hist_layoutc C s� | j }t�t�� t|jj|jdd�}W d Q R X | j|ddd� t�t�� t|jj|jdd�}W d Q R X | j|ddd� t�t�� t|jj|j dd�}W d Q R X | j|ddd� t�t�� t|jj|j d d�}W d Q R X | j|dd d� t�t�� t|jj|j dd�}W d Q R X | j|dd d� t�t�� t|jj|j d d�}W d Q R X | j|dd d� t�t�� t|jj|j d d�}W d Q R X | j|dd d� |jj|j ddd�}| j|dddd� d S )N)r r )r r6 r )�axes_numr6 )� ���)r; r )� r r= )r r< )r r )r; r )r< r= )r r= r; )r= r )� � )r r6 r )r: r6 r ) r7 r r r r r8 r �gender�_check_axes_shape�category� classroom)r r4 �axesr r r �test_hist_layout_with_byL s0 z(TestSeriesPlots.test_hist_layout_with_byc C sp ddl m}m} ttj�d��}ttj�d��}|d� |�� |d� |�� |� }|j}t |�dkslt �d S )Nr )�gcf�subplotr �y �z ) Zmatplotlib.pyplotrF rG r r. r/ r0 r rD r1 r3 )r rF rG �x�yr) rD r r r �test_hist_no_overlapq s z$TestSeriesPlots.test_hist_no_overlapc C s0 | j }|jj|jd�}t| j�� �dks,t�d S )N)r r )r7 r8 r r@ r1 r# � get_fignumsr3 )r r4 rD r r r �test_hist_by_no_extra_plots� s z+TestSeriesPlots.test_hist_by_no_extra_plotsc C sT ddl m} |� }|� }|�d�}d}tjt|d�� | jj||d� W d Q R X d S )Nr )r �o z&passed axis not bound to passed figure)r )r r )Zpylabr Z add_subplotr& r' r3 r r )r r Zfig1Zfig2r* r, r r r �+test_plot_fails_when_ax_differs_from_figure� s z;TestSeriesPlots.test_plot_fails_when_ax_differs_from_figurezhisttype, expected)�barT)� barstackedT)�stepF)� stepfilledTc C s0 t tj�dd��}|j|d�}| j||d� d S )Nr r )�histtype)�filled)r r. r/ �randintr �_check_patches_all_filled)r rU �expected�serr r r r �test_histtype_argument� s z&TestSeriesPlots.test_histtype_argumentz&by, expected_axes_num, expected_layout)Nr )r r )�br )r r c C sd ddg ddg }t tj�d�|dd�}d|j_t|jdd|d �}| j|||d � | � |d� d S ) N� �1�2� �a)r! r r\ T)r �legendr )r: r6 ) r r. r/ r0 r! r r r rA �_check_legend_labels)r r �expected_axes_num�expected_layoutr! �srD r r r �test_hist_with_legend� s z%TestSeriesPlots.test_hist_with_legendr Nr\ c C s` ddg ddg }t tj�d�|dd�}d|j_tjtdd �� |j d |d d � W d Q R X d S ) Nr] r^ r_ r` ra )r! r r\ z Cannot use both legend and label)r T�c)rb r �label) r r. r/ r0 r! r r& r' r( r )r r r! rf r r r � test_hist_with_legend_raises� s z,TestSeriesPlots.test_hist_with_legend_raisesc C s� | j �� \}}| jjjd|d�}t|j�dks2t�| �|j � � d� t � � | j �� \}}| jjjd|d�}| �|j � � d� t � � | j �� \}}| jjjdd|d�}t � � d S ) Nr )r r Z Frequency� horizontal)Z orientationr �leftT)�alignZstackedr )r# r$ r �plotr r1 r2 r3 �_check_text_labels�yaxisZ get_labelr r% Zxaxis)r �_r r r r �test_hist_kwargs� s z TestSeriesPlots.test_hist_kwargsc C s� | j �� \}}| jjjd|d�}| j|dd� |�� }| �|dgt|� � |� � }| �|dgt|� � t | jjj � t | jjj � | j �� \}}| jjj d|d�}| j|dd� |�� }| �|dgt|� � |� � }| �|dgt|� � d S )NT)�logyr �log)rp � ) r# r$ r rn r �_check_ax_scalesZget_xticklabelsro r1 Zget_yticklabelsr �kde�density)r rq r ZxlabelsZylabelsr r r � test_hist_kde� s z TestSeriesPlots.test_hist_kdec C s� | j �� \}}| jjjddd|d�}| j|dd� t|j�dksDt�| j |jdgd d� | j �� \}}| jjj dd|d �}| j|dd� |� � }t|�d ks�t�| � |dg� d S ) NTr r\ )rs r �colorr rt )rp )Z facecolors�r)rs rz r r ) r# r$ r rn r rv r1 r2 r3 Z _check_colorsrw � get_lines)r rq r �linesr r r �test_hist_kde_color� s z#TestSeriesPlots.test_hist_kde_color)�__name__� __module__� __qualname__r r- r5 r9 rE rL rN rP r& �mark� parametrizer[ rg rj rr �tdZskip_if_no_scipyry r~ r r r r r s( % r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d ddgfdd dgdd dgfg�dd� �Z ej � dddddg�dd� �Z ej � dd dg�ej � dd d g�d d � ��Z ej � dd dg�ej � dd d g�d d � ��Zd!d"� Zd#d$� Zd%d&� Zd S )'�TestDataFramePlotsc s� ddl m� t�t�� t| jj� W d Q R X tt j � dd��}t t j j | j| jdt jd��|d< t�t�� t|jdd�}W d Q R X | j|dd d � |d �� r�t�t|dg j� tt j � dd ��}t|j� tt j � dd ��}t t j j | j| jdt jd��|d < t�t�� t|jdd�}W d Q R X | j|ddd � t�t�� t|jddd� W d Q R X t�t�� t|jdd� W d Q R X t�t�� t|jd d� W d Q R X |d }d\}}d\}}|j||||d�}| j|||||d� d\}}d\}}|j||||d�}| j|||||d� t�� |jdddd�}� fdd �|�� D �} t�| d �� d � t�� |jdd �}| j|d d!� t�� t�t�� |jd"d#� W d Q R X d S )$Nr )� Rectangle�d r )�size�dtypeF)r r; )r r )r: r6 )r r r r )r= r )r6 � T)�sharex�sharey)r r )r )r )� � )r` �( )� xlabelsize�xrot� ylabelsize�yrotr= )� cumulativer rx c s g | ]}t |� �r|�qS r )� isinstance)�.0rJ )r� r r � <listcomp>6 s z:TestDataFramePlots.test_hist_df_legacy.<locals>.<listcomp>r< g �?)rt rt )rp rQ )�foo) �matplotlib.patchesr� r r r r r7 r r r. r/ r0 r rW �start_date_to_int64�end_date_to_int64�int64rA � get_visibler3 �_check_ticks_propsr% � get_children�assert_almost_equal� get_heightrv �external_error_raised�AttributeError) r r4 rD rZ �xf�yfr� r� r �rectsr )r� r �test_hist_df_legacy� sl z&TestDataFramePlots.test_hist_df_legacyc C s� t tj�d�tj�ddd�ttjjdddtjd��ttjjdddtjd�dd�d��}|�t�}d }t j t |d �� |� � W d Q R X d S ) Nr r l � �p�/ l @�spc�/ )r� T)�utc)ra r\ rh �dzDhist method requires numerical or datetime columns, nothing to plot.)r ) r r. r/ ZrandrW r r� Zastype�objectr& r' r( r )r r4 Zdf_or, r r r �*test_hist_non_numerical_or_datetime_raisesD s z=TestDataFramePlots.test_hist_non_numerical_or_datetime_raisesc C s> t tj�dd��}ttjj| j| jdtjd��|d< d dd�ddd�ddd�ddd�ddd�d dd�d dd�d dd�d dd�f }x4|D ],}|j |d d�}|d }| j |d|d� q�W d}t j t |d�� |j dd� W d Q R X t�d�}t j t |d�� |j dd� W d Q R X d}t j t |d�� |j dd� W d Q R X d S )Nr� r )r� r� )r r )r6 � expected_size)r= r )r r= )r; r; )r< r= )r= r< )r< r )r r< r6 )r6 r� r; )r: r6 z1Layout of 1x1 must be larger than required size 3)r )r r z)Layout must be a tuple of (rows, columns))r z1At least one dimension of layout must be positive)r< r< )r r. r/ r0 r rW r� r� r� r rA r& r' r( �re�escape)r r4 Zlayout_to_expected_sizeZ layout_testrD rY r, r r r r9 ] s: z#TestDataFramePlots.test_hist_layoutc C sX t tj�dd��}ttjj| j| jdtjd��|d< t |j dd� | j � � t �� d S )Nr� r )r� r� T)r )r r. r/ r0 r rW r� r� r� r r r# Z tight_layoutr r% )r r4 r r r �test_tight_layout� s z$TestDataFramePlots.test_tight_layoutc C sL t dddddgdddddgd��}t|jd d d d d dd�}| j|dd� d S )Ng �?g �?g333333�?g�������?r; �pig�rabbit)�length�animalT�alwaysr� r� r r )r �filterwarnings�columnr r r� )r� )r r r r� )r r4 rD r r r �test_hist_subplot_xrot� s z)TestDataFramePlots.test_hist_subplot_xrotzcolumn, expectedN�widthr� r8 c st t dddddgddddd gd dd d d gd �dddddgd�}t|jd|dd�� � fdd�td �D �}||kspt�d S )Ngffffff�?g�������?g333333�?g�������?g �?g �?g333333�?g�������?r; g333333 @r r )r� r� r8 r� r� �duck�chicken�horse)r! T)r r; )r r� r6 c s g | ]}� d |f � � �qS )r )Z get_title)r� �i)rD r r r� � s zGTestDataFramePlots.test_hist_column_order_unchanged.<locals>.<listcomp>)r r r �ranger3 )r r� rY r4 �resultr )rD r � test_hist_column_order_unchanged� s z3TestDataFramePlots.test_hist_column_order_unchangedzhisttype, expected)rQ T)rR T)rS F)rT Tc C s< t tjjdddd�ddgd�}|j|d�}| j||d � d S ) Nr r )r� r )r� ra r\ )�columns)rU )rV )r r. r/ rW r rX )r rU rY r4 r r r r r[ � s z)TestDataFramePlots.test_histtype_argumentr rh r� r\ c C s� |d kr|d k rdnd}d|f}|p*ddg}|d k r>|gd }t ddg ddg dd �}ttj�d d�|ddgd �}t|jd d ||d �}| j|||d� |d kr�|d kr�|d }x"t||�D ]\} } | � | | � q�W d S )Nr r ra r\ r] r^ r_ rh )r r` )r! r� T)r rb r r� )r: r6 r ) r r r. r/ r0 r r rA �ziprc ) r r r� rd re Zexpected_labelsr! r4 rD Zexpected_labelr r r r rg � s$ z(TestDataFramePlots.test_hist_with_legendc C sh t ddg ddg dd�}ttj�dd�|dd gd �}tjtd d �� |jd ||dd� W d Q R X d S )Nr] r^ r_ rh )r r` r ra r\ )r! r� z Cannot use both legend and label)r Tr� )rb r r� ri ) r r r. r/ r0 r& r' r( r )r r r� r! r4 r r r rj � s z/TestDataFramePlots.test_hist_with_legend_raisesc C sF t tj�dd��}| j�� \}}|jjd|d�}t|j �dksBt �d S )Nr r r )r r ) r r. r/ r0 r# r$ rn r r1 r2 r3 )r r4 rq r r r r �test_hist_df_kwargs� s z&TestDataFramePlots.test_hist_df_kwargsc C s� t �d��$ ttj�dd�ddddgd�}W d Q R X d d gd |d < | j�� \}}|jj d |d �}t |j �dkstt �| j�� \}}|jj |d�}t |j �dks�t �d S )Nr r r= �A�B�C�D)r� rJ rK r �E)r r r� )r r� ) r � RNGContextr r. r/ r0 r# r$ rn r r1 r2 r3 )r r4 rq r r r r � test_hist_df_with_nonnumerics s (z0TestDataFramePlots.test_hist_df_with_nonnumericsc C s� t tj�dd�td�d�}| j�� \}}|d jjd|d�}|d jj|ddd � | j |dd gd � |� � � � stt �|j � � � � s�t �t�� | j�� \}}|d jjdd|d �}|d jj|ddd � | j |jd d gd � |j� � � � r�t �|� � � � s�t �t�� | j�� \}}|d jjdd|d �}|d jj|dd� | j |jd dgd � |j� � � � �sft �|� � � � �sxt �t�� d S )Nr` r= �abcd)r� ra T)rb r r\ )r rb � secondary_yz b (right))�labels)rb r� r z a (right))r rb )r r. r/ r0 �listr# r$ rn r rc Z get_yaxisr� r3 Zright_axr r% Zleft_ax)r r4 rq r r r r �test_hist_secondary_legend s, z-TestDataFramePlots.test_hist_secondary_legend)r r� r� r� r� r9 r� r� r& r� r� r� r[ rg rj r� r� r� r r r r r� � s, U* r� c @ sb e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j � dddddg�dd� �Z dS )�TestDataFrameGroupByPlotsc s ddl m� ddlm} ttj�dd�dgd�}ttjj | j | j dtj d��|d < tj� dd d�|d < d gd |d < ||j |jd�}| j|d dd� t�� |j|jd�}| j|d dd� t�� |jd dd�}| j|ddd� | j|dd� t�� d\}}d\}}||j |jdd ||||dd� }x@|�� D ]4}� fdd�|�� D �} | d �� } t�| d � �q.W | j|||||d � t�� ||j |jdd �}| j|d d � t�� t�t�� ||j |jd!d"� W d Q R X d#} tj t | d$�� |jd d%d&� W d Q R X d S )'Nr )r� )� _grouped_histi� r r� )r� )r� r� r� r= r� �Xr� )r )r r )r: r6 r` )r Zrot)r r )r� )r� r� )r` r� T)r r� r r� r� r� r� rx c s g | ]}t |� �r|�qS r )r� )r� rJ )r� r r r� i s zFTestDataFrameGroupByPlots.test_grouped_hist_legacy.<locals>.<listcomp>r< g �?)r� r� r� r� )r rt rt )rp rQ )r r� z$Specify figure size by tuple instead)r �default)r r ) r� r� Z pandas.plotting._matplotlib.histr� r r. r/ r0 r rW r� r� r� r� r� rA r r% r r� Zravelr� r� r� rv r� r� r& r' r( ) r r� r4 rD r� r� r� r� r r� r8 r, r )r� r �test_grouped_hist_legacy9 s` z2TestDataFrameGroupByPlots.test_grouped_hist_legacyc C s� d}t tjjdd|d��}t tjjdd|d��}t�d�� tjjddg|d�}W d Q R X t|||d ��}|�d �}|� � }t |�d ks�t �t | j � � �d ks�t �t�� d S ) Nr � r� )r� �< �* r r )r8 �weightr@ r@ r )r r. r/ �normalr r� �choicer �groupbyr r1 r3 r# rM r% )r �nr� r8 Z gender_intZdf_int�gbrD r r r �test_grouped_hist_legacy2~ s z3TestDataFrameGroupByPlots.test_grouped_hist_legacy2c C s, | j }d}tjt|d�� |jd|jdd� W d Q R X d}tjt|d�� |jd|jdd� W d Q R X d }tjt|d�� |jd|jd d� W d Q R X t�t �� t |jd|jd d�}W d Q R X | j |d d d � t�t �� t |jd|jdd�}W d Q R X | j |d d d � |jd|jdd�}| j |ddd � |jd|jdd�}| j |ddd � |jd|jddd�}| j |dddd� t� � t�t �� t |jdddd�}W d Q R X | j |ddd � t�t �� t |jdd�}W d Q R X | j |ddd � |jddd �}| j |d dd � |jddd gd �}| j |ddd � d S ) Nz1Layout of 1x1 must be larger than required size 2)r r� )r r )r� r r6 z1Layout of 1x3 must be larger than required size 4r8 )r r; z1At least one dimension of layout must be positive)r< r< )r r r )r: r6 )r r< )r= r r= )r< r )r= r )r> r )r� r r6 r )r: r6 r rC )r r r; )r r@ )r; r )r r6 rB )r� ) r7 r& r' r( r r@ rB r r r r rA r% )r r4 r, rD r r r �test_grouped_hist_layout� sH z2TestDataFrameGroupByPlots.test_grouped_hist_layoutc C s� | j }| j�dd�\}}|jdddg|d d�}| j|ddd � t�||d � |d j|ksbt�|jd |d d �}| j|ddd � t�||d � |d j|ks�t�| j�dd�\}}d }t j t |d�� |jd|d�}W d Q R X d S )Nr r; r8 r� rB r )r� r )r r; )r: r6 rC r )r r z@The number of passed axes must be 1, the same as the output plot)r ) r7 r# r$ r rA r Zassert_numpy_array_equalr r3 r& r' r( )r r4 r) rD �returnedr, r r r � test_grouped_hist_multiple_axes� s z9TestDataFrameGroupByPlots.test_grouped_hist_multiple_axesc C sx | j }|jd|jdd�\}}| �|��||�s2t�| �|��||�sHt�| �|��||�r^t�| �|��||�rtt�d S )Nr8 T)r� r r� )r7 r r@ � get_x_axis�joinedr3 � get_y_axis)r r4 r* r+ r r r �test_axis_share_x� s z+TestDataFrameGroupByPlots.test_axis_share_xc C sx | j }|jd|jdd�\}}| �|��||�s2t�| �|��||�sHt�| �|��||�r^t�| �|��||�rtt�d S )Nr8 T)r� r r� )r7 r r@ r� r� r3 r� )r r4 r* r+ r r r �test_axis_share_y� s z+TestDataFrameGroupByPlots.test_axis_share_yc C sz | j }|jd|jddd�\}}| �|��||�s4t�| �|��||�sJt�| �|��||�s`t�| �|��||�svt�d S )Nr8 T)r� r r� r� )r7 r r@ r� r� r3 r� )r r4 r* r+ r r r �test_axis_share_xy� s z,TestDataFrameGroupByPlots.test_axis_share_xyzhisttype, expected)rQ T)rR T)rS F)rT Tc C s> t tjjdddd�ddgd�}|jd|d�}| j||d � d S ) Nr r )r� r )r� ra r\ )r� )r rU )rV )r r. r/ rW r rX )r rU rY r4 r r r r r[ � s z0TestDataFrameGroupByPlots.test_histtype_argumentN)r r� r� r� r� r� r� r� r� r� r& r� r� r[ r r r r r� 7 s E 6 r� )�__doc__r� �numpyr. r& Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr r r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr r r� ZslowZ pytestmarkZskip_if_no_mplr r� r� r r r r �<module> s" U K
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка