Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/plotting/__pycache__/test_series.cpython-37.pyc
Назад
B �5�g�t � @ s� d Z ddlmZ ddlmZ ddlZddlZddlm m Z ddl Z ddl m Z mZmZ ddlmZ ddlmZmZ ddlmZ ejjZe jG dd� de��Z dS ) z Test cases for Series.plot � )�datetime)�chainN)� DataFrame�Series� date_range)� TestPlotBase�_check_plot_worksc @ sp e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zej�d)d*d+g�d,d-� �Zd.d/� Ze j d0d1� �Z e j d2d3� �Z d4d5� Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$d>d?� Z%d@dA� Z&dBdC� Z'dDdE� Z(dFdG� Z)dHdI� Z*dJdK� Z+dLdM� Z,dNdO� Z-dPdQ� Z.dRdS� Z/dTdU� Z0dVdW� Z1dXdY� Z2ejj3dZd[� �Z4ej�d\d]d^g�d_d`� �Z5dadb� Z6dcdd� Z7ej�dedfdgdhdigdhdgdfdjgfdkdldmdngdodpdqdrgfg�dsdt� �Z8dudv� Z9ej�dwdxdydzg�ej�d\d]d^d{g�d|d}� ��Z:ej�d~e;j<djdgdd��e;j=dfd��e;j=dgd��gg�d�d�� �Z>d�S )��TestSeriesPlotsc C sV t �| |� dd l}|�� t�� | _d| j_t�� | _ d| j _t� � | _ d| j _d S )Nr �ts�series�iseries) r � setup_method� matplotlibZ rcdefaults�tmZmakeTimeSeriesr �nameZmakeStringSeriesr ZmakePeriodSeriesr )�self�method�mpl� r �W/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/plotting/test_series.pyr s z TestSeriesPlots.setup_methodc C s� t | jjdd� t | jjdd� t | jjdd�}| j|dd� t | jjdd d �}| j|d d � t | jjdd d �}| j|d d� t | jjdd d�}| j|d d d� t | jd d� jj� t | jjjdd� t | jj� x$dD ] }t | jd d� j|d� q�W t | jd d� jj � t t t j � d��jjdd�}| j|jd gdgd� t | jjd d�}| j|ddd � t | jjd d d �}| j|ddd � t | jjd d d �}| j|ddd � d S ) NZfoo)�labelF)� use_indexr )�rot)�xrot�.T)�style�logy�log)�yaxis)r Zlogx)�xaxis)r Zloglog)r r � )�stacked)�line�bar�barh�kde�hist�box� )�kind�black)�color)� facecolors)�subplots� )r. r. )�axes_num�layout)���r. )r- r0 )r. r1 )r r �plot�_check_ticks_props�_check_ax_scalesr# �arear r r$ r �np�random�randn� _check_colors�patches�_check_axes_shape)r �axes�axr) r r r � test_plot. s0 zTestSeriesPlots.test_plotc C sD | j �� \}}| jjdd|d�}| �|jd� | j|dddd� d S )NZTest)� � )�title�figsizer= r. )r. r. )r/ r0 rB )�pltr- r r2 �_check_text_labelsrA r; )r �_r= r r r �test_plot_figsize_and_titleQ s z+TestSeriesPlots.test_plot_figsize_and_titlec C sL d}| j j| }| j �� \}}tdddg�j|d� || j j| ksHt�d S )Nzaxes.prop_cycler. � � )r= )rC �rcParamsr- r r2 �AssertionError)r �key�colorsrE r= r r r �test_dont_modify_rcParamsX s z)TestSeriesPlots.test_dont_modify_rcParamsc C s� | j �� \}}| jj|d�}|�� \}}|�� }||d jdd�d d ksPt�||d jdd�d d kspt�t� � | jjd|d�}|�� \}}|�� }||d jdd�d d ks�t�||d jdd�d d ks�t�d S )N)r= r F)�origr1 T)� secondary_yr= ) rC r- r r2 �get_xlim� get_lines�get_datarJ r �close)r �figr= �xmin�xmax�linesr r r �test_ts_line_lim` s z TestSeriesPlots.test_ts_line_limc C s | j �� \}}| jjjd|d�}|�� \}}|�� d jdd�d }||d ksTt�||d ksdt�| j |dd� t � � | j �� \}}| jjjdd|d�}|�� \}}|�� d jdd�d }||d ks�t�||d ks�t�| j |d d� t � � | j� � }|� d ��d �|_| j �� \}}|jjdd|d�}|�� \}}|�� d jdd�d }||d k�sht�||d k�szt�| j |dd� t � � | j �� \}}|jjdd|d �}|�� \}}|�� d jdd�d }||d k�s�t�||d k�s�t�| j |dd� d S ) NF)r! r= r )rN r1 )r T)r! Zx_compatr= � �GMTZCET)r! rO r= )rC r- r r2 r5 rP rQ rR rJ r3 r rS �copyZ tz_localizeZ tz_convert�index)r rE r= rU rV r" Ztz_tsr r r �test_ts_area_limo sB z TestSeriesPlots.test_ts_area_limc C sr | j jdddd�\}\}}t| j�j|dd� t| j�j|dd� | �|��||�sXt�| �|��||�snt�d S )Nr. rG T)Zshareyr5 )r= r) )rC r- �absr r2 Z get_y_axisZjoinedrJ )r rT �ax1�ax2r r r � test_area_sharey_dont_overwrite� s z/TestSeriesPlots.test_area_sharey_dont_overwritec C s: t ddg�}| j�� \}}|jdd|d�}| j|dgd� | j�� | j�� \}}|jd|d�}| j|dgd� | j�� d |_| j�� \}}|jd|d�}| j|d gd� | j�� | j�� \}}|jdd|d �}| j|dgd� | j�� | j�� \}}|jd d|d �}|�� d k�s t�|� � | j|dgd� d S ) Nr. rG ZLABELT)r �legendr= )�labels)rb r= �None�NAME)rb r r= F) r rC r- r2 �_check_legend_labelsrS r Z get_legendrJ rb )r �srE r= r r r � test_label� s. zTestSeriesPlots.test_labelc C sH t dddg�}t|jdd� d}tjt|d�� t|j� W d Q R X d S )NFT)Z include_boolzno numeric data to plot)�match)r r r2 �pytest�raises� TypeError)r rg �msgr r r � test_boolean� s z TestSeriesPlots.test_booleanc C s$ ddt jdg}t|�}t|tjdd�d�}x�||gD ]�}t|j�}|jd �� }t j dddgt j d�}t� t � |j d�|� t� |jt � d d d d g�� t j ddddgt j d�}t|jd d �}t� |jd �� |� t|jj�}t� |jd �� |� t|jjd d �}t� |jd �� |� q4W d S ) Nr. rG rH � )�k)r\ r )�dtypeFT)r! )r6 �nanr r Z makeDateIndexr r2 rW � get_ydata�arrayZfloat64�assert_numpy_array_equal�delete�data�maskr5 ) r �valuesrg r �dr= Zmasked�exp�expectedr r r �test_line_area_nan_series� s" z)TestSeriesPlots.test_line_area_nan_seriesc C s� t dddgdddgd�}d|j_| j�� \}}|jd |d �}|�� }|d ksPt�| j�� \}}|jjd |d �}|�� }|d ks�t�d S ) Nr. rG rH �a�b�c)r\ z The IndexF)r r= � ) r r\ r rC r- r2 � get_xlabelrJ r# )r rg rE r= r r` �label2r r r �test_line_use_index_false� s z)TestSeriesPlots.test_line_use_index_falsec C s� t �ddddddg�}| j�� \}}tddg�jjd |d �}t�|j � � |� t� � | j�� \}}tddg�jj d |d �}t�|j � � |� t� � t �d d d ddddg�}| j�� \}}tddd g�jd d|d�}d}d}|�� }t�|d |� t�|d |� t�|j � � |� t� � | j�� \}}tddd g�jd d|d�}|�� }t�|d |� t�|d |� t�|j � � |� d S )Ng�������?g �?g $@g Y@g @�@g ��@�� i� T)r r= g�h㈵��>g-C ��6?g����MbP?g{�G�z�?r# )r r) r= g!�,�NJ?gЄ�-? �?r r. r$ )r6 rt rC r- r r2 r# r ru r � get_ticklocsrS r$ r Zget_ylimZassert_almost_equalrP )r r| rE r= Zymin�ymax�resr r r � test_bar_log� s2 z TestSeriesPlots.test_bar_logc C sV t ddddgddddgd �}| j�� \}}|jjd |d �}| �|�� d d ddg� d S )Nr. rG rH ro r~ r r� rz )r\ F)r r= �0�1�2�3)r rC r- r2 r# rD �get_xticklabels)r �dfrE r= r r r �test_bar_ignore_index s z%TestSeriesPlots.test_bar_ignore_indexc C sR t ddddg�}|jjddddgd�}dd � |jD �}d d d d g}||ksNt�d S ) Nr. rG rH ro �red�blue)r+ c S s g | ] }|� � �qS r )Z get_facecolor)�.0�pr r r � <listcomp> s z8TestSeriesPlots.test_bar_user_colors.<locals>.<listcomp>)g �?g g g �?)g g g �?g �?)r r2 r# r: rJ )r rg r= �resultr| r r r �test_bar_user_colors s z$TestSeriesPlots.test_bar_user_colorsc C sh t tj�dd��}| j�� \}}|j|d�}| j|dd� | j�� \}}|jd|d�}| j|dd� d S )Nr( )r= r )r rY )r r= )r r6 r7 r8 rC r- r2 r3 )r r� rE r= r< r r r � test_rotation s z TestSeriesPlots.test_rotationc C s� ddl m} tdd�}|dddddd d d d g }ttj�t|��|�}| j� � \}}|j |d �}|� t ddd�d|�}|� dd� ||�� d ks�t�| j|dd� d S )Nr )�DatetimeConverterz1/1/2000z3/1/2000r. rG rH r( � r � � )r= i� r� z1/1/1999z1/1/2001rY )r )Z%pandas.plotting._matplotlib.converterr� r r r6 r7 r8 �lenrC r- r2 �convertr Zset_xlimrP rJ r3 )r r� �rng�serrE r= Zxpr r r �test_irregular_datetime& s z'TestSeriesPlots.test_irregular_datetimec C s� t ddtjddddgddddtjddgd�}| j�� \}}|j|d �}|�� \}}|�� }|t�|d j d d �d �ks|t �|t� |d j d d �d �ks�t �d S ) Ng g �?g @g @g @g @g @)r\ )r= r F)rN ) r r6 rr rC r- r2 rP rQ ZnanminrR rJ Znanmax)r r� rE r= rU rV rW r r r �test_unsorted_index_xlim3 s "z(TestSeriesPlots.test_unsorted_index_xlimc C s� t tj�dd�dddddgdd �}t|jj�}| �|j|j � |� � dksNt �t|jjd d �}| �|jd gd � d d dg}t|jj|d�}d d dd d g}| j |j |d� dddddg}d d dddg}t|jj||d�}| �|j|� | j |j |d� t|jj|ddd�}dd� |j|�� D �}tt�t|j |���}| �|j|� x"|jD ]}|�� dk�sNt ��qNW t dd d d d gdddddgd �}tjtd!d"�� |j�� W d Q R X t dd tjd gddddgdd �}t|jj�}| �|jddd dg� d S )#Nr. r( r~ r r� rz �eZYLABEL)r\ r )rc r� �r�g)rL )r, �A�B�C�D�E�m)rc rL z%.2f� )rL ZautopctZfontsizec S s g | ]}|d d��qS )�d z.2fr )r� rg r r r r� _ s z3TestSeriesPlots.test_pie_series.<locals>.<listcomp>rG r ro r1 )r\ z&pie plot doesn't allow negative values)ri )r r6 r7 �randintr r2 �pierD �textsr\ � get_ylabelrJ r9 r: ry �sum�listr � from_iterable�zipZ get_fontsizerj rk � ValueErrorrr ) r r r= Z color_argsZcolor_expectedrc ZpctsZexpected_texts�tr r r �test_pie_series? s: z TestSeriesPlots.test_pie_seriesc C s\ t dtjddg�}| j�� \}}|jjd|d�}ddddg}dd � |jD �}||ksXt�d S ) Nr. T)rb r= r� r� r� r� c S s g | ] }|� � �qS r )Zget_text)r� �xr r r r� t s z0TestSeriesPlots.test_pie_nan.<locals>.<listcomp>) r r6 rr rC r- r2 r� r� rJ )r rg rE r= r| r� r r r � test_pie_nano s z TestSeriesPlots.test_pie_nanc C sf t tj�dd�td�d�}ttj�d�dd�}| j�� \}}|j|d�}|jdd|d � | j |d d d d gd� |� � � � s~t �|j � � � � s�t �t�� | j�� \}}|j|d�}|j|ddd� | j |d d d d gd� |� � � � s�t �|j � � � � s�t �t�� | j�� \}}|jd|d�}|jdd|d � dddd g}| j |j|d� |j� � � � �r^t �|� � � � �spt �t�� | j�� \}}|jd|d�}|j|ddd� dddd g}| � |j|� |j� � � � �r�t �|� � � � �s�t �t�� | j�� \}}|jdd|d�}|j|ddd� d d d d g}| � |j|� |j� � � � �rHt �|� � � � �sZt �t�� d S )NrY rH �abc)�columnsr� )r )r= T)rb rO r= r~ r r� z x (right))rc )r= rb rO )rO r= z a (right)z b (right)z c (right)F)rO Z mark_rightr= )r r6 r7 r8 r� r rC r- r2 rf Z get_yaxisZ get_visiblerJ Zright_axr rS Zleft_ax)r r� rg rE r= r| r r r � test_df_series_secondary_legendw sP z/TestSeriesPlots.test_df_series_secondary_legendzinput_logy, expected_scale)Tr )�symZsymlogc C s^ t tj�d��}t tj�d��}|j|d�}|jd|d�}|�� |ksJt�|�� |ksZt�d S )NrY )r T)rO r )r r6 r7 r8 r2 Z get_yscalerJ )r Z input_logyZexpected_scale�s1�s2r_ r` r r r �test_secondary_logy� s z#TestSeriesPlots.test_secondary_logyc C sP t tj�d��}| j�� \}}d}tjt|d�� |j dd|d� W d Q R X d S )NrG z�Cannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbol)ri zk--rp )r r+ r= ) r r6 r7 r8 rC r- rj rk r� r2 )r r� rE r= rm r r r �)test_plot_fails_with_dupe_color_and_style� s z9TestSeriesPlots.test_plot_fails_with_dupe_color_and_stylec C s� t �ddd�}t| jjjddd� t| jjjd dd� t| jjjd t �d�d� t| jjjd|d� t| jjjd|d� | j� � \}}| jjjdd||d�}| j |d d � | � |j � � d � d S ) Ni����r� � Zscott)� bw_method�indg �?T)r r� r� r= r )r ZDensity)r6 Zlinspacer r r2 r% �int_ZdensityrC r- r4 rD r Z get_label)r Z sample_pointsrE r= r r r �test_kde_kwargs� s z TestSeriesPlots.test_kde_kwargsc C sJ t tjjdd��}tj|d< t|jj�}tt� |j d � � � �sFt �d S )N�2 )�sizer ) r r6 r7 �uniformrr r r2 r% �any�isnanrW � get_xdatarJ )r rg r< r r r �test_kde_missing_vals� s z%TestSeriesPlots.test_kde_missing_valsc C sj | j �� \}}| jjjd|d�}| j|dd� |�� }| �|| jjg� |� � }| �|dgt |� � d S )NT)r r= r )r r� ) rC r- r r2 r' r4 r� rD r Zget_yticklabelsr� )r rE r= ZxlabelsZylabelsr r r �test_boxplot_series� s z#TestSeriesPlots.test_boxplot_seriesc C sz t td��}tjjtjj }xX|D ]P}| j�� \}}|j||d� | j� � | j�� \}}t |j|�� | j� � q"W d S )NrH )r) r= ) r �range�plotting� PlotAccessor� _common_kinds� _series_kindsrC r- r2 rS �getattr)r rg �kindsr) rE r= r r r �test_kind_both_ways� s z#TestSeriesPlots.test_kind_both_waysc C s\ t td��}| j�� \}}x<tjjD ]0}d}tjt |d�� |j ||d� W d Q R X q$W d S )N�abcdzno numeric data to plot)ri )r) r= ) r r� rC r- r� r� r� rj rk rl r2 )r rg rE r= r) rm r r r �test_invalid_plot_data� s z&TestSeriesPlots.test_invalid_plot_datac C s4 t td�td�}x tjjD ]}t|j|d� qW d S )Nr )rq )r) )r r� �objectr� r� r� r r2 )r rg r) r r r �test_valid_object_plot s z&TestSeriesPlots.test_valid_object_plotc C s` t ddddg�}| j�� \}}x<tjjD ]0}d}tjt|d�� |j ||d� W d Q R X q(W d S )Nr~ r g �?rG zno numeric data to plot)ri )r) r= ) r rC r- r� r� r� rj rk rl r2 )r rg rE r= r) rm r r r � test_partially_invalid_plot_data s z0TestSeriesPlots.test_partially_invalid_plot_datac C s6 t ddg�}tjtdd�� |jdd� W d Q R X d S )Nr. rG z%invalid_kind is not a valid plot kind)ri Z invalid_kind)r) )r rj rk r� r2 )r rg r r r �test_invalid_kind s z!TestSeriesPlots.test_invalid_kindc C sJ t ddd�}t ddd�}|�|�}tj�|j�}t||d�}t|j� d S )Nz1/1/2009ro )�periodsz1/2/2009)r\ ) r �appendr6 r7 r8 r� r r r2 )r Zdr1Zdr2r\ ry rg r r r � test_dup_datetime_index_plot s z,TestSeriesPlots.test_dup_datetime_index_plotc C s� t t�d�dd�}tj�dd�}|j||d�}t�dd� |jd �� D ��}|j t� d dg� |� � � d d� }t �||� d t|�� d �}tjt|d �� |jtj�dd �d� W d Q R X t �� d S )Nr r� )r rG )�yerr�xerrc S s g | ]}|j d d �df �qS )Nr. )Zvertices)r� �ir r r r� ' s z>TestSeriesPlots.test_errorbar_asymmetrical.<locals>.<listcomp>r. r1 z?Asymmetrical error bars should be provided with the shape \(2, z\))ri r� )r� )r r6 �aranger7 Zrandr2 Zvstack� collections� get_paths�Trt Zto_numpyZreshaper ru r� rj rk r� rS )r rg �errr= r� r| rm r r r �test_errorbar_asymmetrical s $ z*TestSeriesPlots.test_errorbar_asymmetricalc C s t t�d�dd�}tj�d�}ttj�dd�|jddgd�}ddg}x�|D ]�}t|jt |�|d �}| j |d d d � t|j||d �}| j |d d d � t|j|� � |d �}| j |d d d � t|j||d �}| j |d d d � t|jd d |d�}| j |d d d � qJW t|j|d�}| j |d d d � t dddd�}t t�d�|dd�}t tj�d�|d�} ttj�dd�|ddgd�} t|j| d�}| j |d d d � t|j| d�}| j |d d d � t � t�� |jt�d�d� W d Q R X dgd }t � t�� |j|d� W d Q R X d S )Nr r� )r rG �y)r\ r� r" r# )r� r) r r. )r� r� g�������?)r� r� r) )r� z1/1/2000z1/1/2001�M)�freqr� )r\ r )r\ )r� r� �zzz)r r6 r� r7 r8 r r\ r r2 Z_check_has_errorbars�tolistr r Zexternal_error_raisedr� rl ) r rg Zs_errZd_errr� r) r= Zixr Zts_errZtd_errr r r �test_errorbar_plot4 s< z"TestSeriesPlots.test_errorbar_plotc C s& t | jjdd� t | jj| jd� d S )NT)�table)r r r2 )r r r r � test_table] s zTestSeriesPlots.test_tablec C s& | � tdddg�tjjtjj � d S )Nr. rG rH )Z_check_grid_settingsr r� r� r� r� )r r r r �test_series_grid_settingsa s z)TestSeriesPlots.test_series_grid_settingsc C s� ddl m} x�dD ]x}|d|d�}||gks0t�|d|gd�}||gksLt�|d|d�}||gd ksjt�|d|gd�}||gd kst�qW d S )Nr )�get_standard_colors)r� r� �greenz#FF0000r. )r+ rH )�!pandas.plotting._matplotlib.styler� rJ )r r� r� r� r r r �test_standard_colorsh s z$TestSeriesPlots.test_standard_colorsc C s& dd l m} ddlm} x�|jD ]x}|d|d�}||gks>t�|d|gd�}||gksZt�|d|d�}||gd ksxt�|d|gd�}||gd ks t�q W x�|jjD ]x}|d|d�}||gks�t�|d|gd�}||gks�t�|d|d�}||gd ks�t�|d|gd�}||gd ks�t�q�W d S )Nr )r� r. )Z num_colorsr+ rH )Zmatplotlib.colorsrL r� r� ZcnamesrJ ZColorConverter)r rL r� r� r� r r r �test_standard_colors_allx s( z(TestSeriesPlots.test_standard_colors_allc C sB | j �� \}}tt�d�d �jd|d�}| j|�� dgd� d S )Nr� r. r� )r+ r= )� linecolors)rC r- r r6 r� r2 r9 rQ )r rE r= r r r � test_series_plot_color_kwargs� s z-TestSeriesPlots.test_series_plot_color_kwargsc C sN | j �� \}}tt�d�d tddd�d�jd|d�}| j|�� dgd� d S ) Nr� r. z1/1/2000)r� )r\ r� )r+ r= )r� ) rC r- r r6 r� r r2 r9 rQ )r rE r= r r r �"test_time_series_plot_color_kwargs� s z2TestSeriesPlots.test_time_series_plot_color_kwargsc C s� dd l }| �|j�}tddd�}tt�dd�|d�}d}| j�� \}}x t |�D ]}|j |d �}qPW | j |� � |d |� d � d S ) Nr z1/1/2000r� )r� r. � )r\ rH )r= )r� ) r Z_unpack_cyclerrI r r r6 r� rC r- r� r2 r9 rQ ) r r Z def_colorsr\ rg ZncolorsrE r= r� r r r �-test_time_series_plot_color_with_empty_kwargs� s z=TestSeriesPlots.test_time_series_plot_color_with_empty_kwargsc C sf t t�d�dd� td�D �d�}| j�� \}}|jddddg|d �}d d� d D �}| �|�� |� d S ) Nr c S s g | ]}d |d���qS )�P�02dr )r� r� r r r r� � s z4TestSeriesPlots.test_xticklabels.<locals>.<listcomp>)r\ r rH r( r� )�xticksr= c S s g | ]}d |d���qS )r� r� r )r� r� r r r r� � s )r rH r( r� ) r r6 r� r� rC r- r2 rD r� )r rg rE r= r{ r r r �test_xticklabels� s z TestSeriesPlots.test_xticklabelsc C s^ t td�dd� td�D �d�}|jjtddd�d�}t�ttddd���}t�||� � � d S ) Nr c S s g | ]}d |d���qS )r� r� r )r� r� r r r r� � s z6TestSeriesPlots.test_xtick_barPlot.<locals>.<listcomp>)r\ r r� rG )r� ) r r� r2 r# r6 rt r� r ru Z get_xticks)r rg r= r{ r r r �test_xtick_barPlot� s z"TestSeriesPlots.test_xtick_barPlotc C s@ ddl m} ttdd�tjdd|dgd�d �d �}t|j� d S ) Nr )�CustomBusinessDayr� �y z 2014-05-01z 2014-06-01z 2014-05-26)Zholidays)�start�endr� )r\ )Zpandas.tseries.offsetsr r r� �pdZ bdate_ranger r2 )r r rg r r r � test_custom_business_day_freq� s z-TestSeriesPlots.test_custom_business_day_freqc C sn t ddddg�}| j�� \}}|j|d�}|j�� }|jddgdd� | j�� \}}|j�� }t�||� d S ) Nr. rG rH ro )r= r T)Zinplace) r rC r- r2 r r� Zdropr ru )r rg rE r= �before�afterr r r �%test_plot_accessor_updates_on_inplace� s z5TestSeriesPlots.test_plot_accessor_updates_on_inplacer) r" r5 c C sT t ddg�}| j�� \}}|j||d� |�� }|d dk s@t�|d dksPt�d S )NrG rH )r) r= r r. )r rC r- r2 rP rJ )r r) rg rE r= �xlimsr r r �test_plot_xlim_for_series� s z)TestSeriesPlots.test_plot_xlim_for_seriesc C sh t td�}|jst�|�� }t|�� �dks0t�|�� d }t|�� �dksPt�t|�� �dksdt�d S )N)rq r. r ) r �int�emptyrJ r2 r� rQ r� rs )r r� r= r" r r r �test_plot_no_rows� s z!TestSeriesPlots.test_plot_no_rowsc C s4 t dddg�}tjtdd�� |�� W d Q R X d S )Nr~ r r� zno numeric data to plot)ri )r rj rk rl r2 )r r� r r r �test_plot_no_numeric_data� s z)TestSeriesPlots.test_plot_no_numeric_dataz data, indexr. rG rH ro r r r� r� rY iv i� i� i� c C sL t ||d�}|jdd�}|�� }dd� t|jdd� d�D �}||ksHt�d S ) N)rw r\ r# )r) c S s g | ]}|� � j�qS r )�get_bboxr� )r� �patchr r r r� s z3TestSeriesPlots.test_plot_order.<locals>.<listcomp>c S s | � � jS )N)r rV )r r r r �<lambda> � z1TestSeriesPlots.test_plot_order.<locals>.<lambda>)rK )r r2 r� �sortedr: rJ )r rw r\ r� r= r| r� r r r �test_plot_order� s z TestSeriesPlots.test_plot_orderc C s4 t ddg�}|jddd�}|jd �� dks0t�d S )Nr. rG rg ZC3)r r+ r )r r2 rW � get_colorrJ )r rg r= r r r �test_style_single_ok s z$TestSeriesPlots.test_style_single_okz index_name, old_label, new_label)Nr� �new)�oldr r )Nr� r� r# c C sx t ddddg�}||j_|j|d�}|�� dks4t�|�� |ksDt�|j|||d�}|�� |ksdt�|�� |kstt�d S )Nr. rG rH ro )r) r� )r) ZylabelZxlabel)r r\ r r2 r� rJ r� )r r) Z index_nameZ old_labelZ new_labelr� r= r r r �test_xlabel_ylabel_series s z)TestSeriesPlots.test_xlabel_ylabel_seriesr\ r� )r r� r� )�daysc C s0 d}t ddg|d�j|d�}|�� dks,t�d S )N)rH r. r. rG )r\ )Zxlim)r r2 rP rJ )r r\ r r= r r r �test_timedelta_index& s z$TestSeriesPlots.test_timedelta_indexN)?�__name__� __module__� __qualname__r r> rF rM rX r] ra rh rn r} r� r� r� r� r� r� r� r� r� r� rj �markZ parametrizer� r� �tdZskip_if_no_scipyr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r Zxfailr r r r r r r r Ztimedelta_rangeZ Timedeltar r r r r r sz #' ! 0? )! $ r ) �__doc__r � itertoolsr �numpyr6 rj Z pandas.util._test_decorators�utilZ_test_decoratorsr" Zpandasr r r r Zpandas._testingZ_testingr Z pandas.tests.plotting.commonr r Zpandas.plottingr� r! ZslowZ pytestmarkZskip_if_no_mplr r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка