a ²<b{×ã@sþdZddlmZmZmZmZddlZddlZddlZddl m Z mZddlm mZddlmZmZmZmZmZmZmZddlmZddlmZmZmZddlm Z m!Z!m"Z"ddl#m$Z$dd l%m&Z&dd l'm(Z(ej)j*Z+ej,Gdd„de&ƒƒZ-dd d„Z.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äeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zej dgd¢¡dd„ƒZdd „Zd!d"„Zej dgd¢¡d#d$„ƒZd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d6„Zd7d8„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.dUdV„Z/dWdX„Z0e1j2dYdZ„ƒZ3d[d\„Z4d]d^„Z5d_d`„Z6dadb„Z7dcdd„Z8dedf„Z9dgdh„Z:didj„Z;dkdl„Zdqdr„Z?dsdt„Z@dudv„ZAdwdx„ZBdydz„ZCd{d|„ZDd}d~„ZEdd€„ZFdd‚„ZGdƒd„„ZHd…d†„ZId‡dˆ„ZJd‰dŠ„ZKd‹dŒ„ZLddŽ„ZMdd„ZNd‘d’„ZOd“d”„ZPd•d–„ZQd—d˜„ZRd™dš„ZSd›dœ„ZTddž„ZUdŸd „ZVd¡d¢„ZWd£d¤„ZXejjYd¥d¦d§d¨„ƒZZd©dª„Z[d«d¬„Z\dS)®Ú TestTSPlotcCs€t ||¡gd¢|_dd„|jDƒ}dd„|Dƒ|_dd„|Dƒ|_gd¢}dd„|Dƒ}dd„|Dƒ|_d d„|Dƒ|_dS) N)ÚSÚTÚHÚDÚWÚMÚQÚAcSsg|]}td|dd‘qS©ú 12/31/1999éd©ÚfreqÚperiods)r©Ú.0Úx©r*úw/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/plotting/test_datetimelike.pyÚ 5óz+TestTSPlot.setup_method..cSs"g|]}ttj t|ƒ¡|ƒ‘qSr*©rÚnpÚrandomÚrandnÚlenr'r*r*r+r,6r-cSs,g|]$}ttj t|ƒd¡|gd¢d‘qS©é©r ÚBÚC©ÚindexÚcolumns©rr/r0r1r2r'r*r*r+r,7sÿ) rrrrrrzQ-DECr Z1B30MincSsg|]}td|dd‘qSr!)rr'r*r*r+r,=r-cSs"g|]}ttj t|ƒ¡|ƒ‘qSr*r.r'r*r*r+r,>r-cSs,g|]$}ttj t|ƒd¡|gd¢d‘qSr3r;r'r*r*r+r,?sÿ)rÚsetup_methodr%Ú period_serÚ period_dfÚdatetime_serÚdatetime_df)ÚselfÚmethodÚidxr%r*r*r+r<1s þþzTestTSPlot.setup_methodcCst ¡dS©N)ÚtmÚclose)rArBr*r*r+Úteardown_methodDszTestTSPlot.teardown_methodcCs°|}tddd|d}tddg|d}t d¡nt|jƒ| ¡}t| ¡ƒd ¡}|dj |dj fd ksrJ‚|d j |d j fdksŽJ‚Wdƒn1s¢0YdS)Nú1/1/2011ér)r&r%Útzgg@g„t@©r9r©rréÿÿÿÿ)ér)rrrEÚassert_produces_warningÚ_check_plot_worksÚplotÚlistÚ get_linesÚ get_xdataÚhourÚminute)rAZtz_aware_fixturerJr9ÚtsÚaxÚxdatar*r*r+Útest_ts_plot_with_tzGs zTestTSPlot.test_ts_plot_with_tzcCs`ttj dd¡tdƒd}|j ¡\}}|jd|d| ¡| ¡D]}| ¡dksFJ‚qFdS)Né é rKrI©ZfontsizerX)rr/r0r1ÚrangeÚpltÚsubplotsrQÚget_xticklabelsZget_yticklabelsZget_fontsize)rAÚdfÚfigrXÚlabelr*r*r+Útest_fontsize_set_correctlyTs z&TestTSPlot.test_fontsize_set_correctlycCsÂtdddd}t|jdd}ttj t|ƒd¡|d}t|j ƒ|dd … |d d…¡}ttj t|ƒd¡|d}t|j ƒtdd dd}t|jdd}ttj t|ƒd¡|d}t|j ƒdS)Nú1/1/1987ZMSr#r$©r%r4rKré(é-écz2008-1-1 00:15:00Z15Tr[)rrÚvaluesrr/r0r1r2rPrQÚunion)rArCrbÚdf2r*r*r+Útest_frame_inferred\s zTestTSPlot.test_frame_inferredcCs:t gd¢¡}t||d}t|j|dt|j|ddS)N©rNrIr4rK)Zxerr)Zyerr)r/ÚarrayrrPrQ)rAÚiÚar*r*r+Útest_is_error_nozeroindexosz$TestTSPlot.test_is_error_nozeroindexcCs¢tdddd}tgd¢gd¢dœ|ƒ}|j ¡\}}|j|dt| ¡ƒd ksTJ‚|j |¡d }tj t |d|d ¡Wdƒn1s”0YdS)Nrfr r4r$)r)ÚyÚzro)r r6©rXrNzno numeric data to plot©Úmatch)rrr_r`rQr2rSrFÚpytestÚraisesÚ TypeError)rArCrbrcrXÚmsgr*r*r+Útest_nonnumeric_excludevsz"TestTSPlot.test_nonnumeric_excludecCs†|j ¡\}}t ¡}|jD]}t|j|dq|jD]}t|j|dq6|j ¡\}}|jd|dd}|| ¡d ¡ks‚J‚dS)NrvÚk©ÚstylerX)çrrrNr) r_r`rEÚmakeTimeSeriesr=rPrQr?rSÚ get_color)rAÚ_rXrWÚsÚcolorr*r*r+Útest_tsplotƒs zTestTSPlot.test_tsplotcCs”t ¡}d}tjt|d|jdddWdƒn1s>0Y|jdd}tjt|d|jdddWdƒn1s†0YdS)NzCannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbolrwzb-z#000099)r€r†T)Zdrop)rEr‚ryrzÚ ValueErrorrQZreset_index)rArWr|r…r*r*r+Útest_both_style_and_color“sÿ,z$TestTSPlot.test_both_style_and_colorcCsVddg}|D]D}|j ¡\}}tdd|d}ttj t|ƒ¡|ƒ}t|j |dqdS)NÚmsÚusú1/1/2012r#©r&r%rv) r_r`rrr/r0r1r2rPrQ)rAZfreaksr%r„rXÚrngÚserr*r*r+Útest_high_freq¢szTestTSPlot.test_high_freqcCsnddlm}|ddƒdusJ‚|ddƒdks0J‚|tddƒdƒtddƒjksPJ‚|ddƒtd dƒjksjJ‚dS) Nr)Ú get_datevalueréÃr rz1987-12rfz1987-1-1)Ú%pandas.plotting._matplotlib.converterr‘rÚordinal)rAr‘r*r*r+Útest_get_datevalueªs zTestTSPlot.test_get_datevaluecCszdd„}tdtddddd}|j ¡\}}|j|d ||d ƒtdtddddd}|j|d ||dƒt ¡dS) NcSs>| ¡d}| ¡dj}| ¡d}|| ||¡ks:J‚dS)Nr)rSrTr”Ú get_ydataZformat_coord)rXZexpected_stringÚ first_lineZfirst_xZfirst_yr*r*r+Úcheck_format_of_first_point³szITestTSPlot.test_ts_plot_format_coord..check_format_of_first_pointrNz 2014-01-01r4zA-DECrrKrvzt = 2014 y = 1.000000rzt = 2014-01-01 y = 1.000000)rrr_r`rQrErF)rAr˜Zannualr„rXZdailyr*r*r+Útest_ts_plot_format_coord²s z$TestTSPlot.test_ts_plot_format_coordcCs |jD]}t|j|jjƒqdSrD)r=rPrQr9r%©rAr…r*r*r+Útest_line_plot_period_seriesÅs z'TestTSPlot.test_line_plot_period_seriesÚfrqncy)Z1SZ3SZ5TZ7HZ4DZ8WZ11MZ3AcCs:td|dd}ttj t|ƒ¡|ƒ}t|j|jj j ƒdS)Nr"r#r$)rrr/r0r1r2rPrQr9r%Ú rule_code)rArœrCr…r*r*r+Ú test_line_plot_period_mlt_seriesÉsz+TestTSPlot.test_line_plot_period_mlt_seriescCs"|jD]}t|j|jjjƒqdSrD)r?rPrQr9r%rršr*r*r+Útest_line_plot_datetime_seriesÓs z)TestTSPlot.test_line_plot_datetime_seriescCs |jD]}t|j|jjƒqdSrD)r>rPrQr9r%)rArbr*r*r+Útest_line_plot_period_frame×s z&TestTSPlot.test_line_plot_period_framecCsRtd|dd}ttj t|ƒd¡|gd¢d}|j |jjj ¡j}t |j|ƒdS)Nr"r#r$r4r5r8)rrr/r0r1r2r9Úasfreqr%rrPrQ)rArœrCrbr%r*r*r+Útest_line_plot_period_mlt_frameÛs z*TestTSPlot.test_line_plot_period_mlt_framecCs0|jD]$}|j |jjj¡j}t|j|ƒqdSrD)r@r9Ú to_periodr%rrPrQ)rArbr%r*r*r+Útest_line_plot_datetime_frameçs z(TestTSPlot.test_line_plot_datetime_framecCsN|jD]B}t|jtt |j¡ƒƒ}t|j|jj ƒ|gd¢}t|jƒqdS)N)rr4éé) r?rrkr r/Zasarrayr9rPrQZ inferred_freq)rArr*r*r+Útest_line_plot_inferred_freqìs z'TestTSPlot.test_line_plot_inferred_freqcCsf|j ¡\}}tddƒ}ttt|ƒƒ|d}t|dd…|dd…gƒ}|j|dt|dƒrbJ‚dS)Nz2001-1-1z 2001-1-10rKr4r¥rvr%) r_r`rrr^r2rrQÚhasattr)rAr„rXrŽrWr*r*r+Útest_fake_inferred_businessôs z&TestTSPlot.test_fake_inferred_businesscCsLt ¡}t|jƒt|jdddd}ttj t |ƒ¡|d}t|jƒdS)NrZBQSr[r$rK)rEr‚rPrQrr9rr/r0r1r2)rArÚdrr*r*r+Útest_plot_offset_freqüs z TestTSPlot.test_plot_offset_freqcCsLttdddƒtdddƒtdddƒgƒ}ttj t|ƒ¡|d}t|jƒdS)NéÐrNr¦érK) r rrr/r0r1r2rPrQ)rArªrr*r*r+Ú test_plot_multiple_inferred_freqs&z+TestTSPlot.test_plot_multiple_inferred_freqc Csºddlmmm}tdddd}ttj t |ƒd¡|d}|j ¡\}}|j|d| ¡}| ¡}| ¡}t||ƒD]8\} } | | ¡ d ¡}t| ¡ƒ}t |ƒr|||ks|J‚q|dS) Nrz2012-6-22 21:59:51.960928ÚLéôr$rIrKrvú%H:%M:%S.%f)r“ÚplottingÚ_matplotlibÚ converterrrr/r0r1r2r_r`rQÚ get_xaxisZget_ticklocsÚget_ticklabelsÚzipZ _from_ordinalÚstrftimeÚstrÚget_text) rAÚconvrCrbr„rXZaxisZtlocsZtlabelsÚlocrdÚxpÚrsr*r*r+Útest_uhf szTestTSPlot.test_uhfc Cs tdddd}ttj t|ƒd¡|d}|jgd¢}|j ¡\}}|j |dt | ¡d ¡dd…d fƒ ¡}d }t |dd…||d|g¡dk ¡s¤J‚|j ¡\}}| ¡}|j t¡|_|j |dt | ¡d ¡dd…d fƒ ¡}t |dd…|¡dk ¡sJ‚dS) Nz2012-6-22 21:59:51rr#r$rIrK)rrNr4érvrg)QÎ ÈEè>rNg:Œ0âŽyE>)rrr/r0r1r2Úilocr_r`rQrrSÚ get_xydataÚdiffÚfabsÚallÚcopyr9ÚastypeÚobject) rArCrbÚirregr„rXZdiffsÚsecrmr*r*r+Ú test_irreg_hfs$,$zTestTSPlot.test_irreg_hfcCslt ¡}|gd¢}|j ¡\}}|j|d}|dus:J‚t| ¡d ¡|jƒD]\}}||ksRJ‚qRdS)N)rrNrIérvr) rEr‚r_r`rQr·rSrTr9)rArr„rXÚretr¾r½r*r*r+Ú"test_irregular_datetime64_repr_bug.s z-TestTSPlot.test_irregular_datetime64_repr_bugcCsnt ¡}|j ¡\}}|j|d| ¡d ¡d|jdjksFJ‚| ¡d ¡}t |djdksjJ‚dS)NrvrrL©Údatar6)rEZmakePeriodSeriesr_r`rQrSrÂr9r”rTrÚfreqstr©rAÚbtsr„rXrCr*r*r+Útest_business_freq:s$zTestTSPlot.test_business_freqcCs€t d¡ d¡}| d¡}|j ¡\}}|j|d| ¡d ¡d|j dj ksXJ‚| ¡d ¡}t|dj dks|J‚dS)Ni,ZBMrrvrrLrÏ)rEr‚r¡r£r_r`rQrSrÂr9r”rTrrÑ)rArÓrWr„rXrCr*r*r+Útest_business_freq_convertBs $z%TestTSPlot.test_business_freq_convertcCs€tƒ}t d¡ |¡}|j ¡\}}|j|d| ¡d ¡}d}t j t|dt|dWdƒn1sr0YdS)Nr¥rvrz)freq not specified and cannot be inferredrwrÏ) rrEr‚r¡r_r`rQrSrTryrzrˆr)rAr%rÓr„rXrCr|r*r*r+Útest_freq_with_no_period_aliasKsz)TestTSPlot.test_freq_with_no_period_aliascCsftddddtdd}tt d¡|d}|j ¡\}}|j|d| ¡d ¡}t |ƒjrbJ‚dS) Nz 2012-12-20érré©ÚminutesrKrvr)rrrr/Úaranger_r`rQrSrTr Ú is_normalized)rArCrbr„rXr¾r*r*r+Útest_nonzero_baseWszTestTSPlot.test_nonzero_basecCsTtdt ¡iƒ}|j ¡\}}|j|d| ¡d ¡}t |j ¡t|ƒ¡dS)Nrrrvr)rrEr‚r_r`rQrSrTÚassert_index_equalr9r£rrÒr*r*r+Útest_dataframe`s zTestTSPlot.test_dataframecsž‡fdd„}t ¡}ˆj ¡\}}|j|d||ƒˆj ¡\}}t||ddœƒ}|j|d||ƒt||ddœƒ}|jdd}|D]}||ƒqŒdS)NcsH| ¡}| |dd|dd¡| ¡}|d|ddksDJ‚|d|ddks\J‚td|jƒtd|jƒf}| dd¡| ¡}t|dƒ|djks¢J‚t|dƒ|djks¼J‚td|jƒtd|jƒf}| tdddƒtdddƒ¡| ¡}t|dƒ|djksJ‚t|dƒ|djks0J‚| ¡}ˆj |¡dS) Nrr¥rNr[ú1/1/2000z4/1/2000r¬rÀ) Úget_xlimÚset_xlimrr%Úintr”rÚ get_figurer_rF)rXZxlimÚresultÚexpectedrc©rAr*r+Ú_tesths"z*TestTSPlot.test_axis_limits.._testrvrN©rrÚbT)r`)rEr‚r_r`rQr)rArèrr„rXrbÚaxesr*rçr+Útest_axis_limitsgszTestTSPlot.test_axis_limitscCs¬ddlmmm}| tdƒ¡|jks0J‚| tdƒ¡|jksHJ‚| tdƒ¡|jks`J‚| tdƒ¡|jksxJ‚| tdƒ¡|j ksJ‚| tdƒ¡|jks¨J‚dS)Nrr6rrrr r) r“r²r³r´Z get_finderrZ _daily_finderZ_monthly_finderZ_quarterly_finderZ_annual_finder)rAr»r*r*r+Útest_get_finderszTestTSPlot.test_get_findercCsægd¢}tdddjgt|ƒ}}g}g}|D]˜}td|d}ttj t|ƒ¡|ƒ}|j ¡\} } |j | d| ¡}| | ¡d¡| ¡\}} | |d| ¡| | ¡d¡|j | ¡¡q0||ksÖJ‚||ksâJ‚dS) N)r[rhéüii¶i¾ é'z1999-1-1r6rg©r&rvrçÍÌÌÌÌÌì?)rr”r2rrr/r0r1r_r`rQrµÚappendÚget_majorticklocsrárârFrä)rAZday_lstÚxpl1Úxpl2Úrs1Úrs2ÚnrŽrr„rXÚxaxisÚvminÚvmaxr*r*r+Útest_finder_daily™s"zTestTSPlot.test_finder_dailycCsìddg}tdƒjgt|ƒ}}g}g}|D]¢}tdt|dƒdd}ttj t|ƒ¡|ƒ}|j ¡\} } |j| d| ¡}| | ¡d ¡| ¡\}} | |d | ¡| | ¡d ¡|j | ¡¡q,||ksÜJ‚||ksèJ‚dS)Ng@rÚ1988Q1Ú1987Q2rÀrrrvrrñ©rr”r2rrãrr/r0r1r_r`rQrµròrórárârFrä©rAZyrsrôrõrör÷rørŽrr„rXrùrúrûr*r*r+Útest_finder_quarterly¯s"z TestTSPlot.test_finder_quarterlycCsìgd¢}tdƒjgt|ƒ}}g}g}|D]¢}tdt|dƒdd}ttj t|ƒ¡|ƒ}|j ¡\} } |j| d| ¡}| | ¡d¡| ¡\}} | |d | ¡| | ¡d¡|j | ¡¡q,||ksÜJ‚||ksèJ‚dS) N)gffffffò?g@rÀrzJan 1988rþérrrvrrñrÿrr*r*r+Útest_finder_monthlyÅs"zTestTSPlot.test_finder_monthlycCsntdddd}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d| ¡}| ¡d}tddƒj}||ksjJ‚dS)Nrýi rrrvrZ1989Q1) rrr/r0r1r2r_r`rQrµrórr”)rArŽrr„rXrùr¾r½r*r*r+Útest_finder_monthly_longÛsz#TestTSPlot.test_finder_monthly_longc Csœgd¢}dd„|Dƒ}g}dD]l}td|dd}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d| ¡}| | ¡d ¡|j | ¡¡q||ks˜J‚dS) N)r’iÄéÆréËiäiizcSsg|]}t|ddj‘qS)r rg)rr”r'r*r*r+r,çr-z1TestTSPlot.test_finder_annual..)r¥r[éé1rjéÇiWiéZ1987r rrvr)rrr/r0r1r2r_r`rQrµròrórFrä) rAr½r¾ZnyearsrŽrr„rXrùr*r*r+Útest_finder_annualåszTestTSPlot.test_finder_annualc Cstd}tdd|d}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d| ¡}| ¡d}tdddj}||kspJ‚dS)Ni@ú1/1/1999ZMinr$rvrrg© rrr/r0r1r2r_r`rQrµrórr”) rAZnminutesrŽrr„rXrùr¾r½r*r*r+Útest_finder_minutelyôszTestTSPlot.test_finder_minutelyc Cstd}tdd|d}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d| ¡}| ¡d}tdddj}||kspJ‚dS)Nérrr$rvrrgr) rAZnhoursrŽrr„rXrùr¾r½r*r*r+Útest_finder_hourlyszTestTSPlot.test_finder_hourlyc Csdt ¡}tj|dd…<|j ¡\}}|j|d| ¡}t|ƒdksHJ‚|d}| ¡}|j sd|js|tjj |t|ƒtjd}t|tjjj ƒsJ‚|j}|dd…df ¡s®J‚|j | ¡¡t ¡}|gd¢}tj|dd…<|j ¡\}}|j|d}| ¡}t|ƒdksJ‚|d}| ¡}|j s4|jsLtjj |t|ƒtjd}t|tjjj ƒsbJ‚|j}|dd…df ¡s‚J‚|j | ¡¡gd¢}ttj t|ƒ¡|ƒ} tj| dd…<|j ¡\}}| j|d| ¡}t|ƒdksòJ‚|d}| ¡}|j s|js*tjj |t|ƒtjd}t|tjjj ƒs@J‚|j}|dd…df ¡s`J‚dS) Nr¥érvrNr©ÚmaskZ fill_value) rrNrIr¥rÌr\réérI)rEr‚r/Únanr_r`rQrSr2rÂÚmpl_ge_3_0_0Úmpl_ge_2_2_3ÚmaÚMaskedArrayr Ú isinstanceÚcorerrÅrFrärr0r1) rArWr„rXÚlinesÚlinerÐrrCrr*r*r+Ú test_gapssVzTestTSPlot.test_gapsc Cst ¡}tj|dd…<|j ¡\}}|j|dt|jd|jddd}t tj t|ƒ¡|ƒ}|jdd | ¡}t|ƒd ks„J‚t|j ¡ƒd ksšJ‚|d}| ¡}|js¶|jsÎtjj|t|ƒtjd}t|tjjjƒsâJ‚|j} | dd…d f ¡sJ‚dS)Nr¥rrvrrMZ12hrgT©Úsecondary_yrNr)rEr‚r/rr_r`rQrr9rr0r1r2rSÚright_axrÂrrrrr rrrrÅ) rAÚlowr„rXÚidxhr…rrrÐrr*r*r+Útest_gap_upsampleBs"zTestTSPlot.test_gap_upsamplec Csfttj d¡ƒ}ttj d¡ƒ}|j ¡\}}|jdd}t|dƒsHJ‚t|dƒrVJ‚| ¡}| ¡d}t| ¡| ¡ƒ}t ||¡| ¡ ¡dksœJ‚|d ¡ ¡r°J‚|j |¡|j ¡\}} |j| d| ¡ ¡|jksìJ‚|j | ¡¡| ¡}|jdd} | ¡ ¡s"J‚t|dƒr2J‚t|dƒsBJ‚t| dƒsRJ‚t| dƒrbJ‚dS) Nr[TrÚleft_axr!rÚrightrv)rr/r0r1r_r`rQr¨Úget_axesrSr–rTrEÚassert_series_equalÚ get_yaxisÚget_ticks_positionÚget_visiblerFÚdefault_tick_positionrä) rArÚser2rcr„rXrërr½Úax2r*r*r+Útest_secondary_yXs0zTestTSPlot.test_secondary_ycCs<tddd}ttj d¡|ƒ}ttj d¡|ƒ}|j ¡\}}|jdd}t|dƒsXJ‚t|dƒrfJ‚| ¡}| ¡d}t| ¡| ¡ƒ ¡} t || ¡| ¡ ¡d ks°J‚|d ¡ ¡rÄJ‚|j |¡|j ¡\}} |j| d | ¡ ¡|jksJ‚|j | ¡¡| ¡}|jdd} | ¡ ¡s8J‚dS)Nràr[rðTrr%r!rr&rv)rrr/r0r1r_r`rQr¨r'rSr–rTZto_timestamprEr(r)r*r+rFr,rä)rArCrr-rcr„rXrërr½r.r*r*r+Útest_secondary_y_tsts*zTestTSPlot.test_secondary_y_tscCsnttj d¡ƒ}|j ¡\}}|jdd|d}t|dƒs)rEr‚Úto_framerÁr_r`rQrSrrTÚequalsr9r£rár”rCr*r*r+Ú test_mixed_freq_regular_first_dfÐs z+TestTSPlot.test_mixed_freq_regular_first_dfcCs¨t ¡ ¡}|jgd¢dd…f}|j ¡\}}|jd|d|j|dt|dƒrXJ‚| ¡}|d ¡}t ||j t ¡j¡|d ¡}t ||j t ¡j¡dSrJ)rEr‚rOrÁr_r`rQr¨rSrTrKr9rÇrÈrkrLr*r*r+Ú"test_mixed_freq_irregular_first_dfász-TestTSPlot.test_mixed_freq_irregular_first_dfcCs˜tdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|j|d| ¡D]}t | ¡djdksvJ‚qvdS) NrémrrrrrvrÏ© rrr/r0r1r2r_r`rQrSrrTr%©rAr#ÚidxlÚhighr"r„rXrr*r*r+Útest_mixed_freq_hf_firstïsz#TestTSPlot.test_mixed_freq_hf_firstcCsˆtdddd}tj d¡}t||d}| d¡ ¡}|j ¡\}}|j |d}|j d |d |j d ¡d|j d ¡dks„J‚dS) Nz2012-01-01 13:00ú 2012-01-02rrgrrKrrvÚrrrrN)rr/r0r1rr¡Zinterpolater_r`rQrrT)rAZts_indZts_datarWZts2r„rXr*r*r+Útest_mixed_freq_alignmentúsz$TestTSPlot.test_mixed_freq_alignmentc Cs^tdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|jd|d|jd|d| ¡D]}t | ¡d jdkszJ‚qz| ¡}t|jƒd ks²J‚|j | ¡¡tdddd}tdd dd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|j|d| ¡D] }t | ¡d jdks8J‚q8dS)NrrSrrrrT)ZlegendrXrÏrIéðrrÀrrv)rrr/r0r1r2r_r`rQrSrrTr%Ú get_legendZtextsrFrä) rAr#rVrWr"r„rXrÚlegr*r*r+Útest_mixed_freq_lf_firsts*z#TestTSPlot.test_mixed_freq_lf_firstcCsbt ¡}|gd¢}tdddd}ttj t|ƒ¡|ƒ}|j ¡\}}|j |d|j |ddS)N)rrNrIr4rÀr¥r¦rÌér\r[rééééz1/3/2000rØr6rrv)rEr‚rrr/r0r1r2r_r`rQ)rArWrÉrŽZpsr„rXr*r*r+Útest_mixed_freq_irreg_period sz'TestTSPlot.test_mixed_freq_irreg_periodcCstdddd}|dd… |dd…¡}ttt|ƒƒ|ƒ}ttt|ƒƒ|ƒ}|jjddd\}\}}|j|d |j|d |jdksŠJ‚|jdks˜J‚|j d ¡d|j d ¡dksÀJ‚|j ¡\}}| ¡}|j|d |j|d |j d ¡d|j d ¡dksJ‚dS)Nú 2015-01-01r4rrrNrIT)ZnrowsÚsharexrvrrL)rrlrr^r2r_r`rQr%rrÂZtwinx)rArFrGrDrErcZax1r.r*r*r+Útest_mixed_freq_shared_ax)s(z$TestTSPlot.test_mixed_freq_shared_axcCs‚|j ¡\}}tdtdgƒ}ttt|ƒƒ|ƒ}|j|d| ¡d ¡}|j ¡t|ƒ ¡ksdJ‚t|ƒ ¡|j ¡ks~J‚dS)Nrfz 2015-01-03rvr) r_r`rr rr^r2rQrSrTr9ÚminÚmax)rAr„rXZdtir…rYr*r*r+Útest_nat_handlingJszTestTSPlot.test_nat_handlingcCsštdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|j|d| ¡D]}t | ¡dj|jksvJ‚qvdS) Nré4rrrrrvrÏrTrUr*r*r+Útest_to_weekly_resamplingVsz$TestTSPlot.test_to_weekly_resamplingcCsútdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|j|d| ¡j tj¡}tj gd¢tjd }| ¡D]P} t| ¡d j|jksºJ‚| jdd} t| ƒdkràt | |¡qœt | |¡qœt ¡dS) Nrrlrrrrrv©iêiïiói÷iûiiii iii©ZdtyperÏF©Úorig)rrr/r0r1r2r_r`rQr£Úasi8rÇÚfloat64rprSrrTr%rErKrF)rAr#rVrWr"r„rXZ expected_hZ expected_lrrYr*r*r+Útest_from_weekly_resamplingas&þz&TestTSPlot.test_from_weekly_resamplingcCs,tdddd}tdddd}ttj t|ƒd¡|gd¢d }ttj t|ƒd¡|gd¢d }d D]Z\}}|j ¡\}}|j|d|d|j|d|dtj gd ¢tj d} tjt| ƒtj d} tdƒD]\}|j |}t| ¡ƒj|jksîJ‚t |jdd| ¡| ||j7} t |jdd| ¡qÈ| ¡j tj ¡} tjt| ƒtj d} tdƒD]f}|j d|}t| ¡dj|jks‚J‚t |jdd| ¡| ||j7} t |jdd| ¡qTq`d D]b\}}|j ¡\}}|j|d|d|j|d|d| ¡j tj ¡} tjt| ƒtj d} tdƒD]b}|j |}t| ¡dj|jksRJ‚t |jdd| ¡| ||j7} t |jdd| ¡q(tj gd ¢tj d} tjt| ƒtj d} tdƒD]f}|j d|} t| ¡dj|jksêJ‚t | jdd| ¡| ||j7} t | jdd| ¡q¼qÂdS)Nrrlrrrrr4)rrNrIr8))rÚarea)rurT)r2ZstackedrXrnroFrprÏ)rrr/r0Úrandr2r_r`rQrprsZzerosr^rrrTr%rErKrkr–r£rrrÇ)rAr#rVrWr"Zkind1Zkind2r„rXZ expected_xZ expected_yrqrrr*r*r+Ú$test_from_resampling_area_line_mixedxsd ñ ñz/TestTSPlot.test_from_resampling_area_line_mixedcCstdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|j|dt| ¡ƒdks‚J‚| ¡D]}t | ¡d jd ksŠJ‚qŠt ¡|j ¡\}}|j|d|j|dt| ¡ƒdksêJ‚| ¡D]}t | ¡d jd ksòJ‚qòdS)Nz2014-07-01 09:00ré2r$Z100Lr°rvrIrÏr¯)rrr/r0r1r2r_r`rQrSrrTr%rErFrUr*r*r+Ú"test_mixed_freq_second_millisecondÓs"z-TestTSPlot.test_mixed_freq_second_millisecondcCs¢tdddƒtdddƒtdddƒg}ttj t|ƒd¡t|tdƒ}t|j ƒt ddd }|gd ¢ t¡}ttj t|ƒd¡|ƒ}|j ¡\}}t|j |ddS)Nr¬rNr¥rr4roràr[rð)rrIr¥r\rv)rrr/r0r1r2r rÈrPrQrrÇr_r`)rArCrbr„rXr*r*r+Útest_irreg_dtypesês" zTestTSPlot.test_irreg_dtypescstddddddƒ‰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 ¡} | |ksœJ‚qœdS)NrNr4rØrrcs"g|]}ˆtt|ƒd ¡‘qS©rÙ©rrãrr'©Útr*r+r,úr-z(TestTSPlot.test_time..rérKrvé<ú%H:%M:%Sú%H:%M)rr/r0ÚrandintÚcumsumrprr1r2r_r`rQÚ get_xticksrar·Údivmodrãrºrr¸©rAÚdeltasrWrbrcrXÚticksÚlabelsÚlÚmr…Úhr¾r½r*r}r+Ú test_time÷s&"ÿzTestTSPlot.test_timecs¾tddddddƒ‰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 ¡} | |ksœJ‚qœ| dd¡| ¡}| ¡}t||ƒD]~\‰}ttˆƒdƒ\} } t| dƒ\}} | ¡}t|ƒdkr:| dkr–t|| | ƒ d¡} nt|| | ƒ d ¡} | |ks:J‚q:dS)NrNr4rØrrcs"g|]}ˆtt|ƒd ¡‘qSr{r|r'r}r*r+r,r-z4TestTSPlot.test_time_change_xlim..rérKrvrr€rz1:30z5:00)rr/r0r‚rƒrprr1r2r_r`rQr„rar·r…rãrºrr¸râr†r*r}r+Útest_time_change_xlims>"ÿ z TestTSPlot.test_time_change_xlimcs‚tddddddƒ‰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.rérKrvrg€„.Aièr±éýÿÿÿr€r)rr/r0r‚rƒrprr1r2r_r`rQr„rar·r…rãÚroundrºrr¸)rAr‡rWrbrcrXrˆr‰rŠr‹r…r‹rŒr¾r½r*r}r+Útest_time_musec8s0"ÿ zTestTSPlot.test_time_museccCsÚtdddd}tdddd}ttj t|ƒ¡|ƒ}ttj t|ƒ¡|ƒ}|j ¡\}}|j|d|jd|d }| ¡D]}t | ¡ƒjdksxJ‚qxt |d ƒs¢J‚t |dƒr°J‚|j ¡D]}t | ¡ƒjdksºJ‚qºdS)NrrSrrrrrvT©r rXr%r!)rrr/r0r1r2r_r`rQrSrrTr%r¨r%rUr*r*r+Útest_secondary_upsampleWsz"TestTSPlot.test_secondary_upsamplecCsZ|j ¡}| d¡}t ¡}|jddg|d| ¡}t| ¡ƒdksJJ‚| ¡d ¡dksbJ‚| ¡d ¡d kszJ‚| ¡d ¡dks’J‚| ¡d ¡d ksªJ‚|j ¡dus¼J‚tƒ}| ¡D]}| | ¡¡qÊt|ƒdksîJ‚|j |¡|j ¡}| d¡}|jddgd|d| ¡}t| ¡ƒdks@J‚| ¡d ¡dksZJ‚| ¡d ¡dkstJ‚| ¡d ¡dksŽJ‚| ¡d ¡d ks¨J‚|j |¡|j ¡\}}|jddg|d| ¡}| ¡d ¡dksöJ‚| ¡d ¡dksJ‚|j |¡|j ¡\}}|jddgd|d| ¡}| ¡d ¡dks`J‚| ¡d ¡dkszJ‚|j |¡|j ¡}| d¡}t ¡}|jdd g|d}| ¡}t| ¡ƒdksÒJ‚|j ¡dusæJ‚tƒ}| ¡D]}| | ¡¡qôt|ƒdksJ‚|j |¡t ¡}|j ¡}| d¡}|jddg|d}| ¡}t| ¡ƒdkstJ‚|j ¡dusˆJ‚tƒ}| ¡D]}| | ¡¡q–t|ƒdks¾J‚|j ¡|j ¡}| d¡}|jdd g|d}| ¡}t| ¡ƒdksJ‚|j ¡dus J‚tƒ}| ¡D]}| | ¡¡q.t|ƒdksVJ‚dS)NéÓr r6r“rÀrz A (right)rNz B (right)rIr7r4rF)r Ú mark_rightrXr5)r2r rX)r2r r–rX)r_ÚfigureÚadd_subplotrEZmakeTimeDataFramerQr]r2rSZ get_textsrºr!ÚsetÚaddrƒrFr`Z makeDataFrame)rArcrXrbr^Úcolorsrr*r*r+Útest_secondary_legendfs z TestTSPlot.test_secondary_legendcCsztdddd}ttj t|ƒd¡|ƒ}|j ¡\}}|j|d}| ¡}| ¡D]$}t| ¡ƒdkrP| ¡dksPJ‚qPdS) NrŒrrrr4rvrrØ) rrr/r0r1r2r_r`rQrµr¶rºZget_rotation)rArŽrbr„rXrùrr*r*r+Útest_format_date_axisÁsz TestTSPlot.test_format_date_axiscCsXtdddd}ttt|ƒƒƒ}|j ¡\}}|j||dd}t t |d ¡ƒ|¡dS)NrYr[r©Ústartr&r%ÚY)rdr)rrRr^r2r_r`rQrErÞrrT)rAr)rtr„rXrr*r*r+Útest_ax_plotËs zTestTSPlot.test_ax_plotc Cstdddƒtdddƒg}t ddd¡}t dd d¡}d ddœ}|j ¡\}}|jd d„|Dƒ|fi|¤Ž|jdd„|Dƒ|fi|¤Ž| ¡\}}tjdd„|Dƒtjd} t | ¡dd…df| ¡tjdd„|Dƒtjd} t | ¡dd…df| ¡dS)NéØréiÙrNg$@g&@gà?g(@ú-rÀ)ÚfmtZlwcSsg|]}| ¡‘qSr*©Ú toordinalr'r*r*r+r,Úr-z0TestTSPlot.test_mpl_nopandas..cSsg|]}| ¡‘qSr*r¦r'r*r*r+r,Ûr-cSsg|]}| ¡‘qSr*r¦r'r*r*r+r,ßr-rorcSsg|]}| ¡‘qSr*r¦r'r*r*r+r,ár-)rr/rÛr_r`Z plot_daterSrprsrErKrÂ) rAÚdatesZvalues1Zvalues2Úkwr„rXZline1Zline2Úexpr*r*r+Útest_mpl_nopandasÒs zTestTSPlot.test_mpl_nopandascCs¦ddlm}t ¡dd…}|gd¢}|j ¡\}}|dd…j|d|dd…j|d| ¡\}}|| |j ¡d|¡ks†J‚|| |j ¡d|¡ks¢J‚dS)Nr©ÚDatetimeConverterr© rNrÀr¥r¦r`r\r[rr@rArrbrcr¥rvÚ©r“rrEr‚r_r`rQráÚconvertr9rirj©rArrWZts_irregularr„rXrHr&r*r*r+Ú test_irregular_ts_shared_ax_xlimäsz+TestTSPlot.test_irregular_ts_shared_ax_xlimcCs„gd¢}gd¢}td|d}td|d}|j ¡\}}|j|d| ¡\}}|jd|d| ¡\} } || kstJ‚|| ks€J‚dS) N)rNrIr4rÀ)r¥r¦rÌr`rNrKrIrvTr“)rr_r`rQrá©rAZindex_1Zindex_2rDrEr„rXÚleft_beforeÚright_beforeÚ left_afterÚright_afterr*r*r+Útest_secondary_y_non_ts_xlimõsz'TestTSPlot.test_secondary_y_non_ts_xlimcCstdddd}tdddd}td|d}td|d}|j ¡\}}|j|d | ¡\}}|jd |d| ¡\} } || ks€J‚|| ksŒJ‚dS)Nú 2000-01-01rÀrržz 2000-01-05rNrKrIrvTr“)rrr_r`rQrár´r*r*r+Ú test_secondary_y_regular_ts_xlimsz+TestTSPlot.test_secondary_y_regular_ts_xlimc Cs€tdddd}td|d}|j ¡\}}|j|d| ¡\}}| d¡ ¡jd |d | ¡\}}||kspJ‚||ks|J‚dS)NrºrïrirrNrKrvrTr“)rrr_r`rQráZresampleÚmean) rArŽrWr„rXrµr¶r·r¸r*r*r+Ú#test_secondary_y_mixed_freq_ts_xlimsz.TestTSPlot.test_secondary_y_mixed_freq_ts_xlimcCs¼ddlm}t ¡dd…}|gd¢}|j ¡\}}|dd…j|d|dd…jd|d|dd…j|d| ¡\}}|| |j ¡d |¡ksœJ‚|| |j ¡d |¡ks¸J‚dS) Nrr¬rr®r¥rvTr“r¯r°r²r*r*r+Ú"test_secondary_y_irregular_ts_xlim$sz-TestTSPlot.test_secondary_y_irregular_ts_xlimcCsZtdddƒtdddƒg}|j ¡\}}| |¡tddddƒtddddƒg}| |¡dS)NirNrIr)rr_r`rQr)rArkr„rXr*r*r+Útest_plot_outofbounds_datetime6s z)TestTSPlot.test_plot_outofbounds_datetimecCsšdd„t d¡Dƒ}tdddd}ttj t|ƒd¡|ƒ}|j ¡\}}|j d|d |j ¡| ¡}d d„|Dƒ}t|ƒt|ƒksŠJ‚||ks–J‚dS)NcSsg|]}d|d›‘qS)z00:00:00.0000000z0>2dr*)r(rqr*r*r+r,Ar-zATestTSPlot.test_format_timedelta_ticks_narrow..r[Ú0Únsrr4rIr]cSsg|]}| ¡‘qSr*©rºr'r*r*r+r,Jr-)r/rÛrrr0r1r2r_r`rQÚdrawra©rAZexpected_labelsrŽrbrcrXr‰Z result_labelsr*r*r+Ú"test_format_timedelta_ticks_narrow?s z-TestTSPlot.test_format_timedelta_ticks_narrowcCsŽgd¢}tdddd}ttj t|ƒd¡|ƒ}|j ¡\}}|jd|d}|j ¡| ¡}d d „|Dƒ}t|ƒt|ƒks~J‚||ksŠJ‚dS)N) z00: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Àr[ú1 drr4rIr]cSsg|]}| ¡‘qSr*rÂr'r*r*r+r,br-z?TestTSPlot.test_format_timedelta_ticks_wide..)rrr/r0r1r2r_r`rQrÃrarÄr*r*r+Ú test_format_timedelta_ticks_wideNs z+TestTSPlot.test_format_timedelta_ticks_widecCs¶ttdƒtdddƒ}|j ¡\}}t|j|dtdddd}ttj t |ƒ¡|ƒ}|j ¡\}}t|j|dtddd d}ttj t |ƒ¡|ƒ}|j ¡\}}t|j|ddS) Nr¥Z1dayrðrvz1 day 2 hr 30 min 10 sr[rÆrz1 ns)rr^rr_r`rPrQr/r0r1r2)rAr…r„rXr9r*r*r+Útest_timedelta_plotfszTestTSPlot.test_timedelta_plotcCs`tdddd}|}t ddd¡}t ddd¡ddd…}|j ¡\}}|j||g||gd dS) NrHr[rrrrNgš™™™™™¹?rM)Úweights)rr/rÛr_r`Úhist)rArŽr)Zw1Zw2r„rXr*r*r+Ú test_histxszTestTSPlot.test_histcCs’tgd¢tdddƒtdddƒtdddƒgd}tgd¢tdddƒtd ddƒtd ddƒgd}|j ¡\}}|j|d|j|d|j|ddS)Nrorrr£r¬iÕrKiÍiÓr¢rv)rrr_r`rQ)rArDrEr„rXr*r*r+Útest_overlapping_datetimes$ ýþ ýþz$TestTSPlot.test_overlapping_datetimez1GH9053 matplotlib does not use ax.xaxis.converter)ÚreasoncCs„ttj d¡tdddd}| ¡}t t¡$|j|j |j ddWdƒn1sX0Y|j\}}t | ¡| ¡¡dS)Nr[z 1970-01-02rðrKrB)r†)rr/r0r1rrQrErOÚDeprecationWarningr9rkrrKrÂ)rAr…rXÚl1Úl2r*r*r+Útest_add_matplotlib_datetime64›s2 z)TestTSPlot.test_add_matplotlib_datetime64cCs–ttj d¡ dd¡ddgd}tdddd |d <|j ¡\}}|jd d|d|j ¡| ¡d}|jrrd}n|jr~d}nd }| ¡|ks’J‚dS)Nrr[rIr)rtr9z 2018-01-01rrr)r)rtrÐz 2017-12-08z 2017-12-12)rr/r0ZRandomStatervrr_r`ZscatterrÃraZmpl_ge_3_2_0rrº)rArbrcrXrdrær*r*r+Ú"test_matplotlib_scatter_datetime64¨s z-TestTSPlot.test_matplotlib_scatter_datetime64cCsütgd¢ƒ}t|gd¢dœƒ}|jddd}|j|ddtgd¢ƒ}t|gd¢dœƒ}|jddd}|j|d d| d¡jdd d}|j|d d| d¡jddd}|j|dd|jddd d d }|j|d d|jddd dd }|j|dddS)N)ú 2020-05-01ú 2020-05-02z 2020-05-03ro)r)rtr)rtr)Zxrot)rÓrÔz 2020-05-04rØT)rtZ use_indexF)r)rtr`rg)rrrQZ_check_ticks_propsZ set_index)rAr)rbrër*r*r+Útest_check_xticks_rot¸s z TestTSPlot.test_check_xticks_rotN)]Ú__name__Ú __module__Ú__qualname__r<rGrZrernrsr}r‡r‰rr•r™r›ryÚmarkZparametrizeržrŸr r¢r¤r§r©r«r®r¿rËrÎrÔrÕrÖrÝrßrìrírürrrr r rrr$r/r0ÚtdZskip_if_no_scipyr4r6r<r=rIrNrQrRrXr[r_rerhrkrmrtrwryrzrrŽr’r”rœrr¡r«r³r¹r»r½r¾r¿rÅrÇrÈrËrÌZxfailrÑrÒrÕr*r*r*r+r/s¸ ÿ ÿ ( 6 ![ )[ rc Os€ddlm}| ¡}zX| ¡| d¡}| d| ¡¡}t|ddƒ} ||i|¤Ž} | dus`J‚| d| ¡¡}|dur¦|jj }t |tƒr|j}| dur¦|j |ks¦J‚|durÄ| durÄ|j |ksÄJ‚| d¡}||d<||i|¤Ž} | dusðJ‚t jdd}| |¡Wdƒn1s0Yt jdd}t ||¡Wdƒn1sX0YW| |¡n| |¡0dS)Nrr•rXr%éÔT)Zreturn_filelike)Zmatplotlib.pyplotZpyplotZgcfZclfr˜ÚpopZgcaÚgetattrr9r%rrrrEZensure_cleanZsavefigÚpickleÚdumprF) Úfr%ZseriesÚargsÚkwargsr_rcrXZorig_axZorig_axfreqrÍZdfreqÚpathr*r*r+rPÓs6 *.rP)NN)/Ú__doc__rrrrrÞZnumpyr/ryZpandas._libs.tslibsrrZpandas.util._test_decoratorsÚutilZ_test_decoratorsrÚZpandasrr r rrr rZpandas._testingZ_testingrEZpandas.core.indexes.datetimesrrrZpandas.core.indexes.periodrrrZpandas.core.indexes.timedeltasrZpandas.tests.plotting.commonrZpandas.tseries.offsetsrrÙZslowZ pytestmarkZskip_if_no_mplrrPr*r*r*r+Ús8$ .