o
    i                     @   s"  d Z ddlmZmZmZmZ ddlZddlZddlZddl	m
Z
mZ ddlm  mZ ddlmZmZmZmZmZmZmZ ddlmZ ddlmZmZmZ ddlm Z m!Z!m"Z" ddl#m$Z$ dd	l%m&Z& zdd
l'm(Z( W n e)yx   dd Z(Y nw ddl*m+Z+ ej,G dd de&Z-dddZ.dS )z< Test cases for time series specific (freq conversion, etc)     )datedatetimetime	timedeltaN)
BaseOffset	to_offset)	DataFrameIndexNaTSeriesconcatisnato_datetime)DatetimeIndexbdate_range
date_range)PeriodPeriodIndexperiod_range)timedelta_range)TestPlotBase)mpl_ge_3_6_0c                   C   s   dS )NT r   r   r   b/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/tests/plotting/test_datetimelike.py<lambda>-   s    r   )WeekOfMonthc                   @   s  e Zd Zejddd Zdd Zdd Zdd	 Z	d
d Z
ejdg ddd Zejdg ddd Zdd Zdd Zejdddgdd Zdd Zdd Zejdg ddd  Zejd!g d"d#d$ Zejdg dd%d& Zejdg dd'd( Zejd!g d"d)d* Zejdg dd+d, Zejdg dd-d. Zd/d0 Zd1d2 Zd3d4 Zejje d5d6d7d8 Zd9d: Z d;d< Z!d=d> Z"d?d@ Z#dAdB Z$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dQdR Z,dSdT Z-ejj.dUdV Z/dWdX Z0dYdZ Z1d[d\ Z2d]d^ Z3d_d` Z4e5j6dadb Z7dcdd Z8dedf Z9dgdh Z:didj Z;dkdl Z<dmdn Z=dodp Z>dqdr Z?dsdt Z@dudv ZAdwdx ZBdydz ZCd{d| ZDd}d~ ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNejje d5d6dd ZOdd ZPdd ZQdd ZRdd ZSdd ZTdd ZUdd ZVdd ZWdd ZXdd ZYdd ZZdd Z[dd Z\ejjdd6dd Z]dd Z^dd Z_dS )
TestTSPlotzignore::UserWarningc                 C   s   |}t ddd|d}tddg|d}t|j | }t| d  }|d j|d jfd	ks3J |d
 j|d
 jfdksAJ d S )N1/1/2011   H)periodsfreqtzg     g@g     t@indexr   r   r   )   r   )	r   r   _check_plot_worksplotlist	get_lines	get_xdatahourminute)selftz_aware_fixturer"   r$   tsaxxdatar   r   r   test_ts_plot_with_tz4   s   
 zTestTSPlot.test_ts_plot_with_tzc                 C   s`   t tjddtdd}| j \}}|jd|d | |	  D ]
}|
 dks-J q#d S )N
   	   r#   r   fontsizer2   )r   nprandomrandnrangepltsubplotsr)   get_xticklabelsget_yticklabelsget_fontsize)r/   dffigr2   labelr   r   r   test_fontsize_set_correctlyA   s   z&TestTSPlot.test_fontsize_set_correctlyc                 C   s   t dddd}t|jd d}ttjt|d|d}t|j	 |dd	 
|d
d }ttjt|d|d}t|j	 t dddd}t|jd d}ttjt|d|d}t|j	 d S )N1/1/1987MSd   r!   r    r!      r#   r   (   -   c   z2008-1-1 00:15:0015Tr5   )r   r   valuesr   r9   r:   r;   lenr(   r)   union)r/   idxrB   df2r   r   r   test_frame_inferredI   s   

zTestTSPlot.test_frame_inferredc                 C   s:   t g d}t||d}t|j|d t|j|d d S )Nr'   r   rK   r#   )xerr)yerr)r9   arrayr   r(   r)   )r/   iar   r   r   test_is_error_nozeroindex\   s   z$TestTSPlot.test_is_error_nozeroindexc                 C   s   t dddd}tg dg dd|}| j \}}|j|d t| d	ks*J | j| d
}tj	t
|d |d   W d    d S 1 sKw   Y  d S )NrF   ArK   rI   )xyzrV   )r]   Br2   r'   zno numeric data to plotmatch)r   r   r=   r>   r)   rQ   r+   closepytestraises	TypeError)r/   rS   rB   rC   r2   msgr   r   r   test_nonnumeric_excludec   s   "z"TestTSPlot.test_nonnumeric_excluder!   )STr   DWMQr]   c                 C   D   t d|dd}ttjt||}| j \}}t|j	|d d S N
12/31/1999rH   rI   rb   )
r   r   r9   r:   r;   rQ   r=   r>   r(   r)   r/   r!   rS   ser_r2   r   r   r   test_tsplot_periodp   s   zTestTSPlot.test_tsplot_period)	rk   rl   r   rm   rn   ro   zQ-DECr]   1B30Minc                 C   rq   rr   )
r   r   r9   r:   r;   rQ   r=   r>   r(   r)   rt   r   r   r   test_tsplot_datetimew   s   zTestTSPlot.test_tsplot_datetimec                 C   sD   t  }| j \}}|jd|d d}|| d  ks J d S )Nkstyler2   )        r}   r}   r'   r   )tmmakeTimeSeriesr=   r>   r)   r+   	get_color)r/   r1   rv   r2   colorr   r   r   test_tsplot   s
   zTestTSPlot.test_tsplotc                 C   s   t  }d}tjt|d |jddd W d    n1 sw   Y  |jdd}tjt|d |jddd W d    d S 1 sDw   Y  d S )NzCannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbolrc   zb-z#000099)r|   r   T)drop)r~   r   rf   rg   
ValueErrorr)   reset_index)r/   r1   ri   sr   r   r   test_both_style_and_color   s   "z$TestTSPlot.test_both_style_and_colormsusc                 C   sD   | j  \}}tdd|d}ttjt||}t|j	|d d S )N1/1/2012rH   r    r!   rb   )
r=   r>   r   r   r9   r:   r;   rQ   r(   r)   )r/   r!   rv   r2   rngru   r   r   r   test_high_freq   s   zTestTSPlot.test_high_freqc                 C   sn   ddl m} |d dd u sJ |dddksJ |tdddtddjks(J |ddtd	djks5J d S )
Nr   )get_datevaluerm     r]   ro   z1987-12rF   z1987-1-1)%pandas.plotting._matplotlib.converterr   r   ordinal)r/   r   r   r   r   test_get_datevalue   s
    zTestTSPlot.test_get_datevaluec                 C   sz   dd }t dtddddd}| j \}}|j|d	 ||d
 t dtddddd}|j|d	 ||d t  d S )Nc                 S   s>   |   d }| d j}| d }|| ||ksJ d S )Nr   )r+   r,   r   	get_ydataformat_coord)r2   expected_string
first_linefirst_xfirst_yr   r   r   check_format_of_first_point   s   zITestTSPlot.test_ts_plot_format_coord.<locals>.check_format_of_first_pointr'   z
2014-01-01rK   zA-DECr   r#   rb   zt = 2014  y = 1.000000rm   zt = 2014-01-01  y = 1.000000)r   r   r=   r>   r)   r~   re   )r/   r   annualrv   r2   dailyr   r   r   test_ts_plot_format_coord   s   

z$TestTSPlot.test_ts_plot_format_coordc                 C   s8   t d|dd}ttjt||}t|j|jj	 d S Nrs   rH   rI   )
r   r   r9   r:   r;   rQ   r(   r)   r$   r!   r/   r!   rS   ru   r   r   r   test_line_plot_period_series   s   z'TestTSPlot.test_line_plot_period_seriesfrqncy)1S3S5T7H4D8W11M3Ac                 C   :   t d|dd}ttjt||}t|j|jj	j
 d S r   )r   r   r9   r:   r;   rQ   r(   r)   r$   r!   	rule_code)r/   r   rS   r   r   r   r    test_line_plot_period_mlt_series   s   z+TestTSPlot.test_line_plot_period_mlt_seriesc                 C   r   r   )r   r   r9   r:   r;   rQ   r(   r)   r$   r!   r   r   r   r   r   test_line_plot_datetime_series   s   z)TestTSPlot.test_line_plot_datetime_seriesc                 C   sB   t d|dd}ttjt|d|g dd}t|j|jj	 d S Nrs   rH   rI   rK   )r]   ra   Cr$   columns)
r   r   r9   r:   r;   rQ   r(   r)   r$   r!   r/   r!   rS   rB   r   r   r   test_line_plot_period_frame   s    z&TestTSPlot.test_line_plot_period_framec                 C   sR   t d|dd}ttjt|d|g dd}|j|jjj	j}t
|j| d S r   )r   r   r9   r:   r;   rQ   r$   asfreqr!   r   r(   r)   )r/   r   rS   rB   r!   r   r   r   test_line_plot_period_mlt_frame   s    z*TestTSPlot.test_line_plot_period_mlt_framec                 C   sR   t d|dd}ttjt|d|g dd}|j|jjj	j}t
|j| d S r   )r   r   r9   r:   r;   rQ   r$   	to_periodr!   r   r(   r)   r   r   r   r   test_line_plot_datetime_frame   s    z(TestTSPlot.test_line_plot_datetime_framec                 C   sf   t d|dd}ttjt||}t|jtt|j	}t
|j|j	j |g d }t
|j d S )Nrs   rH   rI   )r   rK         )r   r   r9   r:   r;   rQ   rP   r	   asarrayr$   r(   r)   inferred_freqr   r   r   r   test_line_plot_inferred_freq   s   z'TestTSPlot.test_line_plot_inferred_freqc                 C   sf   | j  \}}tdd}ttt||d}t|d d |dd  g}|j|d t|dr1J d S )Nz2001-1-1z	2001-1-10r#   rK   r   rb   r!   )	r=   r>   r   r   r<   rQ   r   r)   hasattr)r/   rv   r2   r   r1   r   r   r   test_fake_inferred_business   s   
z&TestTSPlot.test_fake_inferred_businessc                 C   sL   t  }t|j t|jd ddd}ttj	t
||d}t|j d S )Nr   BQSr5   rI   r#   )r~   r   r(   r)   r   r$   r   r9   r:   r;   rQ   )r/   ru   drr   r   r   test_plot_offset_freq   s
   
z TestTSPlot.test_plot_offset_freqc                 C   sL   t tdddtdddtdddg}ttjt||d}t|j d S )N  r'   r      r#   )	r	   r   r   r9   r:   r;   rQ   r(   r)   )r/   r   ru   r   r   r    test_plot_multiple_inferred_freq  s   &z+TestTSPlot.test_plot_multiple_inferred_freqzApi changed)reasonc                 C   s   dd l m  m  m} tdddd}ttjt	|d|d}| j
 \}}|j|d | }| }| }t||D ]\}	}
||	d	}t|
 }t	|rZ||ksZJ q>d S )
Nr   z2012-6-22 21:59:51.960928L  rI   r   r#   rb   %H:%M:%S.%f)r   plotting_matplotlib	converterr   r   r9   r:   r;   rQ   r=   r>   r)   	get_xaxisget_ticklocsget_ticklabelszip_from_ordinalstrftimestrget_text)r/   convrS   rB   rv   r2   axistlocstlabelslocrD   xprsr   r   r   test_uhf  s   zTestTSPlot.test_uhfc           	      C   s  t dddd}ttjt|d|d}|jg d }| j \}}|j	|d t
| d	  d d d	f  }d
}t|dd  ||d |g dk  sRJ | j \}}| }|jt|_|j	|d t
| d	  d d d	f  }t|dd  | dk  sJ d S )Nz2012-6-22 21:59:51rk   rH   rI   r   r#   )r   r'   rK      rb   r   g)QΠE>r'   g:0yE>)r   r   r9   r:   r;   rQ   ilocr=   r>   r)   r   r+   
get_xydatadifffabsallcopyr$   astypeobject)	r/   rS   rB   irregrv   r2   diffssecrT   r   r   r   test_irreg_hf  s   $,$&zTestTSPlot.test_irreg_hfc                 C   sl   t  }|g d }| j \}}|j|d}|d usJ t| d  |jD ]
\}}||ks3J q)d S )N)r   r'   r      rb   r   )	r~   r   r=   r>   r)   r   r+   r,   r$   )r/   ru   rv   r2   retr   r   r   r   r   "test_irregular_datetime64_repr_bug2  s    z-TestTSPlot.test_irregular_datetime64_repr_bugc                 C   sn   t  }| j \}}|j|d | d  d |jd jks#J | d 	 }t
|djdks5J d S )Nrb   r   r%   datara   )r~   makePeriodSeriesr=   r>   r)   r+   r   r$   r   r,   r   freqstrr/   btsrv   r2   rS   r   r   r   test_business_freq>  s   $zTestTSPlot.test_business_freqc                 C   s   t dd}|d}| j \}}|j|d | d  d |j	d j
ks,J | d  }t|djdks>J d S )Ni,  BMro   rb   r   r%   r   )r~   r   r   r   r=   r>   r)   r+   r   r$   r   r,   r   r   )r/   r   r1   rv   r2   rS   r   r   r   test_business_freq_convertF  s   
$z%TestTSPlot.test_business_freq_convertc                 C   s   t  }td|}| j \}}|j|d | d  }d}t	j
t|d t|d W d    d S 1 s:w   Y  d S )Nr   rb   r   z)freq not specified and cannot be inferredrc   r   )r   r~   r   r   r=   r>   r)   r+   r,   rf   rg   r   r   )r/   r!   r   rv   r2   rS   ri   r   r   r   test_freq_with_no_period_aliasO  s   "z)TestTSPlot.test_freq_with_no_period_aliasc                 C   sf   t ddddtdd }ttd|d}| j \}}|j|d | d	 	 }t
|jr1J d S )
Nz
2012-12-20   r   r      minutesr#   rb   r   )r   r   r   r9   aranger=   r>   r)   r+   r,   r	   is_normalized)r/   rS   rB   rv   r2   r   r   r   r   test_nonzero_base[  s   zTestTSPlot.test_nonzero_basec                 C   sT   t dt i}| j \}}|j|d | d  }t|j	
 t| d S )Nr[   rb   r   )r   r~   r   r=   r>   r)   r+   r,   assert_index_equalr$   r   r   r   r   r   r   test_dataframed  s
   zTestTSPlot.test_dataframec                    s    fdd}t  } j \}}|j|d ||  j \}}t||d d}|j|d || t||d d}|jdd}|D ]}|| qFd S )Nc                    sD  |   }| |d d |d d  |   }|d |d d ks"J |d |d d ks.J td| jtd| jf}| dd |   }t|d |d jksQJ t|d |d jks^J td| jtd| jf}| tdddtddd |   }t|d |d jksJ t|d |d jksJ |  } j	| d S )	Nr   r   r'   r5   1/1/2000z4/1/2000r   r   )
get_xlimset_xlimr   r!   intr   r   
get_figurer=   re   )r2   xlimresultexpectedrC   r/   r   r   _testl  s"   z*TestTSPlot.test_axis_limits.<locals>._testrb   r'   r[   bT)r>   )r~   r   r=   r>   r)   r   )r/   r  ru   rv   r2   rB   axesr   r  r   test_axis_limitsk  s   
zTestTSPlot.test_axis_limitsc                 C   s   dd l m  m  m} |td|jksJ |td|jks$J |td|jks0J |td|jks<J |td|j	ksHJ |td|jksTJ d S )Nr   ra   rm   ro   rp   r]   rn   )
r   r   r   r   
get_finderr   _daily_finder_monthly_finder_quarterly_finder_annual_finder)r/   r   r   r   r   test_get_finder  s   zTestTSPlot.test_get_finderc                 C   s   g d}t dddjgt|  }}g }g }|D ]L}td|d}ttjt||}| j	 \}	}
|j
|
d |
 }|| d  |
 \}}|
|d | || d  | j|
  q||kskJ ||ksqJ d S )	N)r5   rL      i  i  i
  '  z1999-1-1ra   rJ   r    rb   r   ?)r   r   rQ   r   r   r9   r:   r;   r=   r>   r)   r   appendget_majorticklocsr   r   re   r  )r/   day_lstxpl1xpl2rs1rs2nr   ru   rv   r2   xaxisvminvmaxr   r   r   test_finder_daily  s"   zTestTSPlot.test_finder_dailyc                 C   s   ddg}t djgt|  }}g }g }|D ]Q}tdt|d dd}ttjt||}| j	
 \}	}
|j|
d |
 }|| d	  |
 \}}|
|d
 | || d	  | j	|
  q||ksnJ ||kstJ d S )Ng      @r   1988Q11987Q2r   rp   r   rb   r   r  r   r   rQ   r   r  r   r9   r:   r;   r=   r>   r)   r   r  r  r   r   re   r  r/   yrsr  r  r  r  r  r   ru   rv   r2   r  r  r   r   r   r   test_finder_quarterly  "   z TestTSPlot.test_finder_quarterlyc                 C   s   g d}t djgt|  }}g }g }|D ]Q}tdt|d dd}ttjt||}| j	
 \}	}
|j|
d |
 }|| d  |
 \}}|
|d	 | || d  | j	|
  q||ksnJ ||kstJ d S )
N)gffffff?g      @r   r   zJan 1988r#     ro   r   rb   r   r  r$  r%  r   r   r   test_finder_monthly  r(  zTestTSPlot.test_finder_monthlyc                 C   sn   t dddd}ttjt||}| j \}}|j|d |	 }|
 d }tddj}||ks5J d S )Nr"  i   ro   r   rb   r   1989Q1)r   r   r9   r:   r;   rQ   r=   r>   r)   r   r  r   r   )r/   r   ru   rv   r2   r  r   r   r   r   r   test_finder_monthly_long  s   z#TestTSPlot.test_finder_monthly_longc           	      C   s   g d}dd |D }g }dD ]6}t d|dd}ttjt||}| j \}}|j|d |	 }|
| d	  | j|  q||ksLJ d S )
N)r   i    r-    i  i  iz  c                 S   s   g | ]	}t |d djqS )r]   rJ   )r   r   .0r^   r   r   r   
<listcomp>  s    z1TestTSPlot.test_finder_annual.<locals>.<listcomp>)r   r5      1   rN      iW  i  1987r]   r   rb   r   )r   r   r9   r:   r;   rQ   r=   r>   r)   r   r  r  re   r  )	r/   r   r   nyearsr   ru   rv   r2   r  r   r   r   test_finder_annual  s   zTestTSPlot.test_finder_annualc           	      C   t   d}t dd|d}ttjt||}| j \}}|j|d |	 }|
 d }tdddj}||ks8J d S )Ni@ 1/1/1999MinrI   rb   r   rJ   r   r   r9   r:   r;   rQ   r=   r>   r)   r   r  r   r   )	r/   nminutesr   ru   rv   r2   r  r   r   r   r   r   test_finder_minutely  s   zTestTSPlot.test_finder_minutelyc           	      C   r8  )N   r9  r   rI   rb   r   rJ   r;  )	r/   nhoursr   ru   rv   r2   r  r   r   r   r   r   test_finder_hourly  s   zTestTSPlot.test_finder_hourlyc           
      C   s6  t  }tj|jdd< | j \}}|j|d | }t	|dks%J |d }|
 }tjj|t|tjd}t|tjjjsCJ |j}|dddf  sRJ | j|  t  }|g d }tj|jdd< | j \}}|j|d}| }t	|dksJ |d }|
 }tjj|t|tjd}t|tjjjsJ |j}|dddf  sJ | j|  g d}ttjt	||}	tj|	jdd< | j \}}|	j|d | }t	|dksJ |d }|
 }tjj|t|tjd}t|tjjjs	J |j}|dddf  sJ d S )	Nr      rb   r'   r   mask
fill_value)	r   r'   r   r   r   r6   r)        r   )r~   r   r9   nanr   r=   r>   r)   r+   rQ   r   maMaskedArrayr   
isinstancecorerC  r   re   r  r   r:   r;   )
r/   r1   rv   r2   linesliner   rC  rS   ru   r   r   r   	test_gaps  sP   zTestTSPlot.test_gapsc           
      C   s   t  }tj|jdd< | j \}}|j|d t|j	d |j	d dd}t
tjt||}|jdd	 | }t|d
ksCJ t|j d
ksNJ |d }| }tjj|t|tjd}t|tjjjslJ |j}	|	ddd
f  s{J d S )Nr   rA  rb   r   r&   12hrJ   Tsecondary_yr'   rB  )r~   r   r9   rG  r   r=   r>   r)   r   r$   r   r:   r;   rQ   r+   right_axr   rH  rI  r   rJ  rK  rC  r   )
r/   lowrv   r2   idxhr   rL  rM  r   rC  r   r   r   test_gap_upsampleD  s    zTestTSPlot.test_gap_upsamplec           
      C   sZ  t tjd}t tjd}| j \}}|jdd}t|ds$J t|dr+J | }|	 d }t |
 | }t|| |  dksNJ |d   rXJ | j| | j \}}	|j|	d |	  d	ksuJ | j|	  | }|jdd}	|  sJ t|drJ t|dsJ t|	dsJ t|	drJ d S )
Nr5   TrP  left_axrR  r   rightrb   left)r   r9   r:   r;   r=   r>   r)   r   get_axesr+   r   r,   r~   assert_series_equal	get_yaxisget_ticks_positionget_visiblere   r  )
r/   ru   ser2rC   rv   r2   r
  rM  r   ax2r   r   r   test_secondary_yY  s0   zTestTSPlot.test_secondary_yc                 C   s6  t ddd}ttjd|}ttjd|}| j \}}|jdd}t|ds,J t|dr3J |	 }|
 d }t| |  }	t||	 |  d	ksXJ |d   rbJ | j| | j \}}
|j|
d
 |
  dksJ | j|
  | }|jdd}
|  sJ d S )Nr   r5   r  TrP  rV  rR  r   rW  rb   rX  )r   r   r9   r:   r;   r=   r>   r)   r   rY  r+   r   r,   to_timestampr~   rZ  r[  r\  r]  re   r  )r/   rS   ru   r^  rC   rv   r2   r
  rM  r   r_  r   r   r   test_secondary_y_tsu  s*   zTestTSPlot.test_secondary_y_tsc                 C   sn   t tjd}| j \}}|jdd|d}t|dsJ t|dr%J | }|d 	 
 dks5J d S )	Nr5   TdensityrQ  kindr2   rV  rR  r'   rW  )r   r9   r:   r;   r=   r>   r)   r   rY  r[  r\  r/   ru   rC   r2   r
  r   r   r   test_secondary_kde  s   zTestTSPlot.test_secondary_kdec                 C   sR   t tjd}| j \}}|jdd|d | }|d  	 dks'J d S )Nr5   Tbarrd  r'   rW  )
r   r9   r:   r;   r=   r>   r)   rY  r[  r\  rf  r   r   r   test_secondary_bar  s
   zTestTSPlot.test_secondary_barc                 C   sx   t tjddg dd}|jddgdd}|d	   d
ks"J |d   dks.J |d   d
ks:J d S )Nr   rK   r[   r	  cr   r[   rk  T)rQ  r>   r   rW  r'   rX  r   r   r9   r:   r;   r)   r[  r\  r/   rB   r
  r   r   r   test_secondary_frame  s
   zTestTSPlot.test_secondary_framec                 C   sz   t tjddg dd}|jdddgdd	}|d
   dks#J |d   dks/J |d   dks;J d S )Nr   rK   rj  rl  rh  r[   rk  T)re  rQ  r>   r   rW  r'   rX  r   rm  rn  r   r   r   test_secondary_bar_frame  s
   z#TestTSPlot.test_secondary_bar_framec                 C   s   t  }|g d }| j \}}|j|d |jd|d}| }t|d  }t|d  }t ||j	
d t ||j	
d | \}	}
|j	
 }|	|d jksZJ |
|d jkscJ d S 	Nr   r   r5   r   r)        rE  rb   gr{   r   r'   ra   r&   )r~   r   r=   r>   r)   r+   r   r,   r   r$   r   r   r   r/   s1s2rv   r2   r_  rL  idx1idx2rX  rW  pidxr   r   r   test_mixed_freq_regular_first  s   
z(TestTSPlot.test_mixed_freq_regular_firstc                 C   s   t  }|g d }| j \}}|jd|d |j|d t|dr%J | }|d  }t ||j	
tj |d  }t ||j	
tj d S Nrr  ru  r{   rb   r!   r   r'   )r~   r   r=   r>   r)   r   r+   r,   assert_numpy_array_equalr$   r   r   rP   r/   rw  rx  rv   r2   rL  x1x2r   r   r   test_mixed_freq_irregular_first  s   z*TestTSPlot.test_mixed_freq_irregular_firstc                 C   s   t   }|jg dd d f }| j \}}|j|d |jd|d}| }t|d 	 }t|d 	 }|
|jdsDJ |
|jdsOJ | \}	}
|j }|	|d jkscJ |
|d jkslJ d S rq  )r~   r   to_framer   r=   r>   r)   r+   r   r,   equalsr$   r   r   r   rv  r   r   r    test_mixed_freq_regular_first_df  s   
z+TestTSPlot.test_mixed_freq_regular_first_dfc                 C   s   t   }|jg dd d f }| j \}}|jd|d |j|d t|dr,J | }|d 	 }t 
||jtj |d 	 }t 
||jtj d S r}  )r~   r   r  r   r=   r>   r)   r   r+   r,   r~  r$   r   r   rP   r  r   r   r   "test_mixed_freq_irregular_first_df  s   z-TestTSPlot.test_mixed_freq_irregular_first_dfc                 C   s   t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |j|d |	 D ]}t
| djdksIJ q;d S )	Nr9  m  rm   r   r)  ro   rb   r   r   r   r9   r:   r;   rQ   r=   r>   r)   r+   r   r,   r!   r/   rT  idxlhighrS  rv   r2   rM  r   r   r   test_mixed_freq_hf_first  s   z#TestTSPlot.test_mixed_freq_hf_firstc                 C   s   t dddd}tjd}t||d}|d }| j \}}|j	|d}|j	d	|d
 |j
d  d |j
d  d ksBJ d S )Nz2012-01-01 13:00
2012-01-02r   rJ   r)  r#   rl   rb   rr{   r   r'   )r   r9   r:   r;   r   r   interpolater=   r>   r)   rL  r,   )r/   ts_indts_datar1   ts2rv   r2   r   r   r   test_mixed_freq_alignment  s   ,z$TestTSPlot.test_mixed_freq_alignmentc           	      C   sZ  t dddd}t dddd}ttjt||}ttjt||}| j \}}|jd|d |jd|d |	 D ]}t
| d	jdksKJ q=| }t|jd
ksYJ | j|  t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |j|d |	 D ]}t
| d	jdksJ qd S )Nr9  r  rm   r   r)  ro   T)legendr2   r   r      rl   r   r   rb   )r   r   r9   r:   r;   rQ   r=   r>   r)   r+   r   r,   r!   
get_legendtextsre   r  )	r/   rT  r  r  rS  rv   r2   rM  legr   r   r   test_mixed_freq_lf_first  s,   z#TestTSPlot.test_mixed_freq_lf_firstc                 C   sb   t  }|g d }tdddd}ttjt||}| j	 \}}|j
|d |j
|d d S )N)r   r'   r   rK   r   r   r   r      r6   r5   rE              z1/3/2000r   ra   r   rb   )r~   r   r   r   r9   r:   r;   rQ   r=   r>   r)   )r/   r1   r   r   psrv   r2   r   r   r   test_mixed_freq_irreg_period!  s   z'TestTSPlot.test_mixed_freq_irreg_periodc                 C   s  t dddd}|d d |dd  }ttt||}ttt||}| jjddd\}\}}|j|d	 |j|d	 |jdksEJ |jdksLJ |j	d
 
 d |j	d
 
 d ks`J | j \}}| }|j|d	 |j|d	 |j	d
 
 d |j	d
 
 d ksJ d S )N
2015-01-01rK   ro   r   r'   r   T)nrowssharexrb   r   r%   )r   rR   r   r<   rQ   r=   r>   r)   r!   rL  r   twinx)r/   ry  rz  rw  rx  rC   ax1r_  r   r   r   test_mixed_freq_shared_ax*  s   (,z$TestTSPlot.test_mixed_freq_shared_axc                 C   s   | j  \}}tdtdg}ttt||}|j|d | d 	 }|j
 t| ks2J t| |j
 ks?J d S )Nr  z
2015-01-03rb   r   )r=   r>   r   r
   r   r<   rQ   r)   r+   r,   r$   minmax)r/   rv   r2   dtir   r3   r   r   r   test_nat_handlingK  s   zTestTSPlot.test_nat_handlingc                 C   s   t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |j|d |	 D ]}t
| dj|jksJJ q;d S )	Nr9  4   rn   r   r)  ro   rb   r   r  r  r   r   r   test_to_weekly_resamplingW  s   z$TestTSPlot.test_to_weekly_resamplingc                 C   s   t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |j|d |	 j
tj}tjg dtjd	}| D ](}	t|	 d
j|jks]J |	jdd}
t|
dkrpt|
| qNt|
| qNt  d S )Nr9  r  rn   r   r)  ro   rb   i  i  i  i  i  i   i  i  i  i  i  i  dtyper   Forig)r   r   r9   r:   r;   rQ   r=   r>   r)   r   asi8r   float64rY   r+   r   r,   r!   r~   r~  re   )r/   rT  r  r  rS  rv   r2   
expected_h
expected_lrM  r3   r   r   r   test_from_weekly_resamplingb  s&   z&TestTSPlot.test_from_weekly_resamplingc                 C   s"  t dddd}t dddd}ttjt|d|g dd	}ttjt|d|g dd	}d
D ]\}}| j \}}|j|d|d |j|d|d tj	g dtj
d}	tjt|	tj
d}
tdD ].}|j| }t| j|jksvJ t|jdd|	 |
|| j7 }
t|jdd|
 qc| jtj
}	tjt|	tj
d}
tdD ]1}|jd|  }t| dj|jksJ t|jdd|	 |
|| j7 }
t|jdd|
 qq0d
D ]\}}| j \}}|j|d|d |j|d|d | jtj
}	tjt|	tj
d}
tdD ]1}|j| }t| dj|jks%J t|jdd|	 |
|| j7 }
t|jdd|
 qtj	g dtj
d}	tjt|	tj
d}
tdD ]3}|jd|  }t| dj|jksqJ t|jdd|	 |
|| j7 }
t|jdd|
 qZqd S )Nr9  r  rn   r   r)  ro   rK   )r   r'   r   r   ))rM  area)r  rM  T)re  stackedr2   r  r  Fr  r   )r   r   r9   r:   randrQ   r=   r>   r)   rY   r  zerosr<   rL  r   r,   r!   r~   r~  rP   r   r   r  r   )r/   rT  r  r  rS  kind1kind2rv   r2   
expected_x
expected_yrZ   rM  rL  r   r   r   $test_from_resampling_area_line_mixedy  sj     

z/TestTSPlot.test_from_resampling_area_line_mixedc                 C   s  t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |j|d t|	 dksAJ |	 D ]}t
| d	jd
ksSJ qEt  | j \}}|j|d |j|d t|	 dksuJ |	 D ]}t
| d	jd
ksJ qyd S )Nz2014-07-01 09:00rk   2   rI   100Lr   rb   r   r   r   )r   r   r9   r:   r;   rQ   r=   r>   r)   r+   r   r,   r!   r~   re   r  r   r   r   "test_mixed_freq_second_millisecond  s$   z-TestTSPlot.test_mixed_freq_second_millisecondc                 C   s   t dddt dddt dddg}ttjt|dt|td}t|j	 t
ddd	}|g d
 t}ttjt|d|}| j \}}t|j	|d d S )Nr   r'   r   rF  rK   r  r   r5   r  )r   r   r   r6   rb   )r   r   r9   r:   r;   rQ   r	   r   r(   r)   r   r   r=   r>   )r/   rS   rB   rv   r2   r   r   r   test_irreg_dtypes  s   " 
zTestTSPlot.test_irreg_dtypesc                    s  t dddddd tjddd }t fdd|D }ttjt|tjt|d|d	}| j	
 \}}|j|d
 | }| }t||D ];\ }tt d\}	}
t|	d\}}	| }t|dkr|
dkrzt||	|
d}n	t||	|
d}||ksJ qNd S )Nr'   rK   r   r   rF  c                    "   g | ]} t t|d   qS r   r   r  r   r/  tr   r   r1       " z(TestTSPlot.test_time.<locals>.<listcomp>r  r#   rb   <   %H:%M:%S%H:%M)r   r9   r:   randintcumsumrY   r   r;   rQ   r=   r>   r)   
get_xticksr?   r   divmodr  r   r   r   r/   deltasr1   rB   rC   r2   tickslabelslmr   hr   r   r   r  r   	test_time  s*   "zTestTSPlot.test_timec                    s  t dddddd tjddd }t fdd|D }ttjt|tjt|d|d	}| j	
 \}}|j|d
 | }| }t||D ];\ }tt d\}	}
t|	d\}}	| }t|dkr|
dkrzt||	|
d}n	t||	|
d}||ksJ qN|dd | }| }t||D ];\ }tt d\}	}
t|	d\}}	| }t|dkr|
dkrt||	|
d}n	t||	|
d}||ksJ qd S )Nr'   rK   r   r   rF  c                    r  r  r  r/  r  r   r   r1    r  z4TestTSPlot.test_time_change_xlim.<locals>.<listcomp>r  r#   rb   r  r  r  z1:30z5:00)r   r9   r:   r  r  rY   r   r;   rQ   r=   r>   r)   r  r?   r   r  r  r   r   r   r   r  r   r  r   test_time_change_xlim  sD   "z TestTSPlot.test_time_change_xlimc                    s|  t dddddd tjddd }t fdd|D }ttjt|tjt|d|d	}| j	
 \}}|j|d
}| }| }t||D ]m\ }tt d\}	}
t t  d }t|	d\}}	| }t|dkr|d dkrt||	|
|d}n.|d dkrt||	|
|dd d }n|
dkrt||	|
|d}n
t||	|
|d}||ksJ qNd S )Nr'   rK   r   r   rF  c                    r  ))microsecondsr  r/  r  r   r   r1  <  r  z.TestTSPlot.test_time_musec.<locals>.<listcomp>r  r#   rb   r  g    .Ai  r   r  r  )r   r9   r:   r  r  rY   r   r;   rQ   r=   r>   r)   r  r?   r   r  r  roundr   r   r   )r/   r  r1   rB   rC   r2   r  r  r  r  r   r   r  r   r   r   r  r   test_time_musec9  s4   "zTestTSPlot.test_time_musecc                 C   s   t dddd}t dddd}ttjt||}ttjt||}| j \}}|j|d |jd|d	}|	 D ]}t
| jdksIJ q<t|d
sQJ t|drXJ |j	 D ]}t
| jdksjJ q]d S )Nr9  r  rm   r   r)  ro   rb   TrQ  r2   rV  rR  )r   r   r9   r:   r;   rQ   r=   r>   r)   r+   r   r,   r!   r   rV  r  r   r   r   test_secondary_upsampleX  s   z"TestTSPlot.test_secondary_upsamplec                 C   sN  | j  }|d}t }|jddg|d | }t| dks%J |	 d 
 dks1J |	 d 
 d	ks=J |	 d
 
 dksIJ |	 d 
 dksUJ |j d u s^J t }| D ]	}||  qet|dkswJ | j | | j  }|d}|jddgd|d | }t| dksJ |	 d 
 dksJ |	 d 
 dksJ |	 d
 
 dksJ |	 d 
 dksJ | j | | j  \}}|jddg|d | }|	 d 
 dksJ |	 d 
 dksJ | j | | j  \}}|jddgd|d | }|	 d 
 dks*J |	 d 
 dks7J | j | | j  }|d}t }|jddg|d}| }t| dkscJ |j d u smJ t }| D ]
}||  qtt|dksJ | j | t }| j  }|d}|jddg|d}| }t| dksJ |j d u sJ t }| D ]
}||  qt|dksJ | j   | j  }|d}|jddg|d}| }t| dks J |j d u s
J t }| D ]
}||  qt|dks%J d S )N   r]   ra   r  r   r   z	A (right)r'   z	B (right)r   r   rK   rm   F)rQ  
mark_rightr2   rh  )re  rQ  r2   )re  rQ  r  r2   )r=   figureadd_subplotr~   makeTimeDataFramer)   r  rQ   r+   	get_textsr   rR  setaddr   re   r>   makeDataFrame)r/   rC   r2   rB   r  colorsrM  r   r   r   test_secondary_legendg  s   










z TestTSPlot.test_secondary_legendc                 C   sz   t dddd}ttjt|d|}| j \}}|j|d}|	 }|
 D ]}t| dkr:| dks:J q(d S )	Nr   r)  ro   r   rK   rb   r   r   )r   r   r9   r:   r;   rQ   r=   r>   r)   r   r   r   get_rotation)r/   r   rB   rv   r2   r  rM  r   r   r   test_format_date_axis  s   z TestTSPlot.test_format_date_axisc                 C   sX   t dddd}ttt|}| j \}}|j||dd}tt	|d 
 | d S )Nr  r5   rm   startr    r!   Y)rD   r   )r   r*   r<   rQ   r=   r>   r)   r~   r   r   r,   )r/   r^   r_   rv   r2   rL  r   r   r   test_ax_plot  s
   zTestTSPlot.test_ax_plotc           
      C   s  t dddt dddg}tddd}tdd	d}d
dd}| j \}}|jdd |D |fi | |jdd |D |fi | | \}}tjdd |D tjd}	t	
| d d df |	 tjdd |D tjd}	t	
| d d df |	 d S )N  r)     i  r'   g      $@g      &@g      ?g      (@-r   )fmtlwc                 S      g | ]}|  qS r   	toordinalr/  r   r   r   r1        z0TestTSPlot.test_mpl_nopandas.<locals>.<listcomp>c                 S   r  r   r  r/  r   r   r   r1    r  c                 S   r  r   r  r/  r   r   r   r1    r  r  r   c                 S   r  r   r  r/  r   r   r   r1    r  )r   r9   r   r=   r>   	plot_dater+   rY   r  r~   r~  r   )
r/   datesvalues1values2kwrv   r2   line1line2expr   r   r   test_mpl_nopandas  s   
 zTestTSPlot.test_mpl_nopandasc                 C   s   ddl m} t d d }|g d }| j \}}|d d j|d |dd  j|d | \}}|||j	
 d|ksCJ |||j	 d|ksQJ d S )Nr   DatetimeConverterrF  r'   r   r   r   r  r6   r5   r)  rs  rt  rE  r  r  r   rb    r   r  r~   r   r=   r>   r)   r   convertr$   r  r  r/   r  r1   ts_irregularrv   r2   rX  rW  r   r   r    test_irregular_ts_shared_ax_xlim  s    z+TestTSPlot.test_irregular_ts_shared_ax_xlimc                 C   s   g d}g d}t d|d}t d|d}| j \}}|j|d | \}}|jd|d | \}	}
||	ks:J ||
k s@J d S )	N)r'   r   rK   r   )r   r   r   r  r'   r#   r   rb   Tr  )r   r=   r>   r)   r   r/   index_1index_2rw  rx  rv   r2   left_beforeright_before
left_afterright_afterr   r   r   test_secondary_y_non_ts_xlim  s   z'TestTSPlot.test_secondary_y_non_ts_xlimc                 C   s   t dddd}t dddd}td|d}td|d}| j \}}|j|d	 | \}}|jd
|d | \}	}
||	ks@J ||
k sFJ d S )N
2000-01-01r   rm   r  z
2000-01-05r'   r#   r   rb   Tr  )r   r   r=   r>   r)   r   r  r   r   r    test_secondary_y_regular_ts_xlim  s   z+TestTSPlot.test_secondary_y_regular_ts_xlimc           	      C   s   t dddd}td|d}| j \}}|j|d | \}}|d jd	|d
 | \}}||ks8J ||ks>J d S )Nr  r  r  r   r'   r#   rb   rm   Tr  )r   r   r=   r>   r)   r   resamplemean)	r/   r   r1   rv   r2   r  r  r  r  r   r   r   #test_secondary_y_mixed_freq_ts_xlim  s   z.TestTSPlot.test_secondary_y_mixed_freq_ts_xlimc                 C   s   ddl m} t d d }|g d }| j \}}|d d j|d |dd  jd|d |d d j|d | \}}|||j	
 d	|ksNJ |||j	 d	|ks\J d S )
Nr   r  rF  r  r   rb   Tr  r  r  r	  r   r   r   "test_secondary_y_irregular_ts_xlim&  s    z-TestTSPlot.test_secondary_y_irregular_ts_xlimc                 C   sZ   t dddt dddg}| j \}}|| tddddtddddg}|| d S )Ni  r'   r   r)  )r   r=   r>   r)   r   )r/   rP   rv   r2   r   r   r   test_plot_outofbounds_datetime8  s
   
z)TestTSPlot.test_plot_outofbounds_datetimec                 C   s   dd t dD }tdddd}tt jt|d|}| j \}}|j	d|d	 | j
  | }d
d |D }t|t|ksEJ ||ksKJ d S )Nc                 S   s   g | ]}d |dqS )z00:00:00.0000000z0>2dr   )r0  rZ   r   r   r   r1  C  s    zATestTSPlot.test_format_timedelta_ticks_narrow.<locals>.<listcomp>r5   0nsr   rK   r   r7   c                 S   r  r   r   r/  r   r   r   r1  L  r  )r9   r   r   r   r:   r;   rQ   r=   r>   r)   drawr?   r/   expected_labelsr   rB   rC   r2   r  result_labelsr   r   r   "test_format_timedelta_ticks_narrowA  s   
z-TestTSPlot.test_format_timedelta_ticks_narrowc                 C   s   g d}t dddd}ttjt|d|}| j \}}|jd|d}| j	  |
 }d	d
 |D }t|t|ks?J ||ksEJ d S )N)	z00:00:00z1 days 03:46:40z2 days 07:33:20z3 days 11:20:00z4 days 15:06:40z5 days 18:53:20z6 days 22:40:00z8 days 02:26:40z9 days 06:13:20r  r5   1 dr   rK   r   r7   c                 S   r  r   r  r/  r   r   r   r1  d  r  z?TestTSPlot.test_format_timedelta_ticks_wide.<locals>.<listcomp>)r   r   r9   r:   r;   rQ   r=   r>   r)   r  r?   r  r   r   r    test_format_timedelta_ticks_wideP  s   
z+TestTSPlot.test_format_timedelta_ticks_widec                 C   s   t tdtddd}| j \}}t|j|d tdddd}t tj	t
||}| j \}}t|j|d tddd	d}t tj	t
||}| j \}}t|j|d d S )
Nr   1dayr  rb   z1 day 2 hr 30 min 10 sr5   r#  r   z1 ns)r   r<   r   r=   r>   r(   r)   r9   r:   r;   rQ   )r/   r   rv   r2   r$   r   r   r   test_timedelta_ploth  s   zTestTSPlot.test_timedelta_plotc                 C   s`   t dddd}|}tddd}tdddd d d }| j \}}|j||g||gd	 d S )
Nr   r5   r   r   r   r'   g?r&   )weights)r   r9   r   r=   r>   hist)r/   r   r^   w1w2rv   r2   r   r   r   	test_histz  s   zTestTSPlot.test_histc                 C   s   t g dtdddtdddtdddgd}t g dtdddtd	ddtd
ddgd}| j \}}|j|d |j|d |j|d d S )NrV   r.  r)  r  r   i  r#   i  i  r  rb   )r   r   r=   r>   r)   )r/   rw  rx  rv   r2   r   r   r   test_overlapping_datetime  s$   





z$TestTSPlot.test_overlapping_datetimez1GH9053 matplotlib does not use ax.xaxis.converterc                 C   s   t tjdtdddd}| }tt |j|j	|j
dd W d    n1 s,w   Y  |j\}}t| |  d S )Nr5   z
1970-01-02r  r#   ru  )r   )r   r9   r:   r;   r   r)   r~   assert_produces_warningDeprecationWarningr$   rP   rL  r~  r   )r/   r   r2   l1l2r   r   r   test_add_matplotlib_datetime64  s   
z)TestTSPlot.test_add_matplotlib_datetime64c                 C   s~   t tjdddddgd}tdddd	|d
< | j \}}|jd
d|d | j	  |
 d }d}| |ks=J d S )Nr   r5   r   r^   r_   rl  z
2018-01-01rm   r   r   )r^   r_   r   )r   r9   r:   RandomStater  r   r=   r>   scatterr  r?   r   )r/   rB   rC   r2   rD   r  r   r   r   "test_matplotlib_scatter_datetime64  s    
z-TestTSPlot.test_matplotlib_scatter_datetime64c                 C   s   t g d}t|g dd}|jddd}| j|dd t g d}t|g dd}|jddd}| j|d	d |djdd
d}| j|d	d |djddd}| j|dd |jddd
d
d}| j|d	d |jddd
dd}| j|dd d S )N)
2020-05-01
2020-05-02z
2020-05-03rV   )r^   r_   r^   r_   r   )xrot)r5  r6  z
2020-05-04r   T)r_   	use_indexF)r^   r_   r>   r  )r   r   r)   _check_ticks_props	set_index)r/   r^   rB   r
  r   r   r   test_check_xticks_rot  s    z TestTSPlot.test_check_xticks_rotN)`__name__
__module____qualname__rf   markfilterwarningsr4   rE   rU   r\   rj   parametrizerw   ry   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'  r*  r,  r7  slowr=  r@  rN  rU  r`  rb  tdskip_if_no_scipyrg  ri  ro  rp  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  r  r  r  r"  r$  r&  r+  r,  r1  r4  r;  r   r   r   r   r   2   s    









	

	
		(


3

	![)[

		
r   c              	   O   s  dd l m} | }z|  |d}|d| }t|dd }	| |i |}
|
d us/J |d| }|d urR|jj	}t
|trG|j}|	d u rR|j	|ksRJ |d ura|	d u ra|j	|ksaJ |d}||d< | |i |}
|
d uswJ tjdd}|| W d    n1 sw   Y  tjdd}t|| W d    n1 sw   Y  W || d S W || d S || w )Nr   r  r2   r!      T)return_filelike)matplotlib.pyplotpyplotgcfclfr  popgcagetattrr$   r!   rJ  r   r   r~   ensure_cleansavefigpickledumpre   )fr!   seriesargskwargsr=   rC   r2   orig_axorig_axfreqr   dfreqpathr   r   r   r(     s>   


r(   )NN)/__doc__r   r   r   r   rQ  numpyr9   rf   pandas._libs.tslibsr   r   pandas.util._test_decoratorsutil_test_decoratorsrD  pandasr   r	   r
   r   r   r   r   pandas._testing_testingr~   pandas.core.indexes.datetimesr   r   r   pandas.core.indexes.periodr   r   r   pandas.core.indexes.timedeltasr   pandas.tests.plotting.commonr   "pandas.plotting._matplotlib.compatr   ImportErrorpandas.tseries.offsetsr   skip_if_no_mplr   r(   r   r   r   r   <module>   sB    $	           (