Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/__pycache__/test_boxplot_method.cpython-37.pyc
Назад
B �5�g�S � @ s� d Z ddlZddlZddlZddlZddlm mZ ddl m Z m Z m Z mZmZ ddlmZ ddlmZmZ ddlmZ ejjZe jG dd� de��Z e jG dd� de��Z dS ) z Test cases for .boxplot method � N)� DataFrame� MultiIndex�Series� date_range�timedelta_range)� TestPlotBase�_check_plot_worksc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z e j �dddddd�ddddd�fddiddifdddddd�fg�dd� �Ze j �d d d d d d d�fd d!d!d"d!d�fg�d#d$� �Ze j �d%ddd&�d'fg�d(d)� �Ze j �d*d+d,d-d.g�d/d0� �Zd1S )2�TestDataFramePlotsc C sL t tj�dd�ttjd d� �ddddgd�}dd gd |d < dd dgd |d < t|jdd� t|jddgdd� t � t �� t|jddgd d� W d Q R X t|jdd d gd� t � t �� t|jd d� W d Q R X t � t �� t|jd d gd� W d Q R X tt j j|d dd� t|jddd� t � t �� t|jd dd� W d Q R X d S )N� � �one�two�three�four)�index�columns�foo�bar� Zindic� Zindic2�dict)� return_type)�columnr )r �by)r )�datar � )�notchr )r r )r �np�random�randn�list�string� ascii_lettersr �boxplot�tm�assert_produces_warning� UserWarning�plottingZ_core)�self�df� r* �_/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/test_boxplot_method.py�test_boxplot_legacy1 s& z'TestDataFramePlots.test_boxplot_legacy1c C sv t tj�dd�ddgd�}tddddddddddg �|d< tdgd �|d < t�t�� t|j dd � W d Q R X | j � � \}}|j dd|d �}|j }||ks�t �| j � � \}}|�d �j |d d �}|j }||d ks�t �| j � � \}}t�t�� |j ddgd|d d�}W d Q R X |d �� |k�s(t �| j � � \}}|j |dd �}ttj�|�� ��}t|�� �t|�k�srt �d S )N� r ZCol1ZCol2)r �A�B�X�Y)r )r �ax�axes)r2 r )r r r2 r r )r r r �randr r$ r% r& r r# �plt�subplotsr3 �AssertionError�groupbyZ get_figurer � itertools�chain� from_iterable�values�len� get_lines)r( r) �figr2 r3 Zax_axes�d�linesr* r* r+ �test_boxplot_legacy27 s, z'TestDataFramePlots.test_boxplot_legacy2c C s | j �� }t|| jj�s t�d S )N)�hist_dfr# � isinstancer5 ZAxesr7 )r( �resultr* r* r+ � test_boxplot_return_type_noneY s z0TestDataFramePlots.test_boxplot_return_type_nonec C s dd l }ttj�dd�ttjd d� �ddddgd�}d }tj t |d �� |j d d � W d Q R X |� � }| � |d � t �d�� |j dd �}W d Q R X | � |d� t �d�� |j d d �}W d Q R X | � |d � t �d�� |j dd �}W d Q R X | � |d� d S )Nr r r r r r r )r r z,return_type must be {'axes', 'dict', 'both'})�matchZ NOT_A_TYPE)r r3 Fr �both)Z matplotlibr r r r r r! r" �pytest�raises� ValueErrorr# �_check_box_return_typer$ r% )r( Zmplr) �msgrE r* r* r+ � test_boxplot_return_type_legacy^ s& z2TestDataFramePlots.test_boxplot_return_type_legacyc C s� dd� }| j �� }tj�dd|jd �|d< |jddgd d �\}}||d |� ||d |� |j|kslt�|jdddgd d �}|d |d |d }}}|d }||d |� ||d |� ||d |� |j|ks�t�|j|ks�t�|jd ks�t�d S )Nc S s0 |� � \}}|| �� ks t�|| �� ks,t�d S )N)Zget_ylim�minr7 �max)�colr2 Zy_minZy_maxr* r* r+ �_check_ax_limits{ s zETestDataFramePlots.test_boxplot_axis_limits.<locals>._check_ax_limitsr � r Zage�height�weight�category)r )r r )r r )r r )r r ) rC �copyr r �randint�shaper# Z_shareyr7 )r( rR r) Z height_axZ weight_ax�pZage_axZdummy_axr* r* r+ �test_boxplot_axis_limitsz s z+TestDataFramePlots.test_boxplot_axis_limitsc C s8 t tj�dd��}tj|jd d �df< t|jdd� d S )NrS r r r3 )r )r r r r �nan�locr r# )r( r) r* r* r+ �test_boxplot_empty_column� s z,TestDataFramePlots.test_boxplot_empty_columnc C sV t tj�dd�dddddgd�}|jd d d �}|jjjd ks@t�|jjj d ksRt�d S )Nr- � r. r/ �C�D�E)r r3 )� � )r Zfigsizerc rd ) r r r r4 r# �figureZ bbox_inches�widthr7 rT )r( r) rE r* r* r+ � test_figsize� s z TestDataFramePlots.test_figsizec C s6 t dddddddgi�}| j|jddd �ddd � d S ) N�ar r r r r_ r � )�fontsize)� xlabelsize� ylabelsize)r �_check_ticks_propsr# )r( r) r* r* r+ � test_fontsize� s z TestDataFramePlots.test_fontsizec C s� t tddd�tj�d�tj�d�d tddd��t�tdddd�tddd�d��}|jd d �}d d � |� � D �d dgks|t �d S )Nz 2012-01-01�d )�periodsr �UTC)rp �tzz1 days)rh �b�cr@ �e�f�box)�kindc S s g | ] }|� � �qS r* )�get_text)�.0�xr* r* r+ � <listcomp>� s z@TestDataFramePlots.test_boxplot_numeric_data.<locals>.<listcomp>rs rt ) r r r r r Zastype�strr �plot�get_xticklabelsr7 )r( r) r2 r* r* r+ �test_boxplot_numeric_data� s z,TestDataFramePlots.test_boxplot_numeric_datazcolors_kwd, expected�rrs �grt )�boxes�whiskers�medians�capsr� c C sR t tj�dd��}|j|dd�}x,|�� D ] \}}|| d �� |ks*t�q*W d S )Nr- r r )�colorr r )r r r r4 r# �items� get_colorr7 )r( Z colors_kwd�expectedr) rE �k�vr* r* r+ �test_color_kwd� s z!TestDataFramePlots.test_color_kwdzscheme,expectedZdark_backgroundz#8dd3c7z#bfbbd9�defaultz#1f77b4z#2ca02cc C sj t tj�dd��}dd lm} |j�|� |jj dd�}x,|� � D ] \}}|| d � � |ksBt �qBW d S )Nr- r r r )r ) r r r r4 Zmatplotlib.pyplotZpyplot�styleZuser~ rw r� r� r7 )r( �schemer� r) r5 rE r� r� r* r* r+ �test_colors_in_theme� s z'TestDataFramePlots.test_colors_in_themezdict_colors, msg)r� Z invalid_keyzinvalid key 'invalid_key'c C s> t tj�dd��}tjt|d�� |j|dd� W d Q R X d S )Nr- r )rG r )r� r )r r r r4 rI rJ rK r# )r( Z dict_colorsrM r) r* r* r+ �test_color_kwd_errors� s z(TestDataFramePlots.test_color_kwd_errorszprops, expected)Zboxpropsr� )Z whiskerpropsr� )Zcappropsr� )Z medianpropsr� c C sN t dd� dD ��}|ddii}|jf ddi|��}|| d �� dksJt�d S ) Nc S s i | ]}t j�d �|�qS )ro )r r )rz r� r* r* r+ � <dictcomp>� s z?TestDataFramePlots.test_specified_props_kwd.<locals>.<dictcomp>�ABCr� ZC1r r r )r r# r� r7 )r( �propsr� r) ZkwdrE r* r* r+ �test_specified_props_kwd� s z+TestDataFramePlots.test_specified_props_kwdN)�__name__� __module__� __qualname__r, rB rF rN r[ r^ rg rn r� rI �mark� parametrizer� r� r� r� r* r* r* r+ r sH " r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e j � dddddddgfdgdddddgfddddd d gfddgdddddddd dd g fd dddddddd dd g fg�d d � �Zd!d"� Zd S )#�TestDataFrameGroupByPlotsc C sl | j jdd�}t�t�� t|jdd�}W d Q R X | jt|j �ddd� t|jddd �}| j|d d d� d S ) N�gender)r r3 )r r )r r )�axes_num�layoutF)r6 r r )r r ) rC r8 r$ r% r& r r# �_check_axes_shaper r< )r( �groupedr3 r* r* r+ r, s z.TestDataFrameGroupByPlots.test_boxplot_legacy1c C s� t tjd d� td��}ttj�dd�t� |�d�}|j dd�}t � t �� t|jdd�}W d Q R X | jt|j�ddd � t|jd dd �}| j|dd d � d S ) Nr- r )r r )�levelr3 )r )r r )r� r� F)r6 r )r r )�zipr! r" �ranger r r r4 r � from_tuplesr8 r$ r% r& r r# r� r r< )r( �tuplesr) r� r3 r* r* r+ rB s z.TestDataFrameGroupByPlots.test_boxplot_legacy2c C s� t tjd d� td��}ttj�dd�t� |�d�}|j dd�j ddd�}t � t�� t|jdd �}W d Q R X | jt|j�dd d � t|jd dd �}| j|ddd � d S )Nr- r )r r )r� r )r� Zaxisr3 )r )r r )r� r� F)r6 r )r r )r� r! r" r� r r r r4 r r� Zunstackr8 r$ r% r& r r# r� r r< )r( r� r) r� r3 r* r* r+ �test_boxplot_legacy3 s z.TestDataFrameGroupByPlots.test_boxplot_legacy3c C s� d}t tjjdd|d��}t tjjdd|d��}t�d�� tjjddg|d�}W d Q R X t|||d ��}|�d �}|� � }t | j � � �d ks�t �t |�d ks�t �t�� |jd d �}t | j � � �dks�t �t |�d ks�t �t�� |�d ��� }t�� d S )Nr- � rS )�size�< �* ZmaleZfemale)rT rU r� r� r r3 )r r )r r r �normalr$ Z RNGContext�choicer r8 r~ r= r5 Z get_fignumsr7 �closer# �hist)r( �nrU rT r� r) �gb�resr* r* r+ �test_grouped_plot_fignums" s" z3TestDataFrameGroupByPlots.test_grouped_plot_fignumsc C s. | j }|jdd�}t|tj�s"t�| j|d dddgd� |�d�jdd�}| j|dd d gd� d �� }t tj � d d �|d�}d�� }|d |d< x�dD ]�}|�d�j|d�}| j||dddgd� |jd|d�}| j||dddgd� |�d�j|d�}| j|||d� |jd|d�}| j|||d� q�W d S )Nr� )r rT rU rV )Z expected_keysr )r ZMaleZFemalezX B C D A G Y N Q O�2 r- )r zA B C D E F G H I Jr_ )r r3 rH � classroomr. r/ r` )r r ) rC r# rD r Zndarrayr7 rL r8 �splitr r r )r( r) rE Zcolumns2Zdf2Z categories2�t�returnedr* r* r+ � test_grouped_box_return_type9 s* z6TestDataFrameGroupByPlots.test_grouped_box_return_typec C s� | j }d}tjt|d�� |jddg|jdd� W d Q R X d}tjt|d�� |jdddgd d d � W d Q R X d }tjt|d�� |jddg|jd d� W d Q R X t�t�� t |� d�jdd d�}W d Q R X | j | j � � jddd� t�t�� t |� d�jdd d�}W d Q R X | j | j � � jddd� t�t�� t |� d�jdd d�}W d Q R X | j | j � � jddd� |jdddgddd�}| j | j � � jddd� x<|d gD ].}| j|�� dd� | j|j�� gdd� �q�W x:|d |d gD ]&}| �|�� � | �|j�� g� �q�W |� d�jdddgd d�}| j | j � � jddd� t�t�� t |� d�jddd d �}W d Q R X | j | j � � jddd� t�t�� t |� d�jdd d d �}W d Q R X | j | j � � jddd� |jdddgdd d�}| j | j � � jdd d� |jdddgdd d�}| j | j � � jdd d� |� d�jdddgd d d �}| j | j � � jdd d� |� d�jdddgd!d d �}| j | j � � jdd"d� d S )#Nz1Layout of 1x1 must be larger than required size 2)rG rU rT )r r )r r r� z7The 'layout' keyword is not supported when 'by' is NonerV )r r r )r r� r z1At least one dimension of layout must be positive)���r� r� )r r r )r r )r� r� r )r r r� r r3 )r r r F)�visible)r r )r r� )r r )r� r )r r )r r )r r� )r r )rC rI rJ rK r# r� r$ r% r& r r8 r� r5 �gcfr3 Z_check_visibler ZxaxisZ get_label)r( r) rM rw r3 r2 r* r* r+ �test_grouped_box_layout[ s~ z1TestDataFrameGroupByPlots.test_grouped_box_layoutc C s� | j }t�t��F | j�dd�\}}|�d�jdd|d� | j| j� � j ddd� W d Q R X | j�dd �\}}t�t��" |jdd dgd d|d d �}W d Q R X t � t |j��}| j|d dd� t�||d � |d j|ks�t�t�t��& |�d�jdd dgd|d d�}W d Q R X t � t |j��}| j|d dd� t�||d � |d j|k�s\t�d}tjt|d��@ | j�dd �\}}t�t�� |�d�j|d�}W d Q R X W d Q R X d S )Nr rV rT r3 )r r r2 r )r r )r� r� r rU r� r )r r r r2 )r r r� r z@The number of passed axes must be 3, the same as the output plot)rG )r2 )rC r$ r% r& r5 r6 r8 r# r� r� r3 r �arrayr r< Zassert_numpy_array_equalre r7 rI rJ rK )r( r) r? r3 r� rM r* r* r+ � test_grouped_box_multiple_axes� s8 " z8TestDataFrameGroupByPlots.test_grouped_box_multiple_axesc C sF t ddddddgddddddgd��}| j|jd d d d �d d d � d S )Nr r r r r_ r r )rh rs rh rs ri )r rj )rk rl )r rm r# )r( r) r* r* r+ rn � s &z'TestDataFrameGroupByPlots.test_fontsizezcol, expected_xticklabelr� z(a, v)z(b, v)z(c, v)z(d, v)z(e, v)�v1z(a, v1)z(b, v1)z(c, v1)z(d, v1)z(e, v1)Nc C sl t tj�td�d�tj�d�tj�d�d��}|�d�}t|jd|dd�}dd � |� � D �}||ksht �d S ) NZabcdero )�catr� r� r� Fr3 )r6 r r c S s g | ] }|� � �qS r* )ry )rz r{ r* r* r+ r| s zQTestDataFrameGroupByPlots.test_groupby_boxplot_subplots_false.<locals>.<listcomp>) r r r r� r r4 r8 r r# r r7 )r( rQ �expected_xticklabelr) r� r3 �result_xticklabelr* r* r+ �#test_groupby_boxplot_subplots_false� s ( z=TestDataFrameGroupByPlots.test_groupby_boxplot_subplots_falsec C s� ddddddddgddddddddgg}t t|� �}tj|ddgd �}ttj�d d �d d dg|d�}ddg}t|j |dd�}ddg}dd� |� � D �}||ks�t �d S )Nr Zbazr Zquxr r �first�second)�namesr rd r. r/ r` )r r )r r )r r r3 )r r z (bar, one)z (bar, two)c S s g | ] }|� � �qS r* )ry )rz r{ r* r* r+ r| 0 s zLTestDataFrameGroupByPlots.test_boxplot_multiindex_column.<locals>.<listcomp>) r r� r r� r r r r r r# r r7 ) r( Zarraysr� r r) rQ r3 r� r� r* r* r+ � test_boxplot_multiindex_column" s z8TestDataFrameGroupByPlots.test_boxplot_multiindex_column)r� r� r� r, rB r� r� r� r� r� rn rI r� r� r� r� r* r* r* r+ r� sH "],r� ) �__doc__r9 r! �numpyr rI Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r r Zpandas._testingZ_testingr$ Z pandas.tests.plotting.commonr r Zpandas.plottingr' r� ZslowZ pytestmarkZskip_if_no_mplr r� r* r* r* r+ �<module> s f
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка