U
|
ý°dKÞ ã @ sö d Z ddlmZmZmZmZ ddlZddlZddlZddl m
|
Z
|
m Z ddl m mZ ddlmZmZmZmZmZmZmZ ddlmZ ddlm Z m Z m Z ddl m Z m!Z!m"Z" ddl#m$Z$ dd l%m&Z& dd
|
l'm(Z( ej)G d d d e&Z*dd dZ+dS )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)Ú WeekOfMonthc @ s e Zd Zej d¡dd Zdd Zdd Zdd Z d
|
d Z
|
ej d d dddddddg¡dd Z ej d d ddddddddg ¡dd Z dd Zd d Zej d d d g¡d!d" Zd#d$ Zd%d& Zej d d dddddddg¡d'd( Zej d)d*d+d,d-d.d/d0d1g¡d2d3 Zej d d ddddddddg ¡d4d5 Zej d d dddddddg¡d6d7 Zej d)d*d+d,d-d.d/d0d1g¡d8d9 Zej d d ddddddddg ¡d:d; Zej d d ddddddddg ¡d<d= Zd>d? Zd@dA ZdBdC Z ejj dDdEdFdG Z dHdI Z dJdK Z dLdM Z!dNdO Z"dPdQ Z#dRdS Z$dTdU Z%dVdW Z&dXdY Z'dZd[ Z(d\d] Z)d^d_ Z*d`da Z+dbdc Z,ejj-ddde Z.dfdg Z/dhdi Z0djdk Z1dldm Z2dndo Z3e4j5dpdq Z6drds Z7dtdu Z8dvdw Z9dxdy Z:dzd{ Z;d|d} Z<d~d Z=dd Z>dd Z?dd
Z@dd ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMejj dDdEd d¡ ZNd¢d£ ZOd¤d¥ ZPd¦d§ ZQd¨d© ZRdªd« ZSd¬d ZTd®d¯ ZUd°d± ZVd²d³ ZWd´dµ ZXd¶d· ZYd¸d¹ ZZdºd» Z[ejj d¼dEd½d¾ Z\d¿dÀ Z]dÁd Z^dÃS )ÄÚ
|
TestTSPlotzignore::UserWarningc C s |}t ddd|d}tddg|d}t|j | ¡ }t| ¡ d ¡ }|d j|d jfd ksft |d
|
j|d
|
jfd kst 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ÚAssertionError)ÚselfZtz_aware_fixturer r ÚtsÚaxÚxdata© r0 ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_datetimelike.pyÚtest_ts_plot_with_tz/ s
|
z TestTSPlot.test_ts_plot_with_tzc C s` t tj dd¡tdd}| j ¡ \}}|jd|d | ¡ | ¡ D ]}|
|
¡ dksFt qFd S )Né
|
é r r ©Zfontsizer. ) r ÚnpÚrandomÚrandnÚrangeÚpltÚsubplotsr% Úget_xticklabelsZget_yticklabelsZ get_fontsizer+ )r, ÚdfÚfigr. Úlabelr0 r0 r1 Útest_fontsize_set_correctly< s
|
z&TestTSPlot.test_fontsize_set_correctlyc C s t dddd}t|jd d}ttj t|d¡|d}t|j |dd
|
|d
|
d
¡}ttj t|d¡|d}t|j t d d dd}t|jd d}ttj t|d¡|d}t|j d S )Nú1/1/1987ZMSéd ©r r ©r é r r é( é- éc z2008-1-1 00:15:00Z15Tr3 ) r r Úvaluesr r6 r7 r8 Úlenr$ r% Úunion)r, Úidxr= Údf2r0 r0 r1 Útest_frame_inferredD s
|
|
z TestTSPlot.test_frame_inferredc C s< t dddg¡}t||d}t|j|d t|j|d d S )Nr# r rE r )Zxerr)Zyerr)r6 Úarrayr r$ r% )r, ÚiÚar0 r0 r1 Útest_is_error_nozeroindexW s z$TestTSPlot.test_is_error_nozeroindexc C s t dddd}tdddgdd dgd
|
|}| j ¡ \}}|j|d t| ¡ dksXt| j |¡ d }t j
|
t |d |d ¡ W 5 Q R X d S )NrA ÚArE rC ÚxÚyÚzr# r )rS ÚB©r. zno numeric data to plot©Úmatch) r r r: r; r% rJ r' r+ ÚcloseÚpytestÚraisesÚ TypeError)r, rL r= r> r. Úmsgr0 r0 r1 Útest_nonnumeric_exclude^ s z"TestTSPlot.test_nonnumeric_excluder ÚSÚTr ÚDÚWÚMÚQrS c C sD t d|dd}ttj t|¡|}| j ¡ \}}t|j |d d S ©Nú
|
12/31/1999rB rC rX )
|
r r r6 r7 r8 rJ r: r; r$ r% ©r, r rL ÚserÚ_r. r0 r0 r1 Útest_tsplot_periodk s z TestTSPlot.test_tsplot_periodzQ-DECZ1B30Minc C sD t d|dd}ttj t|¡|}| j ¡ \}}t|j |d d S rg )
|
r r r6 r7 r8 rJ r: r; r$ r% ri r0 r0 r1 Útest_tsplot_datetimer s z TestTSPlot.test_tsplot_datetimec C sD t ¡ }| j ¡ \}}|jd|d d}|| ¡ d ¡ ks@td S )NÚk©Ústyler. )ç rq rq r# r )ÚtmÚmakeTimeSeriesr: r; r% r' Ú get_colorr+ )r, r- rk r. Úcolorr0 r0 r1 Ú test_tsplot{ s
|
zTestTSPlot.test_tsplotc C sl t ¡ }d}tjt|d |jddd W 5 Q R X |jdd}tjt|d |jddd W 5 Q R X 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 symbolrY zb-z#000099)rp ru T)Zdrop)rr rs r\ r] Ú
|
ValueErrorr% Z reset_index)r, r- r_ Úsr0 r0 r1 Útest_both_style_and_color s ÿ z$TestTSPlot.test_both_style_and_colorÚmsÚusc C sD | j ¡ \}}tdd|d}ttj t|¡|}t|j |d d S )Nú1/1/2012rB ©r r rX )
|
r: r; r r r6 r7 r8 rJ r$ r% )r, r rk r. Úrngrj r0 r0 r1 Útest_high_freq s zTestTSPlot.test_high_freqc C sn ddl m} |d dd ks t|dddks0t|tdddtddjksPt|ddtd djksjtd S )
|
Nr )Ú get_datevaluerc éà rS re z1987-12rA z1987-1-1)Ú%pandas.plotting._matplotlib.converterr r+ r Úordinal)r, r r0 r0 r1 Útest_get_datevalue s
|
z TestTSPlot.test_get_datevaluec C sz dd }t dtddddd}| j ¡ \}}|j|d ||d
|
t dtddd dd}|j|d ||d t ¡ d S ) Nc S s> | ¡ d }| ¡ d j}| ¡ d }|| ||¡ks:td S )Nr )r' r( r Ú get_ydataZ format_coordr+ )r. Zexpected_stringÚ
|
first_lineZfirst_xZfirst_yr0 r0 r1 Úcheck_format_of_first_point s zITestTSPlot.test_ts_plot_format_coord.<locals>.check_format_of_first_pointr# z
|
2014-01-01rE zA-DECr} r rX zt = 2014 y = 1.000000rc z t = 2014-01-01 y = 1.000000)r r r: r; r% rr r[ )r, r Zannualrk r. Zdailyr0 r0 r1 Útest_ts_plot_format_coord s
|
|
z$TestTSPlot.test_ts_plot_format_coordc C s8 t d|dd}ttj t|¡|}t|j|jj d S ©Nrh rB rC )
|
r r r6 r7 r8 rJ r$ r% r r ©r, r rL rj r0 r0 r1 Ú test_line_plot_period_series² s z'TestTSPlot.test_line_plot_period_seriesÚfrqncyZ1SZ3SZ5TZ7HZ4DZ8WZ11MZ3Ac C s: t d|dd}ttj t|¡|}t|j|jj j
|
d S r ) r r r6 r7 r8 rJ r$ r% r r Ú rule_code)r, r rL rx r0 r0 r1 Ú test_line_plot_period_mlt_series¸ s z+TestTSPlot.test_line_plot_period_mlt_seriesc C s: t d|dd}ttj t|¡|}t|j|jj j
|
d S r ) r r r6 r7 r8 rJ r$ r% r r r r r0 r0 r1 Ú test_line_plot_datetime_series s z)TestTSPlot.test_line_plot_datetime_seriesc C sD t d|dd}ttj t|d¡|dddgd}t|j|jj d S © Nrh rB rC rE rS rW ÚC©r Úcolumns)
|
r r r6 r7 r8 rJ r$ r% r r ©r, r rL r= r0 r0 r1 Útest_line_plot_period_frameÊ s "z&TestTSPlot.test_line_plot_period_framec C sT t d|dd}ttj t|d¡|dddgd}|j |jjj ¡j}t
|
|j | d S r ) r r r6 r7 r8 rJ r Úasfreqr r r$ r% )r, r rL r= r r0 r0 r1 Ú test_line_plot_period_mlt_frameÐ s "z*TestTSPlot.test_line_plot_period_mlt_framec C sT t d|dd}ttj t|d¡|dddgd}|j |jjj ¡j}t
|
|j | d S r ) r r r6 r7 r8 rJ r Ú to_periodr r r$ r% r r0 r0 r1 Ú test_line_plot_datetime_frameÜ s "z(TestTSPlot.test_line_plot_datetime_framec C sj t d|dd}ttj t|¡|}t|jtt |j ¡}t
|
|j |j j |ddddg }t
|
|j d S )Nrh rB rC r rE é é ) r r r6 r7 r8 rJ rI r Zasarrayr r$ r% Z inferred_freqr r0 r0 r1 Ú test_line_plot_inferred_freqå s z'TestTSPlot.test_line_plot_inferred_freqc C sf | j ¡ \}}tdd}ttt||d}t|d d
|dd
g}|j|d t|drbt d S )Nz2001-1-1z 2001-1-10r rE r rX r )
|
r: r; r r r9 rJ r r% Úhasattrr+ )r, rk r. r~ r- r0 r0 r1 Útest_fake_inferred_businessñ s
|
z&TestTSPlot.test_fake_inferred_businessc C sL t ¡ }t|j t|jd ddd}ttj t
|
|¡|d}t|j d S )Nr ZBQSr3 rC r ) rr rs r$ r% r r r r6 r7 r8 rJ )r, rj Údrr0 r0 r1 Útest_plot_offset_freqù s
|
|
z TestTSPlot.test_plot_offset_freqc C sL t tdddtdddtdddg}ttj t|¡|d}t|j d S )NéÐ r# r é r ) r r r r6 r7 r8 rJ r$ r% )r, r rj r0 r0 r1 Ú test_plot_multiple_inferred_freq s &z+TestTSPlot.test_plot_multiple_inferred_freqzApi changed in 3.6.0)Úreasonc C sº dd l m m m} tdddd}ttj t |d¡|d}| j
|
¡ \}}|j |d | ¡ }| ¡ }| ¡ }t||D ]8\} }
|
| | ¡ d ¡} t|
|
¡ } t | r|| | ks|tq|d S )
|
Nr z2012-6-22 21:59:51.960928ÚLéô rC r r rX ú %H:%M:%S.%f)r ÚplottingÚ _matplotlibÚ converterr r r6 r7 r8 rJ r: r; r% Ú get_xaxisZ get_ticklocsÚget_ticklabelsÚzipZ _from_ordinalÚstrftimeÚstrÚget_textr+ ) r, ÚconvrL r= rk r. ZaxisZtlocsZtlabelsÚlocr? ÚxpÚrsr0 r0 r1 Útest_uhf s zTestTSPlot.test_uhfc C s$ t dddd}ttj t|d¡|d}|jddd d
|
g }| j ¡ \}}|j |d t
|
| ¡ d ¡ d d
df ¡ }d }t |dd
||d |g ¡d k ¡ s¨t| j ¡ \}}| ¡ }|j t¡|_|j |d t
|
| ¡ d ¡ d d
df ¡ }t |dd
| ¡d k ¡ s td S )Nz2012-6-22 21:59:51ra rB rC r r r r# rE é rX g)QÎ ÈEè>g:0âyE>)r r r6 r7 r8 rJ Úilocr: r; r% r r' Ú
|
get_xydataZdiffÚfabsÚallr+ Úcopyr ÚastypeÚobject) r, rL r= Úirregrk r. ZdiffsÚsecrM r0 r0 r1 Ú test_irreg_hf s $, $zTestTSPlot.test_irreg_hfc C sp t ¡ }|ddddg }| j ¡ \}}|j|d}|d k s>tt| ¡ d ¡ |j D ]\}}||ksVtqVd S )Nr r# r é rX )
|
rr rs r: r; r% r+ r r' r( r )r, rj rk r. Úretr´ r³ r0 r0 r1 Ú"test_irregular_datetime64_repr_bug, s z-TestTSPlot.test_irregular_datetime64_repr_bugc C sn t ¡ }| j ¡ \}}|j|d | ¡ d ¡ d |jd jksFt | ¡ d
|
¡ }t |dj dksjt d S )NrX r r! ©ÚdatarW ) rr ZmakePeriodSeriesr: r; r% r' r¸ r r r+ r( r Úfreqstr©r, Úbtsrk r. rL r0 r0 r1 Útest_business_freq8 s $z TestTSPlot.test_business_freqc C s t d¡ d¡}| d¡}| j ¡ \}}|j|d | ¡ d ¡ d |j d j
|
ksXt | ¡ d ¡ }t |djdks|t d S )Ni, ZBMre rX r r! rÄ )rr rs r r r: r; r% r' r¸ r r r+ r( r rÆ )r, rÈ r- rk r. rL r0 r0 r1 Útest_business_freq_convert@ s
|
$z%TestTSPlot.test_business_freq_convertc C sl t }t d¡ |¡}| j ¡ \}}|j|d | ¡ d ¡ }d}t j
|
t |d t |d W 5 Q R X d S )Nr rX r z)freq not specified and cannot be inferredrY rÄ ) r rr rs r r: r; r% r' r( r\ r] rw r )r, r rÈ rk r. rL r_ r0 r0 r1 Ú test_freq_with_no_period_aliasI s z)TestTSPlot.test_freq_with_no_period_aliasc C sf t ddddtdd }tt d¡|d}| j ¡ \}}|j|d | ¡ d ¡ }t
|
|j rbt d S )
|
Nz
|
2012-12-20é r r} é ©Úminutesr rX r ) r r r r6 Úaranger: r; r% r' r( r Ú is_normalizedr+ )r, rL r= rk r. r´ r0 r0 r1 Útest_nonzero_baseU s z TestTSPlot.test_nonzero_basec C sT t dt ¡ i}| j ¡ \}}|j|d | ¡ d ¡ }t |j
|
¡ t |¡ d S )NrQ rX r ) r rr rs r: r; r% r' r( Úassert_index_equalr r r rÇ r0 r0 r1 Útest_dataframe^ s
|
zTestTSPlot.test_dataframec s fdd}t ¡ } j ¡ \}}|j|d || j ¡ \}}t||d d}|j|d || t||d d}|jdd}|D ] }|| qd S )Nc sH | ¡ }| |d d |d d ¡ | ¡ }|d |d d ksDt|d |d d ks\ttd| jtd| jf}| dd¡ | ¡ }t|d |d jks¢tt|d |d jks¼ttd| jtd| jf}| tdddtddd¡ | ¡ }t|d |d jkstt|d |d jks0t| ¡ } j
|
|¡ d S ) Nr r r# r3 ú1/1/2000z4/1/2000r¡ r¶ ) Úget_xlimÚset_xlimr+ r r Úintr r Ú
|
get_figurer: r[ )r. ZxlimÚresultÚexpectedr> ©r, r0 r1 Ú_testf s" z*TestTSPlot.test_axis_limits.<locals>._testrX r# ©rQ ÚbT)r; )rr rs r: r; r% r )r, rÝ rj rk r. r= Úaxesr0 rÜ r1 Útest_axis_limitse s zTestTSPlot.test_axis_limitsc C s¬ dd l m m m} | td¡|jks0t| td¡|jksHt| td¡|jks`t| td¡|j ksxt| td¡|j
|
kst| td¡|jks¨td S )Nr rW rc re rf rS rd ) r r¨ r© rª Z
|
get_finderr Z _daily_finderr+ Z_monthly_finderZ_quarterly_finderZ_annual_finder)r, r± r0 r0 r1 Útest_get_finder s zTestTSPlot.test_get_finderc C sð dddddddg}t dd d
|
jgt| }}g }g }|D ]}td|d }ttj t|¡|}| j ¡ \} }
|
|j
|
|
|
d |
|
¡ } | | ¡ d ¡ |
|
¡ \} } |
|
| d | ¡ | | ¡ d ¡ | j |
|
¡ ¡ q:||ksàt||ksìtd S )Nr3 rF éü i i¶ i¾
|
é' z1999-1-1rW rD ©r rX r çÍÌÌÌÌÌì?)r r rJ r r r6 r7 r8 r: r; r% r« ÚappendÚget_majorticklocsrÖ r× r[ rÙ r+ )r, Zday_lstÚxpl1Úxpl2Úrs1Úrs2Únr~ rj rk r. ÚxaxisÚvminÚvmaxr0 r0 r1 Útest_finder_daily s" z TestTSPlot.test_finder_dailyc C sì ddg}t djgt| }}g }g }|D ]¢}tdt|d dd}ttj t|¡|}| j
|
¡ \} }
|
|j |
|
d |
|
¡ } | | ¡ d ¡ |
|
¡ \} } |
|
| d
|
| ¡ | | ¡ d ¡ | j |
|
¡ ¡ q,||ksÜt||ksètd S ) Ng @r¢ Ú1988Q1Ú1987Q2r¶ rf r} rX r ræ ©r r rJ r rØ r r6 r7 r8 r: r; r% r« rç rè rÖ r× r[ rÙ r+ ©r, Zyrsré rê rë rì rí r~ rj rk r. rî rï rð r0 r0 r1 Útest_finder_quarterly s" z TestTSPlot.test_finder_quarterlyc C sð ddddg}t djgt| }}g }g }|D ]¢}tdt|d dd }ttj t|¡|}| j
|
¡ \} }
|
|j |
|
d
|
|
|
¡ } | | ¡ d ¡ |
|
¡ \} } |
|
| d | ¡ | | ¡ d ¡ | j |
|
¡ ¡ q0||ksàt||ksìtd S ) Ngffffffò?g @r¶ r¢ zJan 1988ró é re r} rX r ræ rô rõ r0 r0 r1 Útest_finder_monthlyà s" z TestTSPlot.test_finder_monthlyc C sn t dddd}ttj t|¡|}| j ¡ \}}|j|d | ¡ }|
|
¡ d }t ddj }||ksjt d S )Nrò i re r} rX r Z1989Q1)r r r6 r7 r8 rJ r: r; r% r« rè r r r+ )r, r~ rj rk r. rî r´ r³ r0 r0 r1 Útest_finder_monthly_longÙ s z#TestTSPlot.test_finder_monthly_longc C s¨ ddddddddg}dd |D }g }d
|
D ]l}t d |d d }ttj t|¡|}| j ¡ \}}|j|d | ¡ }|
|
| ¡ d ¡ | j | ¡ ¡ q*||ks¤td S )Nr iÄ iÆ éË iä i iz c S s g | ]}t |d djqS )rS rD )r r ©Ú.0rT r0 r0 r1 Ú
|
<listcomp>å s z1TestTSPlot.test_finder_annual.<locals>.<listcomp>)r r3 é é1 rH éÇ iW ié Z1987rS r} rX r )r r r6 r7 r8 rJ r: r; r% r« rç rè r[ rÙ r+ ) r, r³ r´ Znyearsr~ rj rk r. rî r0 r0 r1 Útest_finder_annualã s z TestTSPlot.test_finder_annualc C st d}t dd|d}ttj t|¡|}| j ¡ \}}|j|d | ¡ }|
|
¡ d }t dddj }||kspt d S )Ni@ ú1/1/1999ZMinrC rX r rD ©r r r6 r7 r8 rJ r: r; r% r« rè r r r+ ) r, Znminutesr~ rj rk r. rî r´ r³ r0 r0 r1 Útest_finder_minutelyò s z TestTSPlot.test_finder_minutelyc C st d}t dd|d}ttj t|¡|}| j ¡ \}}|j|d | ¡ }|
|
¡ d }t dddj }||kspt d S )Né r r rC rX r rD r ) r, Znhoursr~ rj rk r. rî r´ r³ r0 r0 r1 Útest_finder_hourlyÿ s z TestTSPlot.test_finder_hourlyc
|
|
C sZ t ¡ }tj|jdd
< | j ¡ \}}|j|d | ¡ }t |dksJt
|
|d }| ¡ }tj j |t|tjd}t|tj jj st
|
|j}|dd
df ¡ s¤t
|
| j | ¡ ¡ t ¡ }|dddddd d
|
d d g }tj|jdd
< | j ¡ \}}|j|d}| ¡ }t |dkst
|
|d }| ¡ }tj j |t|tjd}t|tj jj sXt
|
|j}|dd
df ¡ sxt
|
| j | ¡ ¡ dddddd d
|
d d g }ttj t |¡|} tj| jdd
< | j ¡ \}}| j|d | ¡ }t |dksøt
|
|d }| ¡ }tj j |t|tjd}t|tj jj s6t
|
|j}|dd
df ¡ sVt
|
d S ) Nr é rX r# r ©ÚmaskZ
|
fill_valuer rÁ r4 r÷ é é )rr rs r6 Únanr· r: r; r% r' rJ r+ r¸ ÚmaÚ MaskedArrayr Ú
|
isinstanceÚcorer rº r[ rÙ r r7 r8 )
|
r, r- rk r. ÚlinesÚlinerÅ r rL rj r0 r0 r1 Ú test_gaps sP zTestTSPlot.test_gapsc
|
C sú t ¡ }tj|jdd
< | j ¡ \}}|j|d t|j d |j d dd}t
|
tj t |¡|}|jdd | ¡ }t |d
|
kstt |j ¡ d
|
kst|d }| ¡ }tjj|t|tjd }t|tjjjsØt|j} | dd
d
|
f ¡ sötd S ) Nr r rX r r" Z12hrD T©Ú secondary_yr# r )rr rs r6 r r· r: r; r% r r r r7 r8 rJ r' r+ Úright_axr¸ r r r r r r rº )
|
r, Úlowrk r. Úidxhrx r r rÅ r r0 r0 r1 Útest_gap_upsample> s z TestTSPlot.test_gap_upsamplec
|
C sd t tj d¡}t tj d¡}| j ¡ \}}|jdd}t|dsHtt|drVt| ¡ }|
|
¡ d }t | ¡ | ¡ }t ||¡ | ¡ ¡ dkst|d ¡ ¡ r°t| j |¡ | j ¡ \}} |j| d | ¡ ¡ d ksêt| j | ¡ ¡ | ¡ }|jdd} | ¡ ¡ s tt|dr0tt|ds@tt| dsPtt| dr`td S )
|
Nr3 Tr Úleft_axr r ÚrightrX Úleft)r r6 r7 r8 r: r; r% r r+ Úget_axesr' r
r( rr Úassert_series_equalÚ get_yaxisÚget_ticks_positionÚ get_visibler[ rÙ )
|
r, rj Úser2r> rk r. rà r r³ Úax2r0 r0 r1 Útest_secondary_yS s0 zTestTSPlot.test_secondary_yc C s8 t ddd}ttj d¡|}ttj d¡|}| j ¡ \}}|jdd}t|dsXt t|drft |
|
¡ }| ¡ d }t| ¡ | ¡ ¡ } t || ¡ | ¡ ¡ d ks°t |d ¡ ¡ rÄt | j |¡ | j ¡ \}}
|
|j|
|
d
|
|
|
¡ ¡ d ksþt | j |
|
¡ ¡ | ¡ }|jdd}
|
| ¡ ¡ s4t d S ) NrÕ r3 rå Tr r r r r rX r )r r r6 r7 r8 r: r; r% r r+ r r' r
r( Z to_timestamprr r r r r! r[ rÙ ) r, rL rj r" r> rk r. rà r r³ r# r0 r0 r1 Útest_secondary_y_tso s* z TestTSPlot.test_secondary_y_tsc C sn t tj d¡}| j ¡ \}}|jdd|d}t|ds<tt|drJt| ¡ }|d
|
¡ ¡ dksjtd S ) Nr3 TZdensity©r Úkindr. r r r# r ) r r6 r7 r8 r: r; r% r r+ r r r ©r, rj r> r. rà r0 r0 r1 Útest_secondary_kde s z TestTSPlot.test_secondary_kdec C sR t tj d¡}| j ¡ \}}|jdd|d | ¡ }|d ¡ ¡ dksNt
|
d S )Nr3 TÚbarr& r# r ) r r6 r7 r8 r: r; r% r r r r+ r( r0 r0 r1 Útest_secondary_bar s
|
z TestTSPlot.test_secondary_barc C sz t tj dd¡dddgd}|jddgdd}|d ¡ ¡ d
|
ksFt|d ¡ ¡ d ks^t|d ¡ ¡ d
|
ksvtd S )Nr rE rQ rß Úc©r T)r r; r r r# r r ©r r6 r7 r8 r% r r r+ ©r, r= rà r0 r0 r1 Útest_secondary_frame s
|
z TestTSPlot.test_secondary_framec C s| t tj dd¡dddgd}|jdddgdd }|d
|
¡ ¡ d ksHt|d ¡ ¡ d ks`t|d ¡ ¡ d ksxtd S )Nr rE rQ rß r, r- r* T)r' r r; r r r# r r r. r/ r0 r0 r1 Útest_secondary_bar_frame s
|
z#TestTSPlot.test_secondary_bar_framec C sÖ t ¡ }|ddddddddg }| j ¡ \}}|j|d |jd
|
|d }| ¡ }t|d ¡ }t|d ¡ }t ||j
|
d ¡¡ t ||j
|
d ¡¡ | ¡ \} }
|
|j
|
¡ } | | d j ksÀt |
|
| d j ksÒt d S ©Nr r r3 r¢ r÷ é é r
|
rX Úgro r# rW r" )rr rs r: r; r% r' r r( rÓ r r rÖ r r+ © r, Ús1Ús2rk r. r# r Úidx1Úidx2r r Zpidxr0 r0 r1 Ú test_mixed_freq_regular_first§ s
|
z(TestTSPlot.test_mixed_freq_regular_firstc C s¦ t ¡ }|ddddddddg }| j ¡ \}}|jd |d
|
|j|d t|d rVt| ¡ }|d ¡ }t ||j
|
t ¡j ¡ |d ¡ }t ||j
|
t ¡j ¡ d S ©Nr r r3 r¢ r÷ r3 r4 r
|
r5 ro rX r r# )rr rs r: r; r% r r+ r' r( Úassert_numpy_array_equalr r¼ r½ rI ©r, r7 r8 rk r. r Úx1Zx2r0 r0 r1 Ú test_mixed_freq_irregular_first½ s z*TestTSPlot.test_mixed_freq_irregular_firstc C sè t ¡ ¡ }|jddddddddgd d
f }| j ¡ \}}|j|d |jd
|
|d }| ¡ }t|d ¡ }t|d ¡ }|
|
|j d ¡¡st |
|
|j d ¡¡sªt | ¡ \} }
|
|j ¡ } | | d jksÒt |
|
| d jksät d S r2 )rr rs Úto_framer· r: r; r% r' r r( Úequalsr r r+ rÖ r r6 r0 r0 r1 Ú test_mixed_freq_regular_first_dfÊ s "
|
z+TestTSPlot.test_mixed_freq_regular_first_dfc C s´ t ¡ ¡ }|jddddddddgd d
f }| j ¡ \}}|jd |d
|
|j|d t|d rdt| ¡ }|d
|
¡ }t ||j t¡j¡ |d
|
¡ }t ||j t¡j¡ d S r< )rr rs rA r· r: r; r% r r+ r' r( r= r r¼ r½ rI r> r0 r0 r1 Ú"test_mixed_freq_irregular_first_dfÛ s " z-TestTSPlot.test_mixed_freq_irregular_first_dfc C s t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |j|d | ¡ D ] }t
|
| ¡ dj dksvt qvd S ) Nr ém rc r} r÷ re rX rÄ ©r r r6 r7 r8 rJ r: r; r% r' r r( r r+ ©r, r ÚidxlÚhighr rk r. r r0 r0 r1 Útest_mixed_freq_hf_firsté s z#TestTSPlot.test_mixed_freq_hf_firstc C s t dddd}tj d¡}t||d}| d¡ ¡ }| j ¡ \}}|j |d}|j d |d
|
|j
|
d ¡ d |j
|
d ¡ d kst d S ) Nz2012-01-01 13:00ú
|
2012-01-02r rD r÷ r rb rX Úrro r r# ) r r6 r7 r8 r r Z interpolater: r; r% r r( r+ )r, Zts_indZts_datar- Zts2rk r. r0 r0 r1 Útest_mixed_freq_alignmentô s z$TestTSPlot.test_mixed_freq_alignmentc C s^ t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|jd|d |jd|d | ¡ D ] }t
|
| ¡ d j dkszt qz| ¡ }t|jd
|
ks²t | j | ¡ ¡ t dd d d}t dd dd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |j|d | ¡ D ] }t
|
| ¡ d j d ks8t q8d S )Nr rE rc r} r÷ re T)Zlegendr. rÄ r éð rb r¶ r rX )r r r6 r7 r8 rJ r: r; r% r' r r( r r+ Ú
|
get_legendZtextsr[ rÙ ) r, r rH rI r rk r. r Úlegr0 r0 r1 Útest_mixed_freq_lf_first s* z#TestTSPlot.test_mixed_freq_lf_firstc C s~ t ¡ }|ddddddddd d
|
d d d dddg }tdddd}ttj t|¡|}| j ¡ \}}|j
|
|d |j
|
|d d S )Nr r# r rE r¶ r r rÁ é r4 r3 r
|
é é é é z1/3/2000rÍ rW r} rX ) rr rs r r r6 r7 r8 rJ r: r; r% )r, r- r¾ r~ Zpsrk r. r0 r0 r1 Ú test_mixed_freq_irreg_period s ( z'TestTSPlot.test_mixed_freq_irreg_periodc C s t dddd}|d d
|dd
¡}ttt||}ttt||}| jjddd\}\}}|j|d |j|d |jdkst |jdkst |j
|
d
|
¡ d |j
|
d
|
¡ d ksÀt | j ¡ \}}| ¡ }|j|d |j|d |j
|
d
|
¡ d |j
|
d
|
¡ d kst d S ) Nú
|
2015-01-01rE re r} r# r T)ZnrowsÚsharexrX r r! ) r rK r r9 rJ r: r; r% r r+ r r¸ Ztwinx)r, r9 r: r7 r8 r> Zax1r# r0 r0 r1 Útest_mixed_freq_shared_ax" s ( z$TestTSPlot.test_mixed_freq_shared_axc C s | j ¡ \}}tdtdg}ttt||}|j|d | ¡ d ¡ }|j
|
¡ t| ¡ ksdt t| ¡ |j
|
¡ ks~t d S )NrX z
|
2015-01-03rX r )r: r; r r
|
r r9 rJ r% r' r( r Úminr+ Úmax)r, rk r. Zdtirx r/ r0 r0 r1 Útest_nat_handlingB s z TestTSPlot.test_nat_handlingc C s t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |j|d | ¡ D ] }t
|
| ¡ dj |j ksvt qvd S ) Nr é4 rd r} r÷ re rX rÄ rF rG r0 r0 r1 Útest_to_weekly_resamplingM s z$TestTSPlot.test_to_weekly_resamplingc C s t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |j|d | ¡ j
|
tj ¡}tj dd d
|
d d d ddddddg tj d}| ¡ D ]P} t| ¡ dj|jksÎt| jdd}
|
t|
|
dkrôt |
|
|¡ q°t |
|
|¡ q°t ¡ d S )Nr r^ rd r} r÷ re rX éê éï éó é÷ éû é é é é é é é ©ZdtyperÄ F©Úorig)r r r6 r7 r8 rJ r: r; r% r Úasi8r¼ Úfloat64rO r' r r( r r+ rr r= r[ ) r, r rH rI r rk r. Z
|
expected_hZ
|
expected_lr r/ r0 r0 r1 Útest_from_weekly_resamplingX s& þ z&TestTSPlot.test_from_weekly_resamplingc C sZ t dddd}t dddd}ttj t|d¡|dd d
|
gd }ttj t|d¡|dd d
|
gd }d D ]p\}}| j ¡ \}}|j|d |d |j|d |d tj ddddddddddddg tj
|
d} tj t| tj
|
d}
|
t dD ]^} |j | } t| ¡ j|jkstt | jd d | ¡ |
|
|| j7 }
|
t | jd d |
|
¡ qà| ¡ j tj
|
¡} tj t| tj
|
d}
|
t dD ]f} |j d| } t| ¡ d j|jkstt | jd d | ¡ |
|
|| j7 }
|
t | jd d |
|
¡ qnqdd D ]v\}}| j ¡ \}}|j|d |d |j|d |d | ¡ j tj
|
¡} tj t| tj
|
d}
|
t dD ]b} |j | } t| ¡ d j|jksltt | jd d | ¡ |
|
|| j7 }
|
t | jd d |
|
¡ qBtj ddddddddddddg tj
|
d} tj t| tj
|
d}
|
t dD ]f} |j d| } t| ¡ d j|jkstt | jd d | ¡ |
|
|| j7 }
|
t | jd d |
|
¡ qêqÜd S ) Nr r^ rd r} r÷ re rE r r# r r ))r Úarea)rr r T)r' Zstackedr. r` ra rb rc rd re rf rg rh ri rj rk rl Frm rÄ )r r r6 r7 ÚrandrJ r: r; r% rO rp Úzerosr9 r r r( r r+ rr r= rI r
r ro r¼ )r, r rH rI r Zkind1Zkind2rk r. Z
|
expected_xZ
|
expected_yrP r r r0 r0 r1 Ú$test_from_resampling_area_line_mixedo s ""ôñ
|
|
ôñ z/TestTSPlot.test_from_resampling_area_line_mixedc C s t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |j|d t| ¡ dkst
|
| ¡ D ] }t | ¡ d j d
|
kst
|
qt ¡ | j ¡ \}}|j|d |j|d t| ¡ dksêt
|
| ¡ D ] }t | ¡ d j d
|
ksòt
|
qòd S ) Nz2014-07-01 09:00ra é2 rC Z100Lr¦ rX r rÄ r¥ )r r r6 r7 r8 rJ r: r; r% r' r+ r r( r rr r[ rG r0 r0 r1 Ú"test_mixed_freq_second_millisecondÊ s" z-TestTSPlot.test_mixed_freq_second_millisecondc C s¦ t dddt dddt dddg}ttj t|d¡t|td}t|j t
|
ddd }|d
|
d dd g t¡}ttj t|d¡|}| j ¡ \}}t|j |d d S )Nr¡ r# r r rE rl rÕ r3 rå r r r4 rX )r r r6 r7 r8 rJ r r½ r$ r% r r¼ r: r; )r, rL r= rk r. r0 r0 r1 Útest_irreg_dtypesá s "
|
z TestTSPlot.test_irreg_dtypesc s t dddddd tj ddd¡ ¡ }t fdd|D ¡}ttj t|¡tj t|¡d|d }| j
|
¡ \}}|j |d
|
| ¡ }| ¡ }t||D ]v\}} tt|d \}
|
} t|
|
d \} }
|
| ¡ } t| dkr| dkrôt| |
|
| d ¡}nt| |
|
| d ¡}|| kstqd S )Nr# rE rÍ r r c s" g | ]} t t|d ¡ qS ©rÎ ©r rØ r rû ©Útr0 r1 rý ñ s z(TestTSPlot.test_time.<locals>.<listcomp>rÞ r rX é< ú%H:%M:%Sú%H:%M)r r6 r7 ÚrandintÚcumsumrO r r8 rJ r: r; r% Ú
|
get_xticksr< r ÚdivmodrØ r° r r® r+ ©r, Údeltasr- r= r> r. ÚticksÚlabelsÚ_tickÚ_labelÚmrx Úhr´ r³ r0 r{ r1 Ú test_timeî s( ÿ zTestTSPlot.test_timec s¾ t dddddd tj ddd¡ ¡ }t fdd|D ¡}ttj t|¡tj t|¡d|d }| j
|
¡ \}}|j |d
|
| ¡ }| ¡ }t||D ]v\}} tt|d \}
|
} t|
|
d \} }
|
| ¡ } t| dkr| dkrôt| |
|
| d ¡}nt| |
|
| d ¡}|| kstq| dd¡ | ¡ }| ¡ }t||D ]~\}} tt|d \}
|
} t|
|
d \} }
|
| ¡ } t| dkr:| dkrt| |
|
| d ¡}nt| |
|
| d ¡}|| ks:tq:d S )Nr# rE rÍ r r c s" g | ]} t t|d ¡ qS ry rz rû r{ r0 r1 rý s z4TestTSPlot.test_time_change_xlim.<locals>.<listcomp>rÞ r rX r} r~ r z1:30z5:00)r r6 r7 r r rO r r8 rJ r: r; r% r r< r r rØ r° r r® r+ r× r r0 r{ r1 Útest_time_change_xlim s@ ÿ
|
z TestTSPlot.test_time_change_xlimc s t dddddd tj ddd¡ ¡ }t fdd|D ¡}ttj t|¡tj t|¡d|d }| j
|
¡ \}}|j |d
|
}| ¡ }| ¡ }t||D ]à\}} tt|d \}
|
} t|t| d } t|
|
d \} }
|
| ¡ }t|dkr| d dkrt| |
|
| | d¡}n`| d dkr<t| |
|
| | d¡d d
}n4| dkr\t| |
|
| | d¡}nt| |
|
| | d¡}||kstqd S )Nr# rE rÍ r r c s" g | ]} t t|d ¡ qS ))Ú microsecondsrz rû r{ r0 r1 rý 2 s z.TestTSPlot.test_time_musec.<locals>.<listcomp>rÞ r rX r} g .Aiè r§ éýÿÿÿr~ r )r r6 r7 r r rO r r8 rJ r: r; r% r r< r r rØ Úroundr° r r® r+ )r, r
r- r= r> r. r r r r r rx r{ r r´ r³ r0 r{ r1 Útest_time_musec/ s2 ÿ
|
zTestTSPlot.test_time_musecc C sÚ t dddd}t dddd}ttj t|¡|}ttj t|¡|}| j ¡ \}}|j|d |jd|d }| ¡ D ]}t
|
| ¡ j dksxt qxt|d
|
s¢t t|d r°t |j ¡ D ]}t
|
| ¡ j dksºt qºd S ) Nr rE rc r} r÷ re rX T©r r. r r )r r r6 r7 r8 rJ r: r; r% r' r r( r r+ r r rG r0 r0 r1 Útest_secondary_upsampleN s z"TestTSPlot.test_secondary_upsamplec C sZ | j ¡ }| d¡}t ¡ }|jddg|d | ¡ }t| ¡ dksJt |
|
¡ d ¡ dksbt |
|
¡ d ¡ d kszt |
|
¡ d
|
¡ d kst |
|
¡ d ¡ d ksªt |j ¡ d ks¼t t }| ¡ D ]}| | ¡ ¡ qÊt|dksît | j |¡ | j ¡ }| d¡}|jdd gd|d | ¡ }t| ¡ dks@t |
|
¡ d ¡ dksZt |
|
¡ d ¡ dkstt |
|
¡ d
|
¡ d kst |
|
¡ d ¡ d ks¨t | j |¡ | j ¡ \}}|jddg|d | ¡ }|
|
¡ d ¡ dksöt |
|
¡ d ¡ dkst | j |¡ | j ¡ \}}|jddgd|d | ¡ }|
|
¡ d ¡ dks`t |
|
¡ d ¡ dkszt | j |¡ | j ¡ }| d¡}t ¡ }|jd d g|d}| ¡ }t| ¡ dksÒt |j ¡ d ksæt t }| ¡ D ]}| | ¡ ¡ qôt|dks t | j |¡ t ¡ }| j ¡ }| d¡}|jddg|d}| ¡ }t| ¡ dkstt |j ¡ d kst t }| ¡ D ]}| | ¡ ¡ qt|dks¾t | j ¡ | j ¡ }| d¡}|jd d g|d}| ¡ }t| ¡ dks t |j ¡ d ks t t }| ¡ D ]}| | ¡ ¡ q.t|dksVt d S )NéÓ rS rW r r¶ r z A (right)r# z B (right)r r rE rc F)r Ú
|
mark_rightr. r* )r' r r. )r' r r r. )r: ÚfigureÚ add_subplotrr ZmakeTimeDataFramer% rO rJ r' r+ Z get_textsr° r ÚsetÚaddrt r[ r; Z makeDataFrame)r, r> r. r= rP Úcolorsr r0 r0 r1 Útest_secondary_legend] s
|
|
|
|
|
|
|
|
|
|
|
z TestTSPlot.test_secondary_legendc C sz t dddd}ttj t|d¡|}| j ¡ \}}|j|d}| ¡ }|
|
¡ D ]$}t| ¡ dkrP| ¡ dksPt qPd S ) Nr| r÷ re r} rE rX r rÍ )r r r6 r7 r8 rJ r: r; r% r« r¬ r° Z get_rotationr+ )r, r~ r= rk r. rî r r0 r0 r1 Útest_format_date_axis¸ s z TestTSPlot.test_format_date_axisc C sX t dddd}ttt|}| j ¡ \}}|j||dd}t t |d
|
¡ |¡ d S )NrK r3 rc ©Ústartr r ÚY)r? r ) r r& r9 rJ r: r; r% rr rÓ r r( )r, rT rU rk r. r r0 r0 r1 Ú test_ax_plotà s
|
zTestTSPlot.test_ax_plotc
|
C sü t dddt dddg}t ddd¡}t dd d¡}d
|
d d }| j ¡ \}}|jd d |D |f| |jdd |D |f| | ¡ \}}tjdd |D tjd} t
|
| ¡ d d
df | ¡ tjdd |D tjd} t
|
| ¡ d d
df | ¡ d S )NéØ r÷ é iÙ r# g $@g &@g à?g (@ú-r¶ )ÚfmtZlwc S s g | ] }| ¡ qS r0 ©Ú toordinalrû r0 r0 r1 rý Ò s z0TestTSPlot.test_mpl_nopandas.<locals>.<listcomp>c S s g | ] }| ¡ qS r0 r¥ rû r0 r0 r1 rý Ó s c S s g | ] }| ¡ qS r0 r¥ rû r0 r0 r1 rý × s rl r c S s g | ] }| ¡ qS r0 r¥ rû r0 r0 r1 rý Ù s ) r r6 rÐ r: r; Z plot_dater' rO rp rr r= r¸ )
|
r, ÚdatesZvalues1Zvalues2Úkwrk r. Zline1Zline2Úexpr0 r0 r1 Útest_mpl_nopandasÊ s
|
z TestTSPlot.test_mpl_nopandasc C s¼ ddl m} t ¡ d d
}|dddddd d
|
d d d dddg }| j ¡ \}}|d d
j|d |dd
j|d | ¡ \}}|| |j
|
¡ d|¡kst || |j ¡ d|¡ks¸t d S )Nr ©ÚDatetimeConverterr r# r¶ r r rR r4 r3 r÷ r3 r4 r
|
rT rU rX Ú © r r¬ rr rs r: r; r% rÖ Úconvertr r[ r+ r\ ©r, r¬ r- Z ts_irregularrk r. r r r0 r0 r1 Ú test_irregular_ts_shared_ax_xlimÜ s " z+TestTSPlot.test_irregular_ts_shared_ax_xlimc C s ddddg}ddddg}t d|d }t d|d }| j ¡ \}}|j|d
|
| ¡ \}}|jd |d | ¡ \} }
|
|| ks|t||
|
k std S ) Nr# r rE r¶ r r rÁ rR r rX Tr )r r: r; r% rÖ r+ © r, Zindex_1Zindex_2r7 r8 rk r. Ú left_beforeÚ right_beforeÚ
|
left_afterÚ right_afterr0 r0 r1 Ú test_secondary_y_non_ts_xlimí s z'TestTSPlot.test_secondary_y_non_ts_xlimc C s t dddd}t dddd}td|d}td|d}| j ¡ \}}|j|d | ¡ \}}|jd
|
|d | ¡ \} }
|
|| kst||
|
k std S ) Nú
|
2000-01-01r¶ rc r z
|
2000-01-05r# r r rX Tr )r r r: r; r% rÖ r+ r² r0 r0 r1 Ú test_secondary_y_regular_ts_xlimý s z+TestTSPlot.test_secondary_y_regular_ts_xlimc C s t dddd}td|d}| j ¡ \}}|j|d | ¡ \}}| d¡ ¡ jd |d
|
| ¡ \}}||kspt||ks|td S ) Nr¸ rä r[ r} r# r rX rc Tr ) r r r: r; r% rÖ ZresampleZmeanr+ ) r, r~ r- rk r. r³ r´ rµ r¶ r0 r0 r1 Ú#test_secondary_y_mixed_freq_ts_xlim s z.TestTSPlot.test_secondary_y_mixed_freq_ts_xlimc C sÒ ddl m} t ¡ d d
}|dddddd d
|
d d d dddg }| j ¡ \}}|d d
j|d |dd
jd|d |d d
j|d | ¡ \}}|| |j
|
¡ d|¡ks²t || |j ¡ d|¡ksÎt d S )Nr r« r r# r¶ r r rR r4 r3 r÷ r3 r4 r
|
rT rU rX Tr r r® r° r0 r0 r1 Ú"test_secondary_y_irregular_ts_xlim s " z-TestTSPlot.test_secondary_y_irregular_ts_xlimc C sZ t dddt dddg}| j ¡ \}}| |¡ tddddtddddg}| |¡ d S )Ni r# r r÷ )r r: r; r% r )r, rI rk r. r0 r0 r1 Ú test_plot_outofbounds_datetime. s
|
|
z)TestTSPlot.test_plot_outofbounds_datetimec C s dd t d¡D }tdddd}tt j t|d¡|}| j ¡ \}}|j d|d | j
|
¡ | ¡ }d
|
d |D }t|t|kst ||kst d S ) Nc S s g | ]}d |dqS )z00:00:00.0000000z0>2dr0 )rü rP r0 r0 r1 rý 8 s zATestTSPlot.test_format_timedelta_ticks_narrow.<locals>.<listcomp>r3 Ú0Únsr} rE r r5 c S s g | ] }| ¡ qS r0 ©r° rû r0 r0 r1 rý A s ) r6 rÐ r r r7 r8 rJ r: r; r% Údrawr< r+ ©r, Zexpected_labelsr~ r= r> r. r Z result_labelsr0 r0 r1 Ú"test_format_timedelta_ticks_narrow7 s
|
z-TestTSPlot.test_format_timedelta_ticks_narrowc C s ddddddddd g }t d
|
d d d }ttj t|d¡|}| j ¡ \}}|jd|d}| j ¡ |
|
¡ }dd |D }t|t|kst ||kst d S )Nz00:00:00z1 days 03:46:40z2 days 07:33:20z3 days 11:20:00z4 days 15:06:40z5 days 18:53:20z6 days 22:40:00z8 days 02:26:40z9 days 06:13:20r½ r3 ú1 dr} rE r r5 c S s g | ] }| ¡ qS r0 r¿ rû r0 r0 r1 rý Y s z?TestTSPlot.test_format_timedelta_ticks_wide.<locals>.<listcomp>) r r r6 r7 r8 rJ r: r; r% rÀ r< r+ rÁ r0 r0 r1 Ú test_format_timedelta_ticks_wideE s& ÷
|
z+TestTSPlot.test_format_timedelta_ticks_widec C s¶ t tdtddd}| j ¡ \}}t|j|d tdddd}t tj t
|
|¡|}| j ¡ \}}t|j|d tddd d}t tj t
|
|¡|}| j ¡ \}}t|j|d d S )
|
Nr Z1dayrå rX z1 day 2 hr 30 min 10 sr3 rà r} z1 ns) r r9 r r: r; r$ r% r6 r7 r8 rJ )r, rx rk r. r r0 r0 r1 Útest_timedelta_plot] s z TestTSPlot.test_timedelta_plotc C s` t dddd}|}t ddd¡}t ddd¡d d d
}| j ¡ \}}|j||g||gd d S )
|
Nr r3 r r} r r# g¹?r" )Úweights)r r6 rÐ r: r; Úhist)r, r~ rT Zw1Zw2rk r. r0 r0 r1 Ú test_histo s zTestTSPlot.test_histc C s t dddgtdddtdddtdddgd }t dddgtd
|
ddtd ddtd ddgd }| j ¡ \}}|j|d |j|d |j|d d S )Nr# r rE rú r÷ r¢ r¡ iÕ r iÍ iÓ r¡ rX )r r r: r; r% )r, r7 r8 rk r. r0 r0 r1 Útest_overlapping_datetimex s$
|
|
|
ýþ
|
|
|
ýþ z$TestTSPlot.test_overlapping_datetimez1GH9053 matplotlib does not use ax.xaxis.converterc C sp t tj d¡tdddd}| ¡ }t t¡ |j|j |j
|
dd W 5 Q R X |j \}}t | ¡ | ¡ ¡ d S )Nr3 z
|
1970-01-02rå r r5 )ru )r r6 r7 r8 r r% rr Zassert_produces_warningÚDeprecationWarningr rI r r= r¸ )r, rx r. Úl1Úl2r0 r0 r1 Ú test_add_matplotlib_datetime64 s
|
z)TestTSPlot.test_add_matplotlib_datetime64c C s~ t tj d¡ dd¡ddgd}tdddd |d
|
< | j ¡ \}}|jd
|
d|d | j ¡ |
|
¡ d }d}| ¡ |kszt d S ) Nr r3 r rT rU r- z
|
2018-01-01rc r} r )rT rU rÅ ) r r6 r7 Z RandomStaters r r: r; ZscatterrÀ r< r° r+ )r, r= r> r. r? rÛ r0 r0 r1 Ú"test_matplotlib_scatter_datetime64 s
|
z-TestTSPlot.test_matplotlib_scatter_datetime64c C s t dddg}t|dddgd}|jdd d}| j|d
|
d t ddd g}t|dddgd}|jdd d}| j|d d | d¡jd dd}| j|d d | d¡jd dd}| j|d
|
d |jdd ddd}| j|d d |jdd ddd}| j|d
|
d d S )Nz
|
2020-05-01z
|
2020-05-02z
|
2020-05-03r# r rE )rT rU rT rU r )Zxrotz
|
2020-05-04rÍ T)rU Z use_indexF)rT rU r; rY )r r r% Z_check_ticks_propsZ set_index)r, rT r= rà r0 r0 r1 Útest_check_xticks_rotª s z TestTSPlot.test_check_xticks_rotN)_Ú__name__Ú
|
__module__Ú __qualname__r\ ÚmarkÚfilterwarningsr2 r@ rN rR r` Z parametrizerl rm rv ry r r r r r r r r r r r r r£ Zxfailrµ rÀ rà rÉ rÊ rË rÒ rÔ rá râ rñ rö rø rù r Zslowr r r r r$ r% ÚtdZskip_if_no_scipyr) r+ r0 r1 r; r@ rC rD rJ rM rQ rW rZ r] r_ rq ru rw rx r r r r r r r rª r± r· r¹ rº r» r¼ r rÄ rÅ rÈ rÉ rÍ rÎ rÏ r0 r0 r0 r1 r - sì
|
|
|
ÿ
|
|
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ
|
ÿ
|
|
(
|
|
|
3
|
[ ) [
|
|
|
r c O sB dd l m} | ¡ }z| ¡ | d¡}| d| ¡ ¡}t|dd } | ||}
|
|
|
d k s\t | d| ¡ ¡}|d k r¢|j
|
j } t | t r| j} | d kr¢|j | ks¢t |d k rÀ| d krÀ|j |ksÀt | d¡}||d< | ||}
|
|
|
d k sèt tjdd} | | ¡ W 5 Q R X tjdd} t || ¡ W 5 Q R X W 5 | |¡ X d S )Nr r r. r éÔ T)Zreturn_filelike)Zmatplotlib.pyplotZpyplotZgcfr[ Zclfr ÚpopZgcaÚgetattrr+ r r r r r rr Z ensure_cleanZsavefigÚpickleÚdump) Úfr ZseriesÚargsÚkwargsr: r> r. Zorig_axZ orig_axfreqr ZdfreqÚpathr0 r0 r1 r$ Å s6
|
|
|
|
|
r$ )NN),Ú__doc__r r r r rÙ Únumpyr6 r\ Zpandas._libs.tslibsr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrÕ Zpandasr r r
|
r r r r Zpandas._testingZ_testingrr Z pandas.core.indexes.datetimesr r r Zpandas.core.indexes.periodr r r Z pandas.core.indexes.timedeltasr Z pandas.tests.plotting.commonr Zpandas.tseries.offsetsr Zskip_if_no_mplr r$ r0 r0 r0 r1 Ú<module> s6 $ "
|
