a
    j=ic                     @   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)y   dd Z(Y n0 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   x/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/plotting/test_datetimelike.py<lambda>-       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sfJ |d
 j|d
 jfdksJ 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)selfZ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sFJ qFd S )N
   	   r$   r   Zfontsizer2   )r   nprandomrandnrangepltsubplotsr*   get_xticklabelsZget_yticklabelsZget_fontsize)r0   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/1987ZMSd   r"   r!   r"      r$   r   (   -   c   z2008-1-1 00:15:00Z15Tr5   )r   r   valuesr   r8   r9   r:   lenr)   r*   union)r0   idxr?   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   rG   r$   )Zxerr)Zyerr)r8   arrayr   r)   r*   )r0   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sTJ | j| d
}tj	t
|d |d   W d    n1 s0    Y  d S )NrC   ArG   rE   )xyzrQ   )rV   Br2   r(   zno numeric data to plotmatch)r   r   r<   r=   r*   rL   r,   closepytestraises	TypeError)r0   rN   r?   r@   r2   msgr   r   r   test_nonnumeric_excludec   s    z"TestTSPlot.test_nonnumeric_excluder"   )STr    DWMQrV   c                 C   sD   t d|dd}ttjt||}| j \}}t|j	|d d S N
12/31/1999rD   rE   r[   )
r   r   r8   r9   r:   rL   r<   r=   r)   r*   r0   r"   rN   ser_r2   r   r   r   test_tsplot_periodp   s    zTestTSPlot.test_tsplot_period)	rd   re   r    rf   rg   rh   zQ-DECrV   Z1B30Minc                 C   sD   t d|dd}ttjt||}| j \}}t|j	|d d S rj   )
r   r   r8   r9   r:   rL   r<   r=   r)   r*   rl   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   )        rt   rt   r(   r   )tmmakeTimeSeriesr<   r=   r*   r,   	get_color)r0   r1   rn   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>0    Y  |jdd}tjt|d |jddd W d    n1 s0    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 symbolr\   zb-z#000099)rs   rx   T)Zdrop)ru   rv   r_   r`   
ValueErrorr*   Zreset_index)r0   r1   rb   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/2012rD   r!   r"   r[   )
r<   r=   r   r   r8   r9   r:   rL   r)   r*   )r0   r"   rn   r2   rngrm   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s0J |tdddtddjksPJ |ddtd	djksjJ d S )
Nr   )get_datevaluerf     rV   rh   z1987-12rC   z1987-1-1)%pandas.plotting._matplotlib.converterr   r   ordinal)r0   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_ydataZformat_coord)r2   Zexpected_string
first_lineZfirst_xZ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-01rG   zA-DECr   r$   r[   zt = 2014  y = 1.000000rf   zt = 2014-01-01  y = 1.000000)r   r   r<   r=   r*   ru   r^   )r0   r   Zannualrn   r2   Z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 Nrk   rD   rE   )
r   r   r8   r9   r:   rL   r)   r*   r%   r"   r0   r"   rN   rm   r   r   r   test_line_plot_period_series   s    z'TestTSPlot.test_line_plot_period_seriesfrqncy)Z1SZ3SZ5TZ7HZ4DZ8WZ11MZ3Ac                 C   s:   t d|dd}ttjt||}t|j|jj	j
 d S r   )r   r   r8   r9   r:   rL   r)   r*   r%   r"   	rule_code)r0   r   rN   r{   r   r   r    test_line_plot_period_mlt_series   s    z+TestTSPlot.test_line_plot_period_mlt_seriesc                 C   s:   t d|dd}ttjt||}t|j|jj	j
 d S r   )r   r   r8   r9   r:   rL   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 Nrk   rD   rE   rG   )rV   rZ   Cr%   columns)
r   r   r8   r9   r:   rL   r)   r*   r%   r"   r0   r"   rN   r?   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   r8   r9   r:   rL   r%   asfreqr"   r   r)   r*   )r0   r   rN   r?   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   r8   r9   r:   rL   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 )Nrk   rD   rE   )r   rG         )r   r   r8   r9   r:   rL   rK   r	   Zasarrayr%   r)   r*   Z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rbJ d S )Nz2001-1-1z	2001-1-10r$   rG   r   r[   r"   )	r<   r=   r   r   r;   rL   r   r*   hasattr)r0   rn   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   ZBQSr5   rE   r$   )ru   rv   r)   r*   r   r%   r   r8   r9   r:   rL   )r0   rm   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   r8   r9   r:   rL   r)   r*   )r0   r   rm   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 ]8\}	}
||	d	}t|
 }t	|r|||ks|J q|d S )
Nr   z2012-6-22 21:59:51.960928L  rE   r   r$   r[   %H:%M:%S.%f)r   plotting_matplotlib	converterr   r   r8   r9   r:   rL   r<   r=   r*   	get_xaxisZget_ticklocsget_ticklabelszipZ_from_ordinalstrftimestrget_text)r0   convrN   r?   rn   r2   ZaxisZtlocsZtlabelslocrA   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  sJ | j \}}| }|jt|_|j	|d t
| d	  d d d	f  }t|dd  | dk  sJ d S )Nz2012-6-22 21:59:51rd   rD   rE   r   r$   )r   r(   rG      r[   r   g)QΠE>r(   g:0yE>)r   r   r8   r9   r:   rL   ilocr<   r=   r*   r   r,   
get_xydatadifffabsallcopyr%   astypeobject)	r0   rN   r?   irregrn   r2   diffssecrO   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sRJ qRd S )N)r   r(   r      r[   r   )	ru   rv   r<   r=   r*   r   r,   r-   r%   )r0   rm   rn   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sFJ | d 	 }t
|djdksjJ d S )Nr[   r   r&   datarZ   )ru   ZmakePeriodSeriesr<   r=   r*   r,   r   r%   r   r-   r   freqstrr0   btsrn   r2   rN   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sXJ | d  }t|djdks|J d S )Ni,  ZBMrh   r[   r   r&   r   )ru   rv   r   r   r<   r=   r*   r,   r   r%   r   r-   r   r   )r0   r   r1   rn   r2   rN   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    n1 sr0    Y  d S )Nr   r[   r   z)freq not specified and cannot be inferredr\   r   )r   ru   rv   r   r<   r=   r*   r,   r-   r_   r`   rz   r   )r0   r"   r   rn   r2   rN   rb   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rbJ d S )
Nz
2012-12-20   r    r      minutesr$   r[   r   )r   r   r   r8   aranger<   r=   r*   r,   r-   r	   is_normalized)r0   rN   r?   rn   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 )NrT   r[   r   )r   ru   rv   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 ]}|| qd S )Nc                    sH  |   }| |d d |d d  |   }|d |d d ksDJ |d |d d ks\J td| jtd| jf}| dd |   }t|d |d jksJ t|d |d jksJ td| jtd| jf}| tdddtddd |   }t|d |d jksJ t|d |d jks0J |  } j	| d S )	Nr   r   r(   r5   1/1/2000z4/1/2000r   r   )
get_xlimset_xlimr   r"   intr   r   
get_figurer<   r^   )r2   Zxlimresultexpectedr@   r0   r   r   _testl  s"    z*TestTSPlot.test_axis_limits.<locals>._testr[   r(   rT   bT)r=   )ru   rv   r<   r=   r*   r   )r0   r   rm   rn   r2   r?   axesr   r   r   test_axis_limitsk  s    zTestTSPlot.test_axis_limitsc                 C   s   dd l m  m  m} |td|jks0J |td|jksHJ |td|jks`J |td|jksxJ |td|j	ksJ |td|jksJ d S )Nr   rZ   rf   rh   ri   rV   rg   )
r   r   r   r   
get_finderr   Z_daily_finderZ_monthly_finderZ_quarterly_finderZ_annual_finder)r0   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 ]}td|d}ttjt||}| j	 \}	}
|j
|
d |
 }|| d  |
 \}}|
|d | || d  | j|
  q0||ksJ ||ksJ d S )	N)r5   rH      i  i  i
  '  z1999-1-1rZ   rF   r!   r[   r   ?)r   r   rL   r   r   r8   r9   r:   r<   r=   r*   r   appendget_majorticklocsr   r   r^   r   )r0   Zday_lstxpl1xpl2rs1rs2nr   rm   rn   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 ]}tdt|d dd}ttjt||}| j	
 \}	}
|j|
d |
 }|| d	  |
 \}}|
|d
 | || d	  | j	|
  q,||ksJ ||ksJ d S )Ng      @r   1988Q11987Q2r   ri   r   r[   r   r   r   r   rL   r   r   r   r8   r9   r:   r<   r=   r*   r   r   r   r   r   r^   r   r0   Zyrsr   r   r   r   r   r   rm   rn   r2   r   r   r   r   r   r   test_finder_quarterly  s"    z TestTSPlot.test_finder_quarterlyc                 C   s   g d}t djgt|  }}g }g }|D ]}tdt|d dd}ttjt||}| j	
 \}	}
|j|
d |
 }|| d  |
 \}}|
|d	 | || d  | j	|
  q,||ksJ ||ksJ d S )
N)gffffff?g      @r   r   zJan 1988r      rh   r   r[   r   r   r   r   r   r   r   test_finder_monthly  s"    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sjJ d S )Nr   i   rh   r   r[   r   Z1989Q1)r   r   r8   r9   r:   rL   r<   r=   r*   r   r   r   r   )r0   r   rm   rn   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 ]l}t d|dd}ttjt||}| j \}}|j|d |	 }|
| d	  | j|  q||ksJ d S )
N)r   i    r     i  i  iz  c                 S   s   g | ]}t |d djqS )rV   rF   )r   r   .0rW   r   r   r   
<listcomp>  r   z1TestTSPlot.test_finder_annual.<locals>.<listcomp>)r   r5      1   rJ      iW  i  Z1987rV   r   r[   r   )r   r   r8   r9   r:   rL   r<   r=   r*   r   r   r   r^   r   )	r0   r   r   Znyearsr   rm   rn   r2   r   r   r   r   test_finder_annual  s    zTestTSPlot.test_finder_annualc           	      C   st   d}t dd|d}ttjt||}| j \}}|j|d |	 }|
 d }tdddj}||kspJ d S )Ni@ 1/1/1999ZMinrE   r[   r   rF   r   r   r8   r9   r:   rL   r<   r=   r*   r   r   r   r   )	r0   Znminutesr   rm   rn   r2   r   r   r   r   r   r   test_finder_minutely  s    zTestTSPlot.test_finder_minutelyc           	      C   st   d}t dd|d}ttjt||}| j \}}|j|d |	 }|
 d }tdddj}||kspJ d S )N   r	  r    rE   r[   r   rF   r
  )	r0   Znhoursr   rm   rn   r2   r   r   r   r   r   r   test_finder_hourly  s    zTestTSPlot.test_finder_hourlyc           
      C   s>  t  }tj|jdd< | j \}}|j|d | }t	|dksJJ |d }|
 }tjj|t|tjd}t|tjjjsJ |j}|dddf  sJ | j|  t  }|g d }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sJJ |j}|dddf  sjJ | 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      r[   r(   r   maskZ
fill_value)	r   r(   r   r   r   r6   r         r   )ru   rv   r8   nanr   r<   r=   r*   r,   rL   r   maMaskedArrayr   
isinstancecorer  r   r^   r   r   r9   r:   )
r0   r1   rn   r2   linesliner   r  rN   rm   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sJ t|j d
ksJ |d }| }tjj|t|tjd}t|tjjjsJ |j}	|	ddd
f  sJ d S )Nr   r  r[   r   r'   Z12hrF   Tsecondary_yr(   r  )ru   rv   r8   r  r   r<   r=   r*   r   r%   r   r9   r:   rL   r,   right_axr   r  r  r   r  r  r  r   )
r0   lowrn   r2   idxhr{   r  r  r   r  r   r   r   test_gap_upsampleD  s     zTestTSPlot.test_gap_upsamplec           
      C   sd  t tjd}t tjd}| j \}}|jdd}t|dsHJ t|drVJ | }|	 d }t |
 | }t|| |  dksJ |d   rJ | j| | j \}}	|j|	d |	  d	ksJ | j|	  | }|jdd}	|  s J t|dr0J t|ds@J t|	dsPJ t|	dr`J d S )
Nr5   Tr  left_axr  r   rightr[   left)r   r8   r9   r:   r<   r=   r*   r   get_axesr,   r   r-   ru   assert_series_equal	get_yaxisget_ticks_positionget_visibler^   r   )
r0   rm   ser2r@   rn   r2   r   r  r   ax2r   r   r   test_secondary_yY  s0    zTestTSPlot.test_secondary_yc                 C   s8  t ddd}ttjd|}ttjd|}| j \}}|jdd}t|dsXJ t|drfJ |	 }|
 d }t| |  }	t||	 |  d	ksJ |d   rJ | j| | j \}}
|j|
d
 |
  dksJ | j|
  | }|jdd}
|  s4J d S )Nr   r5   r   Tr  r!  r  r   r"  r[   r#  )r   r   r8   r9   r:   r<   r=   r*   r   r$  r,   r   r-   Zto_timestampru   r%  r&  r'  r(  r^   r   )r0   rN   rm   r)  r@   rn   r2   r   r  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rJJ | }|d 	 
 dksjJ d S )	Nr5   TZdensityr  kindr2   r!  r  r(   r"  )r   r8   r9   r:   r<   r=   r*   r   r$  r&  r'  r0   rm   r@   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sNJ d S )Nr5   Tbarr-  r(   r"  )
r   r8   r9   r:   r<   r=   r*   r$  r&  r'  r/  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sDJ |d   dks\J |d   d
kstJ d S )Nr   rG   rT   r   cr   rT   r4  T)r  r=   r   r"  r(   r#  r   r   r8   r9   r:   r*   r&  r'  r0   r?   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sFJ |d   dks^J |d   dksvJ d S )Nr   rG   r3  r5  r1  rT   r4  T)r.  r  r=   r   r"  r(   r#  r   r6  r7  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sJ |
|d jksJ d S 	Nr   r   r5   r   r         r  r[   grr   r   r(   rZ   r'   )ru   rv   r<   r=   r*   r,   r   r-   r   r%   r   r   r   r0   s1s2rn   r2   r*  r  idx1idx2r#  r"  Z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rJJ | }|d  }t ||j	
tj |d  }t ||j	
tj d S Nr;  r>  rr   r[   r"   r   r(   )ru   rv   r<   r=   r*   r   r,   r-   assert_numpy_array_equalr%   r   r   rK   r0   r@  rA  rn   r2   r  x1Z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sJ |
|jdsJ | \}	}
|j }|	|d jksJ |
|d jksJ d S r:  )ru   rv   to_framer   r<   r=   r*   r,   r   r-   equalsr%   r   r   r   r?  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rXJ | }|d 	 }t 
||jtj |d 	 }t 
||jtj d S rE  )ru   rv   rJ  r   r<   r=   r*   r   r,   r-   rF  r%   r   r   rK   rG  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svJ qvd S )	Nr	  m  rf   r   r   rh   r[   r   r   r   r8   r9   r:   rL   r<   r=   r*   r,   r   r-   r"   r0   r  idxlhighr  rn   r2   r  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sJ d S )Nz2012-01-01 13:00
2012-01-02r    rF   r   r$   re   r[   rrr   r   r(   )r   r8   r9   r:   r   r   interpolater<   r=   r*   r  r-   )r0   Zts_indZts_datar1   Zts2rn   r2   r   r   r   test_mixed_freq_alignment  s    z$TestTSPlot.test_mixed_freq_alignmentc           	      C   s^  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szJ qz| }t|jd
ksJ | 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s8J q8d S )Nr	  rN  rf   r   r   rh   T)legendr2   r   r      re   r   r    r[   )r   r   r8   r9   r:   rL   r<   r=   r*   r,   r   r-   r"   
get_legendZtextsr^   r   )	r0   r  rQ  rR  r  rn   r2   r  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   rG   r   r   r   r      r6   r5   r              z1/3/2000r   rZ   r   r[   )ru   rv   r   r   r8   r9   r:   rL   r<   r=   r*   )r0   r1   r   r   Zpsrn   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sJ |jdksJ |j	d
 
 d |j	d
 
 d ksJ | j \}}| }|j|d	 |j|d	 |j	d
 
 d |j	d
 
 d ksJ d S )N
2015-01-01rG   rh   r   r(   r   T)Znrowssharexr[   r   r&   )r   rM   r   r;   rL   r<   r=   r*   r"   r  r   Ztwinx)r0   rB  rC  r@  rA  r@   Z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sdJ t| |j
 ks~J d S )Nrc  z
2015-01-03r[   r   )r<   r=   r   r
   r   r;   rL   r*   r,   r-   r%   minmax)r0   rn   r2   Z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svJ qvd S )	Nr	  4   rg   r   r   rh   r[   r   rO  rP  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 ]P}	t|	 d
j|jksJ |	jdd}
t|
dkrt|
| qt|
| qt  d S )Nr	  ri  rg   r   r   rh   r[   i  i  i  i  i  i   i  i  i  i  i  i  Zdtyper   Forig)r   r   r8   r9   r:   rL   r<   r=   r*   r   asi8r   float64rR   r,   r   r-   r"   ru   rF  r^   )r0   r  rQ  rR  r  rn   r2   Z
expected_hZ
expected_lr  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 ]Z\}}| 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sJ t|jdd|	 |
|| j7 }
t|jdd|
 q| jtj
}	tjt|	tj
d}
tdD ]f}|jd|  }t| dj|jksJ t|jdd|	 |
|| j7 }
t|jdd|
 qTq`d
D ]b\}}| j \}}|j|d|d |j|d|d | jtj
}	tjt|	tj
d}
tdD ]b}|j| }t| dj|jksRJ 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 ]f}|jd|  }t| dj|jksJ t|jdd|	 |
|| j7 }
t|jdd|
 qqd S )Nr	  ri  rg   r   r   rh   rG   )r   r(   r   r   ))r  area)rr  r  T)r.  Zstackedr2   rk  rl  Frm  r   )r   r   r8   r9   randrL   r<   r=   r*   rR   rp  Zzerosr;   r  r   r-   r"   ru   rF  rK   r   r   ro  r   )r0   r  rQ  rR  r  Zkind1Zkind2rn   r2   Z
expected_xZ
expected_yrS   r  r  r   r   r   $test_from_resampling_area_line_mixedy  sd      

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sJ |	 D ]}t
| d	jd
ksJ qt  | j \}}|j|d |j|d t|	 dksJ |	 D ]}t
| d	jd
ksJ qd S )Nz2014-07-01 09:00rd   2   rE   Z100Lr   r[   r   r   r   )r   r   r8   r9   r:   rL   r<   r=   r*   r,   r   r-   r"   ru   r^   rP  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   r  rG   rl  r   r5   r   )r   r   r   r6   r[   )r   r   r8   r9   r:   rL   r	   r   r)   r*   r   r   r<   r=   )r0   rN   r?   rn   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 ]v\ }tt d\}	}
t|	d\}}	| }t|dkr|
dkrt||	|
d}nt||	|
d}||ksJ qd S )Nr(   rG   r   r   r  c                    s"   g | ]} t t|d   qS r   r   r   r   r  tr   r   r    r   z(TestTSPlot.test_time.<locals>.<listcomp>r   r$   r[   <   %H:%M:%S%H:%M)r   r8   r9   randintcumsumrR   r   r:   rL   r<   r=   r*   
get_xticksr>   r   divmodr   r   r   r   r0   deltasr1   r?   r@   r2   tickslabelslmr{   hr   r   r   rz  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 ]v\ }tt d\}	}
t|	d\}}	| }t|dkr|
dkrt||	|
d}nt||	|
d}||ksJ q|dd | }| }t||D ]~\ }tt d\}	}
t|	d\}}	| }t|dkr:|
dkrt||	|
d}nt||	|
d}||ks:J q:d S )Nr(   rG   r   r   r  c                    s"   g | ]} t t|d   qS rx  ry  r  rz  r   r   r    r   z4TestTSPlot.test_time_change_xlim.<locals>.<listcomp>r   r$   r[   r|  r}  r~  z1:30z5:00)r   r8   r9   r  r  rR   r   r:   rL   r<   r=   r*   r  r>   r   r  r   r   r   r   r   r  r   rz  r   test_time_change_xlim  s>    "
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 ]\ }tt d\}	}
t t  d }t|	d\}}	| }t|dkr|d dkrt||	|
|d}n`|d dkr<t||	|
|dd d }n4|
dkr\t||	|
|d}nt||	|
|d}||ksJ qd S )Nr(   rG   r   r   r  c                    s"   g | ]} t t|d   qS ))microsecondsry  r  rz  r   r   r  <  r   z.TestTSPlot.test_time_musec.<locals>.<listcomp>r   r$   r[   r|  g    .Ai  r   r}  r~  )r   r8   r9   r  r  rR   r   r:   rL   r<   r=   r*   r  r>   r   r  r   roundr   r   r   )r0   r  r1   r?   r@   r2   r  r  r  r  r{   r~   r  r   r   r   rz  r   test_time_musec9  s0    "
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sxJ qxt|d
sJ t|drJ |j	 D ]}t
| jdksJ qd S )Nr	  rN  rf   r   r   rh   r[   Tr  r2   r!  r  )r   r   r8   r9   r:   rL   r<   r=   r*   r,   r   r-   r"   r   r!  rP  r   r   r   test_secondary_upsampleX  s    z"TestTSPlot.test_secondary_upsamplec                 C   sZ  | j  }|d}t }|jddg|d | }t| dksJJ |	 d 
 dksbJ |	 d 
 d	kszJ |	 d
 
 dksJ |	 d 
 dksJ |j d u sJ t }| D ]}||  qt|dksJ | j | | j  }|d}|jddgd|d | }t| dks@J |	 d 
 dksZJ |	 d 
 dkstJ |	 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szJ | j | | j  }|d}t }|jddg|d}| }t| dksJ |j d u sJ t }| D ]}||  qt|dksJ | j | t }| j  }|d}|jddg|d}| }t| dkstJ |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sVJ d S )N   rV   rZ   r  r   r   z	A (right)r(   z	B (right)r   r   rG   rf   F)r  
mark_rightr2   r1  )r.  r  r2   )r.  r  r  r2   )r<   Zfigureadd_subplotru   ZmakeTimeDataFramer*   rZ  rL   r,   Z	get_textsr   r  setaddrw   r^   r=   ZmakeDataFrame)r0   r@   r2   r?   r[  colorsr  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rP| dksPJ qPd S )	Nr   r   rh   r   rG   r[   r   r   )r   r   r8   r9   r:   rL   r<   r=   r*   r   r   r   Zget_rotation)r0   r   r?   rn   r2   r   r  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 )NrT  r5   rf   startr!   r"   Y)rA   r   )r   r+   r;   rL   r<   r=   r*   ru   r   r   r-   )r0   rW   rX   rn   r2   r  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   )fmtZlwc                 S   s   g | ]}|  qS r   	toordinalr  r   r   r   r    r   z0TestTSPlot.test_mpl_nopandas.<locals>.<listcomp>c                 S   s   g | ]}|  qS r   r  r  r   r   r   r    r   c                 S   s   g | ]}|  qS r   r  r  r   r   r   r    r   rl  r   c                 S   s   g | ]}|  qS r   r  r  r   r   r   r    r   )r   r8   r   r<   r=   Z	plot_dater,   rR   rp  ru   rF  r   )
r0   datesZvalues1Zvalues2kwrn   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sJ |||j	 d|ksJ d S )Nr   DatetimeConverterr  r(   r   r   r   r]  r6   r5   r   r<  r=  r  r_  r`  r   r[    r   r  ru   rv   r<   r=   r*   r   convertr%   rf  rg  r0   r  r1   Zts_irregularrn   r2   r#  r"  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stJ ||
k sJ d S )	N)r(   r   rG   r   )r   r   r   r]  r(   r$   r   r[   Tr  )r   r<   r=   r*   r   r0   Zindex_1Zindex_2r@  rA  rn   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sJ ||
k sJ d S )N
2000-01-01r   rf   r  z
2000-01-05r(   r$   r   r[   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spJ ||ks|J d S )Nr  r   rf  r   r(   r$   r[   rf   Tr  )r   r   r<   r=   r*   r   Zresamplemean)	r0   r   r1   rn   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sJ |||j	 d	|ksJ d S )
Nr   r  r  r  r   r[   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   )r0   rK   rn   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sJ ||ksJ d S )Nc                 S   s   g | ]}d |dqS )z00:00:00.0000000z0>2dr   )r  rS   r   r   r   r  C  r   zATestTSPlot.test_format_timedelta_ticks_narrow.<locals>.<listcomp>r5   0nsr   rG   r   r7   c                 S   s   g | ]}|  qS r   r   r  r   r   r   r  L  r   )r8   r   r   r   r9   r:   rL   r<   r=   r*   drawr>   r0   Zexpected_labelsr   r?   r@   r2   r  Z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sJ 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   rG   r   r7   c                 S   s   g | ]}|  qS r   r  r  r   r   r   r  d  r   z?TestTSPlot.test_format_timedelta_ticks_wide.<locals>.<listcomp>)r   r   r8   r9   r:   rL   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   Z1dayr   r[   z1 day 2 hr 30 min 10 sr5   r  r   z1 ns)r   r;   r   r<   r=   r)   r*   r8   r9   r:   rL   )r0   r{   rn   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   r8   r   r<   r=   hist)r0   r   rW   Zw1Zw2rn   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 )NrQ   r  r   r  r   i  r$   i  i  r  r[   )r   r   r<   r=   r*   )r0   r@  rA  rn   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 sX0    Y  |j\}}t| |  d S )Nr5   z
1970-01-02r   r$   r>  )rx   )r   r8   r9   r:   r   r*   ru   Zassert_produces_warningDeprecationWarningr%   rK   r  rF  r   )r0   r{   r2   l1l2r   r   r   test_add_matplotlib_datetime64  s    2
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szJ d S )Nr   r5   r   rW   rX   r5  z
2018-01-01rf   r   r   )rW   rX   r   )r   r8   r9   ZRandomStaters  r   r<   r=   Zscatterr  r>   r   )r0   r?   r@   r2   rA   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-03rQ   )rW   rX   rW   rX   r   )Zxrot)r  r  z
2020-05-04r   T)rX   Z	use_indexF)rW   rX   r=   rd  )r   r   r*   Z_check_ticks_propsZ	set_index)r0   rW   r?   r   r   r   r   test_check_xticks_rot  s     z TestTSPlot.test_check_xticks_rotN)`__name__
__module____qualname__r_   markfilterwarningsr4   rB   rP   rU   rc   Zparametrizero   rp   ry   r|   r   r   r   r   r   r   r   r   r   r   r   r   r   Zxfailr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  Zslowr  r  r  r   r+  r,  tdZskip_if_no_scipyr0  r2  r8  r9  rD  rI  rL  rM  rS  rW  r\  rb  re  rh  rj  rq  rt  rv  rw  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   2   s   









	

	
		(


3

	![)[

		
r   c              	   O   s  dd l m} | }zX|  |d}|d| }t|dd }	| |i |}
|
d us`J |d| }|d ur|jj	}t
|tr|j}|	d u r|j	|ksJ |d ur|	d u r|j	|ksJ |d}||d< | |i |}
|
d usJ tjdd}|| W d    n1 s0    Y  tjdd}t|| W d    n1 sX0    Y  W || n|| 0 d S )Nr   r  r2   r"      T)Zreturn_filelike)Zmatplotlib.pyplotZpyplotZgcfZclfr  popZgcagetattrr%   r"   r  r   r   ru   Zensure_cleanZsavefigpickledumpr^   )fr"   Zseriesargskwargsr<   r@   r2   Zorig_axZorig_axfreqr   Zdfreqpathr   r   r   r)     s6    


*.r)   )NN)/__doc__r   r   r   r   r  numpyr8   r_   Zpandas._libs.tslibsr   r   Zpandas.util._test_decoratorsutilZ_test_decoratorsr  Zpandasr   r	   r
   r   r   r   r   Zpandas._testingZ_testingru   Zpandas.core.indexes.datetimesr   r   r   Zpandas.core.indexes.periodr   r   r   Zpandas.core.indexes.timedeltasr   Zpandas.tests.plotting.commonr   Z"pandas.plotting._matplotlib.compatr   ImportErrorZpandas.tseries.offsetsr   Zskip_if_no_mplr   r)   r   r   r   r   <module>   s>   $	           (