o
    1فi                     @   s   d Z ddlmZ ddlmZ ddlZddlZddlmZ ddl	m
Z
 ddlm  mZ ddlZddlmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$ e%d	Z&e%d
Z'ej(dd Z)ej(dd Z*G dd dZ+dS )z Test cases for Series.plot     )datetime)chainN)is_platform_linux)np_version_gte1p24)	DataFrameSeries
date_rangeperiod_rangeplotting)_check_ax_scales_check_axes_shape_check_colors_check_grid_settings_check_has_errorbars_check_legend_labels_check_plot_works_check_text_labels_check_ticks_props_unpack_cycler
get_y_axis
matplotlibzmatplotlib.pyplotc                   C   s"   t tjdtjdtdddddS )N
   dtype
2020-01-01periodstsindexname)r   nparangefloat64r    r$   r$   b/home/test-dt/pcm-url-check/venv/lib/python3.10/site-packages/pandas/tests/plotting/test_series.pyr   '   s
   
r   c                   C   s$   t tdtjddd tdD dS )N   seriesc                 S   s   g | ]}d | qS )i_r$   .0ir$   r$   r%   
<listcomp>3   s    zseries.<locals>.<listcomp>)r   r    r   )r   ranger!   r#   r$   r$   r$   r%   r'   0   s   r'   c                   @   s  e Zd Zejjejdddiddigdd Zejjdd	 Zejjejd
ddiddigddiddigddidddggdd Z	ejjdd Z
ejjdd Zdd Zejddddejded d!d"d#gd$d% Zd&d' Zd(d) Zejdi d*d+id*d,igd-d. Zd/d0 Zd1d2 Zejdi d3dig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!ejdLdMe"dNdOdPgdQdR Z#dSdT Z$dUdV Z%ejj&e'oe( dWddXejdYdZd[gd\d] Z)ejj&e'o
e( dWddXejd^g d_g d`gdadb Z*dcdd Z+dedf Z,dgdh Z-didj Z.dkdl Z/dmdn Z0dodp Z1dqdr Z2dsdt Z3dudv Z4dwdx Z5dydz Z6d{d| Z7d}d~ Z8dd Z9dd Z:dd Z;dd Z<dd Z=ejdddgdd Z>dd Z?ejdddgdMdgdMe@Adgde@Bdddggdd ZCdd ZDdd ZEdd ZFejj&dddd ZGejdeHjIjJeHjIjK dd ZLejdeHjIjJeHjIjK dd ZMejdeHjIjJdd ZNejdeHjIjJdd ZOejdeHjIjJdd ZPdd ZQdd ZRdd ZSejjejdddgejdeTe@Ue@jVWdXde@Ue@jVWdXdeYe@Ue@jVWdXdeZe@Ue@jVWdXdddgdgdd Z[ejjdd Z\ejjejdeTe@Ue@jVWdXdeZe@Ue@jVWdXdddgdgdd Z]ejjdd Z^ejjddĄ Z_ejjddƄ Z`ejjddȄ Zaejjddʄ Zbejdg d̢dd΄ ZcddЄ Zddd҄ ZeddԄ Zfddք Zgdd؄ Zhddڄ Zidd܄ Zjejj&dddd߄ Zkejdddgdd Zldd Zmdd Znejdg dg dfg dg dfgdd Zodd Zpejdg dejdg ddd ZqejdLerjsdddderjtdderjtddggdd Zudd Zvejjdd ZwdMS (   TestSeriesPlotskwargslabelfoo	use_indexFc                 C   s   t |jfi | d S Nr   plot)selfr   r/   r$   r$   r%   	test_plot8   s   zTestSeriesPlots.test_plotc                 C   s   t |jdd}t|dd d S )Nr   )rotxrot)r   r5   r   )r6   r   axesr$   r$   r%   test_plot_tick_props=   s   z$TestSeriesPlots.test_plot_tick_propszscale, exp_scalelogyTyaxisloglogxxaxisloglog)rA   r>   c                 C   s*   t |jfddi|}t|fi | d S )Nstyle.)r   r5   r   )r6   r   scale	exp_scaleaxr$   r$   r%   test_plot_scalesB   s   
z TestSeriesPlots.test_plot_scalesc                 C      t |d d jj d S Nr   )r   r5   barr6   r   r$   r$   r%   test_plot_ts_barO   s   z TestSeriesPlots.test_plot_ts_barc                 C   s   t |jjdd d S )NFstacked)r   r5   arearL   r$   r$   r%   test_plot_ts_area_stackedS   s   z)TestSeriesPlots.test_plot_ts_area_stackedc                 C   s$   t tdtddd}t|j d S )N   r   r   )r   r-   r	   r   r5   )r6   serr$   r$   r%   test_plot_iseriesW   s   z!TestSeriesPlots.test_plot_iserieskindlinerK   barhkdescipy)markshistboxc                 C   s   t |d d j|d d S )NrR   rU   r4   )r6   r'   rU   r$   r$   r%   test_plot_series_kinds[   s   z&TestSeriesPlots.test_plot_series_kindsc                 C   rI   rJ   )r   r5   rW   r6   r'   r$   r$   r%   test_plot_series_barhi   s   z%TestSeriesPlots.test_plot_series_barhc                 C   s<   t ttjddjjdd}t|j	d gdgd d S )N   r   blackcolorr   
facecolors)
r   r   r!   randomdefault_rngstandard_normalr5   rK   r   patches)r6   rG   r$   r$   r%   test_plot_series_bar_axl   s   z'TestSeriesPlots.test_plot_series_bar_axlayout)   )rn   rm   c                 C   s(   t |jfddi|}t|ddd d S )NsubplotsTrn   rn   rn   )axes_numrl   )r   r5   r   )r6   r   r/   rG   r$   r$   r%   test_plot_6951r   s   zTestSeriesPlots.test_plot_6951c                 C   s>   t j \}}|jdd|d}t|jd t|dddd d S )NTest)      )titlefigsizerG   rn   rp   )rq   rl   rw   )mplpyplotro   r5   r   rv   r   )r6   r'   _rG   r$   r$   r%   test_plot_figsize_and_titlex   s   z+TestSeriesPlots.test_plot_figsize_and_titlec                 C   sJ   d}t jj| }t j \}}tg dj|d |t jj| ks#J d S )Nzaxes.prop_cyclern   ra      rG   )rx   ry   rcParamsro   r   r5   )r6   keycolorsrz   rG   r$   r$   r%   test_dont_modify_rcParams   s
   z)TestSeriesPlots.test_dont_modify_rcParamssecondary_yc                 C   sz   t j \}}|jdd|i|}| \}}| }||d jddd d ks+J ||d jddd d ks;J d S )NrG   r   Forigrm   r$   )rx   ry   ro   r5   get_xlim	get_linesget_data)r6   r   r/   rz   rG   xminxmaxlinesr$   r$   r%   test_ts_line_lim   s    $z TestSeriesPlots.test_ts_line_limc                 C   sr   t j \}}|jjd|d}| \}}| d jddd }||d ks)J ||d ks1J t|dd d S )NF)rO   rG   r   r   rm   r9   	rx   ry   ro   r5   rP   r   r   r   r   r6   r   rz   rG   r   r   rV   r$   r$   r%   test_ts_area_lim   s   z TestSeriesPlots.test_ts_area_limc                 C   st   t j \}}|jjdd|d}| \}}| d jddd }||d ks*J ||d ks2J t|dd d S )	NFTrO   x_compatrG   r   r   rm      r9   r   r   r$   r$   r%   test_ts_area_lim_xcompat   s   z(TestSeriesPlots.test_ts_area_lim_xcompatc                 C      |  }|dd|_tj \}}|jjdd|d}|	 \}}|
 d jddd }||d ks7J ||d ks?J t|dd	 d S )
NGMTCETFTr   r   r   rm   r9   copytz_localize
tz_convertr   rx   ry   ro   r5   rP   r   r   r   r   r6   r   tz_tsrz   rG   r   r   rV   r$   r$   r%   test_ts_tz_area_lim_xcompat      z+TestSeriesPlots.test_ts_tz_area_lim_xcompatc                 C   r   )
Nr   r   FT)rO   r   rG   r   r   rm   r9   r   r   r$   r$   r%   'test_ts_tz_area_lim_xcompat_secondary_y   r   z7TestSeriesPlots.test_ts_tz_area_lim_xcompat_secondary_yc                 C   st   t jjdddd\}\}}t|j|dd t|j|dd t|||s)J t|||s3J t| d S )Nrn   ra   T)shareyrP   )rG   rU   )	rx   ry   ro   absr5   r   joinedpltclose)r6   r   figax1ax2r$   r$   r%   test_area_sharey_dont_overwrite   s   z/TestSeriesPlots.test_area_sharey_dont_overwritec                 C   sH   t ddg}tj \}}|jdd|d}t|dgd tjd d S )Nrn   ra   LABELT)r0   legendrG   labelsallr   rx   ry   ro   r5   r   r   r6   srz   rG   r$   r$   r%   
test_label   s
   zTestSeriesPlots.test_labelc                 C   sF   t ddg}tj \}}|jd|d}t|dgd tjd d S )Nrn   ra   Tr   rG    r   r   r   r   r$   r$   r%   test_label_none   s
   zTestSeriesPlots.test_label_nonec                 C   sJ   t ddgdd}tj \}}|jd|d}t|dgd tjd d S )	Nrn   ra   NAMEr    Tr   r   r   r   r   r$   r$   r%   test_label_ser_name   s
   z#TestSeriesPlots.test_label_ser_namec                 C   sL   t ddgdd}tj \}}|jdd|d}t|dgd tjd	 d S )
Nrn   ra   r   r   Tr   r   r0   rG   r   r   r   r   r$   r$   r%   test_label_ser_name_override   s
   z,TestSeriesPlots.test_label_ser_name_overridec                 C   sd   t ddgdd}tj \}}|jdd|d}| d u sJ |  t|dgd tjd	 d S )
Nrn   ra   r   r   Fr   r   r   r   )	r   rx   ry   ro   r5   
get_legendr   r   r   r   r$   r$   r%   &test_label_ser_name_override_dont_draw   s   z6TestSeriesPlots.test_label_ser_name_override_dont_drawc                 C   s\   t g d}t|jdd d}tjt|d t|j W d    d S 1 s'w   Y  d S )N)FFTT)include_boolno numeric data to plotmatch)r   r   r5   pytestraises	TypeError)r6   r   msgr$   r$   r%   test_boolean   s   "zTestSeriesPlots.test_booleanr   Nr      r   c                 C   s   ddt jdg}t||d}t|j}|jd  }t jg dt jd}t	
t |jd| t	
|jt g d t jg d	t jd}t|jd
d}t	
|jd  | t|jj}t	
|jd  | t|jjdd}t	
|jd  | d S )Nrn   ra   r}   r   r   r|   r   )FFTF)rn   ra   r   r}   TrN   F)r!   nanr   r   r5   r   	get_ydataarrayr#   tmassert_numpy_array_equaldeletedatamaskrP   )r6   r   valuesdrG   maskedexpexpectedr$   r$   r%   test_line_area_nan_series   s   
z)TestSeriesPlots.test_line_area_nan_seriesc                 C   sP   t g dg dd}d|j_tj \}}|jd|d}| }|dks&J d S Nr|   abcr   z	The IndexFr2   rG   r   )r   r   r    rx   ry   ro   r5   
get_xlabel)r6   r   rz   rG   r0   r$   r$   r%   test_line_use_index_false  s   z)TestSeriesPlots.test_line_use_index_falsec                 C   sR   t g dg dd}d|j_tj \}}|jjd|d}| }|dks'J d S r   )	r   r   r    rx   ry   ro   r5   rK   r   )r6   r   rz   rG   r   label2r$   r$   r%   "test_line_use_index_false_diff_var  s   z2TestSeriesPlots.test_line_use_index_false_diff_varzWeird rounding problems)reasonstrictz
axis, meth)r>   rK   )rA   rW   c                 C   sR   t g d}tj \}}ttddgj|d|d}t	t||
 | d S )N)皙?      ?      $@g      Y@g     @@g     @   i  T)r?   rG   )r!   r   rx   ry   ro   getattrr   r5   r   r   get_ticklocs)r6   axismethr   rz   rG   r$   r$   r%   test_bar_log  s   zTestSeriesPlots.test_bar_logzaxis, kind, res_meth)r>   rK   get_ylim)rA   rW   r   c           
      C   s   t g d}tj \}}tg djd||d}d}d}t|| }	t	|	d | t	|	d | t
t|| | d S )	N)gh㈵>g-C6?MbP?{Gz?r   r   r   )r   r   r   T)r?   rU   rG   g!,NJ?gЄ-??r   rn   )r!   r   rx   ry   ro   r   r5   r   r   assert_almost_equalr   r   )
r6   r   rU   res_methr   rz   rG   yminymaxresr$   r$   r%   test_bar_log_kind_bar$  s   z%TestSeriesPlots.test_bar_log_kind_barc                 C   sH   t g dg dd}tj \}}|jjd|d}t| g d d S )Nrn   ra   r}   r   r   r   r   r   r   Fr   )0123)r   rx   ry   ro   r5   rK   r   get_xticklabels)r6   dfrz   rG   r$   r$   r%   test_bar_ignore_index:  s   z%TestSeriesPlots.test_bar_ignore_indexc                 C   sF   t g d}|jjg dd}dd |jD }g d}||ks!J d S )Nr   )redbluer   r   rc   c                 S      g | ]}|  qS r$   )get_facecolor)r*   pr$   r$   r%   r,   C      z8TestSeriesPlots.test_bar_user_colors.<locals>.<listcomp>)r           r  r   r  r  r   r   r  r  )r   r5   rK   rj   )r6   r   rG   resultr   r$   r$   r%   test_bar_user_colors@  s
   z$TestSeriesPlots.test_bar_user_colorsc                 C   s@   t tjdd}tj \}}|j|d}t	|dd d S )Nra   rR   rR   r~   r   r9   
r   r!   rg   rh   ri   rx   ry   ro   r5   r   r6   r   rz   rG   r;   r$   r$   r%   test_rotation_defaultL  s   z%TestSeriesPlots.test_rotation_defaultc                 C   sB   t tjdd}tj \}}|jd|d}t	|dd d S )Nra   r  r   )r8   rG   r9   r  r	  r$   r$   r%   test_rotation_30S  s   z TestSeriesPlots.test_rotation_30c                 C   s   ddl m} tdd}|g d }ttjdt||}t	j
 \}}|j|d}|tdd	d	d
|}|dd || d ksGJ t|dd d S )Nr   )DatetimeConverter1/1/2000z3/1/2000)	r   rn   ra   r}   rR   	   r         ra   r~   i  rn   r   z1/1/19991/1/2001r   r9   )%pandas.plotting._matplotlib.converterr  r   r   r!   rg   rh   ri   lenrx   ry   ro   r5   convertr   set_xlimr   r   )r6   r  rngrS   rz   rG   xpr$   r$   r%   test_irregular_datetimeY  s   
z'TestSeriesPlots.test_irregular_datetimec              	   C   s   t ddtjddddgddddtjddgd}tj \}}|j|d	}| \}}| }|t	|d
 j
ddd
 ks>J |t|d
 j
ddd
 ksOJ d S )Nr  r   g      @g      @g      @g      @g       @r   r~   r   Fr   )r   r!   r   rx   ry   ro   r5   r   r   nanminr   nanmax)r6   rS   rz   rG   r   r   r   r$   r$   r%   test_unsorted_index_xlimf  s   "&z(TestSeriesPlots.test_unsorted_index_xlimc                 C   sP   t tjdddg ddd}t|jj}t|j	|j
 | dks&J d S )Nra   rn   rR   r   r   r   r   eYLABELr   )r   r!   rg   rh   integersr   r5   pier   textsr   
get_ylabelr6   r'   rG   r$   r$   r%   test_pie_seriesr  s   zTestSeriesPlots.test_pie_seriesc                 C   sH   t tjdddg ddd}t|jjd d}t|j	dgd  d S )	Nra   rn   rR   r  r  r   r   r   )
r   r!   rg   rh   r  r   r5   r   r   r!  r#  r$   r$   r%   test_pie_series_no_label~  s   z(TestSeriesPlots.test_pie_series_no_labelc                 C   sT   t tjdddg ddd}g d}t|jj|d}g d	}t|j	|d
 d S )Nra   rn   rR   r  r  r   )rgr   r   )r&  r'  r   r&  r'  re   )
r   r!   rg   rh   r  r   r5   r   r   rj   )r6   r'   
color_argsrG   color_expectedr$   r$   r%   )test_pie_series_less_colors_than_elements  s   z9TestSeriesPlots.test_pie_series_less_colors_than_elementsc                 C   sb   t tjdddg ddd}g d}g d}t|jj||d	}t|j	| t
|j|d
 d S )Nra   rn   rR   r  r  r   )ABCDEr&  r'  r   r   m)r   r   re   )r   r!   rg   rh   r  r   r5   r   r   r!  r   rj   )r6   r'   r   r)  rG   r$   r$   r%   !test_pie_series_labels_and_colors  s   z1TestSeriesPlots.test_pie_series_labels_and_colorsc                 C   s   t tjdddg ddd}g d}t|jj|dd	d
}dd |j|	  D }t
tt|j|}t|j| |jD ]
}| d	ksIJ q?d S )Nra   rn   rR   r  r  r   r1  z%.2f   )r   autopctfontsizec                 S   s   g | ]}|d  dqS )d   z.2fr$   )r*   r   r$   r$   r%   r,         zHTestSeriesPlots.test_pie_series_autopct_and_fontsize.<locals>.<listcomp>)r   r!   rg   rh   r  r   r5   r   r   sumlistr   from_iterablezipr   r   r!  get_fontsize)r6   r'   r)  rG   pctsexpected_textstr$   r$   r%   $test_pie_series_autopct_and_fontsize  s   
z4TestSeriesPlots.test_pie_series_autopct_and_fontsizec                 C   sR   t g dg dd}tjtdd |j  W d    d S 1 s"w   Y  d S )N)rn   ra   r   r   rm   r  r   z&pie plot doesn't allow negative valuesr   )r   r   r   
ValueErrorr5   r   r_   r$   r$   r%   test_pie_series_negative_raises  s   "z/TestSeriesPlots.test_pie_series_negative_raisesc                 C   s<   t ddtjdgg ddd}t|jj}t|jg d d S )Nrn   ra   r   r   r  r   )r   r   r   r   )r   r!   r   r   r5   r   r   r!  r#  r$   r$   r%   test_pie_series_nan  s   z#TestSeriesPlots.test_pie_series_nanc                 C   sX   t dtjddg}tj \}}|jjd|d}g d}dd |jD }||ks*J d S )Nrn   Tr   )r   r   r   r   c                 S   r   r$   )get_text)r*   xr$   r$   r%   r,     r  z0TestSeriesPlots.test_pie_nan.<locals>.<listcomp>)	r   r!   r   rx   ry   ro   r5   r   r!  )r6   r   rz   rG   r   r  r$   r$   r%   test_pie_nan  s   zTestSeriesPlots.test_pie_nanc                 C   s   t tjddtdd}ttjdddd}tj	 \}}|j
|d}|j
d	d	|d
 t|g dd |  sAJ |j  sJJ d S )Nra   r   r}   abccolumnsr   rF  r   r~   Tr   r   rG   r   r   r   	x (right)r   r   r!   rg   rh   ri   r:  r   rx   ry   ro   r5   r   	get_yaxisget_visibleright_axr6   r   r   rz   rG   r$   r$   r%   test_df_series_secondary_legend  s   z/TestSeriesPlots.test_df_series_secondary_legendc                 C   s   t tjddtdd}ttjdddd}tj	 \}}|j
|d}|j
|d	d	d
 t|g dd |  sAJ |j  sJJ d S )Nra   rH  rI  rJ  r   rF  r   r~   TrG   r   r   rM  r   rO  rS  r$   r$   r%   )test_df_series_secondary_legend_with_axes  s   z9TestSeriesPlots.test_df_series_secondary_legend_with_axesc                 C   s   t tjddtdd}ttjdddd}tj	 \}}|j
d|d	}|j
dd|d
 g d}t|j|d |j  rFJ |  sNJ d S )Nra   rH  rI  rJ  r   rF  r   Tr   rG   rL  z	a (right)z	b (right)z	c (right)rN  r   r   r!   rg   rh   ri   r:  r   rx   ry   ro   r5   r   left_axrP  rQ  r6   r   r   rz   rG   r   r$   r$   r%   $test_df_series_secondary_legend_both  s   z4TestSeriesPlots.test_df_series_secondary_legend_bothc                 C   s   t tjddtdd}ttjdddd}tj	 \}}|j
d|d	}|j
|ddd
 g d}t|j| |j  rEJ |  sMJ d S )Nra   rH  rI  rJ  r   rF  r   TrW  rU  rX  rY  r[  r$   r$   r%   .test_df_series_secondary_legend_both_with_axis  s   z>TestSeriesPlots.test_df_series_secondary_legend_both_with_axisc                 C   s   t tjddtdd}ttjdddd}tj	 \}}|j
dd	|d
}|j
|ddd g d}t|j| |j  rFJ |  sNJ d S )Nra   rH  rI  rJ  r   rF  r   TF)r   
mark_rightrG   rU  rM  rY  r[  r$   r$   r%   0test_df_series_secondary_legend_both_with_axis_2  s   z@TestSeriesPlots.test_df_series_secondary_legend_both_with_axis_2zinput_logy, expected_scale)Tr?   )symsymlogc                 C   sj   t tjdd}t tjdd}|j|d}|jd|d}| |ks+J | |ks3J d S )Nra   r7  )r=   T)r   r=   )r   r!   rg   rh   ri   r5   
get_yscale)r6   
input_logyexpected_scales1s2r   r   r$   r$   r%   test_secondary_logy  s   z#TestSeriesPlots.test_secondary_logyc                 C   sl   t tjdd}tj \}}d}tj	t
|d |jdd|d W d    d S 1 s/w   Y  d S )Nra   zCannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbolr   zk--k)rC   rd   rG   )r   r!   rg   rh   ri   rx   ry   ro   r   r   rB  r5   )r6   rF  rz   rG   r   r$   r$   r%   )test_plot_fails_with_dupe_color_and_style'  s   "z9TestSeriesPlots.test_plot_fails_with_dupe_color_and_stylezbw_method, indscottr&         ?r7  c                 C   s    t d t|jj||d d S )NrY   	bw_methodind)r   importorskipr   r5   rX   )r6   r   rn  ro  r$   r$   r%   test_kde_kwargs2  s   

zTestSeriesPlots.test_kde_kwargsc                 C   s.   t d tddd}t|jjd|d d S )NrY   rl  r7  r&   rk  rm  )r   rp  r!   linspacer   r5   density)r6   r   sample_pointsr$   r$   r%   test_density_kwargs?  s   
z#TestSeriesPlots.test_density_kwargsc                 C   sZ   t d tj \}}tddd}|jjdd||d}t	|dd	 t
|j d
 d S )NrY   rl  r7  r&   Trk  )r=   rn  ro  rG   r?   r>   Density)r   rp  rx   ry   ro   r!   rr  r5   rX   r   r   r>   	get_label)r6   r   rz   rG   rt  r$   r$   r%   test_kde_kwargs_check_axesD  s   
z*TestSeriesPlots.test_kde_kwargs_check_axesc                 C   sZ   t d ttjdjdd}tj|d< t|j	j
}tt|jd   s+J d S )NrY   ra   2   )sizer   )r   rp  r   r!   rg   rh   uniformr   r   r5   rX   anyisnanr   	get_xdata)r6   r   r;   r$   r$   r%   test_kde_missing_valsL  s
   

"z%TestSeriesPlots.test_kde_missing_valszApi changed in 3.6.0)r   c                 C   s`   t j \}}|jjd|d}t|dd | }t||jg |	 }t|dgt
|  d S )NT)r=   rG   r?   rv  r   )rx   ry   ro   r5   r\   r   r   r   r    get_yticklabelsr  )r6   r   rz   rG   xlabelsylabelsr$   r$   r%   test_boxplot_seriesU  s   z#TestSeriesPlots.test_boxplot_seriesc                 C   s@   t d ttd}tj \}}|j||d tj  d S )NrY   r}   rU   rG   )	r   rp  r   r-   rx   ry   ro   r5   r   r6   rU   r   rz   rG   r$   r$   r%   test_kind_kwarg_  
   
zTestSeriesPlots.test_kind_kwargc                 C   s@   t d ttd}tj \}}t|j|  tj	  d S )NrY   r}   )
r   rp  r   r-   rx   ry   ro   r   r5   r   r  r$   r$   r%   test_kind_attrj  r  zTestSeriesPlots.test_kind_attrc                 C   s`   t td}tj \}}d}tjt|d |j||d W d    d S 1 s)w   Y  d S )Nabcdr   r   r  )	r   r:  rx   ry   ro   r   r   r   r5   r6   rU   r   rz   rG   r   r$   r$   r%   test_invalid_plot_datau     "z&TestSeriesPlots.test_invalid_plot_datac                 C   s,   t d ttdtd}t|j|d d S )NrY   r   r   r]   )r   rp  r   r-   objectr   r5   )r6   rU   r   r$   r$   r%   test_valid_object_plot}  s   
z&TestSeriesPlots.test_valid_object_plotc                 C   s`   t g d}tj \}}d}tjt|d |j||d W d    d S 1 s)w   Y  d S )N)r   r   r   ra   r   r   r  )r   rx   ry   ro   r   r   r   r5   r  r$   r$   r%    test_partially_invalid_plot_data  r  z0TestSeriesPlots.test_partially_invalid_plot_datac                 C   sL   t ddg}tjtdd |jdd W d    d S 1 sw   Y  d S )Nrn   ra   z%invalid_kind is not a valid plot kindr   invalid_kindr]   )r   r   r   rB  r5   r6   r   r$   r$   r%   test_invalid_kind  s   "z!TestSeriesPlots.test_invalid_kindc                 C   sP   t ddd}t ddd}||}tjd|j}t||d}t|j	 d S )Nz1/1/2009r   r   z1/2/2009ra   r   )
r   appendr!   rg   rh   ri   r{  r   r   r5   )r6   dr1dr2r   r   r   r$   r$   r%   test_dup_datetime_index_plot  s   
z,TestSeriesPlots.test_dup_datetime_index_plotc                 C   s   t tddd}tjdd}|j||d}tdd |jd	  D }|j	t
d
d	g | d
d	 }t|| dt| d}tjt|d |jtjddd W d    d S 1 shw   Y  d S )Nr   rF  r   ra   )ra   r   )yerrxerrc                 S   s   g | ]}|j d d df qS )Nrn   )verticesr)   r$   r$   r%   r,     s    z>TestSeriesPlots.test_errorbar_asymmetrical.<locals>.<listcomp>rn   rm   z?Asymmetrical error bars should be provided with the shape \(2, z\)r   )ra   r  r  )r   r!   r"   rg   rh   r5   vstackcollections	get_pathsTr   to_numpyreshaper   r   r  r   r   rB  )r6   r   errrG   r  r   r   r$   r$   r%   test_errorbar_asymmetrical  s   $"z*TestSeriesPlots.test_errorbar_asymmetricalr  ra   r   )r   ra   rF  yrJ  c                 C   s4   t tddd}t|j||d}t|ddd d S )Nr   rF  r   )r  rU   r   rn   r  r  )r   r!   r"   r   r5   r   )r6   rU   r  r   rG   r$   r$   r%   test_errorbar_plot  s   z"TestSeriesPlots.test_errorbar_plotc                 C   sJ   t tddd}ttjdd}t|j|d}t	|ddd d S )	Nr   rF  r   ra   )r  rn   r   r  )
r   r!   r"   r   rg   rh   ri   r   r5   r   )r6   r   s_errrG   r$   r$   r%   test_errorbar_plot_yerr_0  s   z)TestSeriesPlots.test_errorbar_plot_yerr_0r  )r  ra   c                 C   sH   t dddd}ttd|dd}||_t|j|d}t|d	d
d d S )Nr  r  ME)freqr  rF  r   r  r   rn   r  )r   r   r!   r"   r   r   r5   r   )r6   r  ixr   rG   r$   r$   r%   test_errorbar_plot_ts  s
   z%TestSeriesPlots.test_errorbar_plot_tsc                 C   sT   t tddd}tt |jtdd W d    d S 1 s#w   Y  d S )Nr   rF  r   r  r  )r   r!   r"   r   external_error_raisedrB  r5   r  r$   r$   r%   %test_errorbar_plot_invalid_yerr_shape  s   "z5TestSeriesPlots.test_errorbar_plot_invalid_yerr_shapec                 C   sX   t tddd}dgd }tt |j|d W d    d S 1 s%w   Y  d S )Nr   rF  r   zzzr  )r   r!   r"   r   r  r   r5   )r6   r   r  r$   r$   r%   test_errorbar_plot_invalid_yerr  s
   
"z/TestSeriesPlots.test_errorbar_plot_invalid_yerrc                 C   s   t |jdd d S )NTtabler4   r_   r$   r$   r%   test_table_true     zTestSeriesPlots.test_table_truec                 C   s   t |j|d d S )Nr  r4   r_   r$   r$   r%   test_table_self  r  zTestSeriesPlots.test_table_selfc                 C   s,   t d ttg dtjjtjj  d S )NrY   r|   )r   rp  r   r   r
   PlotAccessor_series_kinds_common_kinds)r6   r$   r$   r%   test_series_grid_settings  s
   

z)TestSeriesPlots.test_series_grid_settingsr   )r&  r   greenz#FF0000c                 C   s   ddl m} |d|d}||gksJ |d|gd}||gks!J |d|d}||gd ks0J |d|gd}||gd ks@J d S )Nr   get_standard_colorsrn   rc   r}   )!pandas.plotting._matplotlib.styler  )r6   r   r  r  r$   r$   r%   test_standard_colors  s   z$TestSeriesPlots.test_standard_colorsc                 C   s  ddl m} ddlm} |jD ]<}|d|d}||gksJ |d|gd}||gks,J |d|d}||gd ks;J |d|gd}||gd ksKJ q|jjD ]<}|d|d}||gks_J |d|gd}||gksmJ |d|d}||gd ks|J |d|gd}||gd ksJ qPd S )Nr   r(  r  rn   )
num_colorsrd   r}   )r   r   r  r  cnamesColorConverter)r6   r   r  r   r  r$   r$   r%   test_standard_colors_all  s*   
z(TestSeriesPlots.test_standard_colors_allc                 C   s@   t j \}}ttdd jd|d}t| dgd d S )Nr  rn   r  rd   rG   
linecolors)	rx   ry   ro   r   r!   r"   r5   r   r   r6   rz   rG   r$   r$   r%   test_series_plot_color_kwargs(  s   z-TestSeriesPlots.test_series_plot_color_kwargsc                 C   sL   t j \}}ttdd tddddjd|d}t|	 dgd d S )	Nr  rn   r  r   r   r  r  r  )
rx   ry   ro   r   r!   r"   r   r5   r   r   r  r$   r$   r%   "test_time_series_plot_color_kwargs.  s
   z2TestSeriesPlots.test_time_series_plot_color_kwargsc           	      C   sz   dd l }t|j}tddd}ttdd|d}d}|j \}}t	|D ]}|j
|d	}q&t| |d | d
 d S )Nr   r  r  r   rn      r   r}   r~   r  )r   r   r   r   r   r!   r"   ry   ro   r-   r5   r   r   )	r6   rx   
def_colorsr   r   ncolorsrz   rG   r+   r$   r$   r%   -test_time_series_plot_color_with_empty_kwargs6  s   
z=TestSeriesPlots.test_time_series_plot_color_with_empty_kwargsc                 C   s`   t tddd tdD d}tj \}}|jg d|d}dd dD }t|	 | d S )Nr   c                 S      g | ]}d |dqS P02dr$   r)   r$   r$   r%   r,   F  r8  z4TestSeriesPlots.test_xticklabels.<locals>.<listcomp>r   )r   r}   rR   r  )xticksrG   c                 S   r  r  r$   r)   r$   r$   r%   r,   I  r8  )
r   r!   r"   r-   rx   ry   ro   r5   r   r   )r6   r   rz   rG   r   r$   r$   r%   test_xticklabelsD  s
    z TestSeriesPlots.test_xticklabelsc                 C   s^   t tddd tdD d}|jjtdddd}tttddd}t||	  d S )	Nr   c                 S   r  r  r$   r)   r$   r$   r%   r,   N  r8  z6TestSeriesPlots.test_xtick_barPlot.<locals>.<listcomp>r   r   r  ra   )r  )
r   r-   r5   rK   r!   r   r:  r   r   
get_xticks)r6   r   rG   r   r$   r$   r%   test_xtick_barPlotL  s   z"TestSeriesPlots.test_xtick_barPlotc                 C   s@   ddl m} ttddtjdd|dgdd	d
}t|j d S )Nr   )CustomBusinessDayr7  y   z
2014-05-01z
2014-06-01z
2014-05-26)holidays)startendr  r   )pandas.tseries.offsetsr  r   r-   pdbdate_ranger   r5   )r6   r  r   r$   r$   r%   test_custom_business_day_freqS  s   
	z-TestSeriesPlots.test_custom_business_day_freqziGH#24426, see also github.com/pandas-dev/pandas/commit/ef1bd69fa42bbed5d09dd17f08c44fc8bfc2b685#r61470674c                 C   sj   t g d}tj \}}|j|d}|j }|jddgdd tj \}}|j }t	|| d S )Nr   r~   r   rn   T)inplace)
r   rx   ry   ro   r5   rA   r   dropr   r   )r6   rS   rz   rG   beforeafterr$   r$   r%   %test_plot_accessor_updates_on_inplaceb  s   

z5TestSeriesPlots.test_plot_accessor_updates_on_inplacerP   c                 C   sT   t ddg}tj \}}|j||d | }|d dk s J |d dks(J d S )Nra   r}   r  r   rn   )r   rx   ry   ro   r5   r   )r6   rU   r   rz   rG   xlimsr$   r$   r%   test_plot_xlim_for_seriesr  s   z)TestSeriesPlots.test_plot_xlim_for_seriesc                 C   sh   t td}|js
J | }t| dksJ | d }t| dks(J t| dks2J d S )Nr   rn   r   )r   intemptyr5   r  r   r  r   )r6   r   rG   rV   r$   r$   r%   test_plot_no_rows~  s   

z!TestSeriesPlots.test_plot_no_rowsc                 C   sH   t g d}tjtdd |  W d    d S 1 sw   Y  d S )Nr   r   r   )r   r   r   r   r5   )r6   r   r$   r$   r%   test_plot_no_numeric_data  s   
"z)TestSeriesPlots.test_plot_no_numeric_datazdata, indexr   )r}   ra   rn   r   )r   rz  r&   r   )iv  i  i  i  c                 C   sL   t ||d}|jdd}| }dd t|jdd dD }||ks$J d S )	N)r   r   rK   r]   c                 S   s   g | ]}|  jqS r$   )get_bboxr   )r*   patchr$   r$   r%   r,     s    z3TestSeriesPlots.test_plot_order.<locals>.<listcomp>c                 S   s
   |   jS r3   )r  r   )r  r$   r$   r%   <lambda>  s   
 z1TestSeriesPlots.test_plot_order.<locals>.<lambda>)r   )r   r5   tolistsortedrj   )r6   r   r   rS   rG   r   r  r$   r$   r%   test_plot_order  s   	zTestSeriesPlots.test_plot_orderc                 C   s4   t ddg}|jddd}|jd  dksJ d S )Nrn   ra   r   C3)rC   rd   r   )r   r5   r   	get_color)r6   r   rG   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   )rV   rP   rK   rW   r[   c                 C   s   t g d}||j_|j|d}|dkr%| dksJ | |ks$J n%|dkr:| dks1J | dks9J n| dksBJ | |ksJJ |j|||d}| |ksZJ | |ksbJ d S )Nr   r]   rW   r   r[   	Frequency)rU   ylabelxlabel)r   r   r    r5   r   r"  )r6   rU   
index_name	old_label	new_labelrS   rG   r$   r$   r%   test_xlabel_ylabel_series  s   z)TestSeriesPlots.test_xlabel_ylabel_seriesr   r/  )r  r   r  rn   )daysc                 C   s0   d}t ddg|dj|d}| dksJ d S )N)r}   rn   rn   ra   r   )xlim)r   r5   r   )r6   r   r  rG   r$   r$   r%   test_timedelta_index  s   	z$TestSeriesPlots.test_timedelta_indexc                 C   s@   t g d}|jd d}ttjjd d }t| |d d S )Nr|   rc   rn   r  )r   r5   r   rx   ry   r   r   r   )r6   r'   rG   r   r$   r$   r%   test_series_none_color  s   z&TestSeriesPlots.test_series_none_colorc                 C   s8   t d | }W d    d S 1 sw   Y  d S )NF)r   assert_produces_warningr5   )r6   r   rz   r$   r$   r%   test_plot_no_warning  s   
"z$TestSeriesPlots.test_plot_no_warning)x__name__
__module____qualname__r   markslowparametrizer7   r<   rH   rM   rQ   rT   paramtd
skip_if_nor^   r`   rk   rr   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   xfailr   r   r   r   r   r  r
  r  r  r  r$  r%  r+  r3  rA  rC  rD  rG  rT  rV  r\  r]  r_  rg  ri  r!   int_rr  rq  ru  ry  r  r  r
   r  r  r  r  r  r  r  r  r  r  r  r   r   rg   rh   ri   r:  r   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  timedelta_range	Timedeltar  r  r   r$   r$   r$   r%   r.   7   s   





	

	

	

		
	






	





!




r.   ),__doc__r   	itertoolsr   numpyr!   r   pandas.compatr   pandas.compat.numpyr   pandas.util._test_decoratorsutil_test_decoratorsr  pandasr  r   r   r   r	   r
   pandas._testing_testingr   pandas.tests.plotting.commonr   r   r   r   r   r   r   r   r   r   r   rp  rx   r   fixturer   r'   r.   r$   r$   r$   r%   <module>   s&    4



