U /e @s0ddlmZddlZddlZddlmZddlm Z iZ ej j rHde d<d9ddZ ejd d d d d dgeeddgdddgddgdddddgddgddgddZejd dddd gd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zejd5ddddd6gd7d8ZdS):)productN) assert_eqFZ check_freqmeancKst|j|f||SN)getattrresample)dffreqhowkwargsr N/tmp/pip-unpacked-wheel-dbjnr7gq/dask/dataframe/tseries/tests/test_resample.pyrsrobjmethod npartitionsr closedlabelseriesframecountZohlcZ30ThdwMrightleftc Cstjdddd}|tjdddd}|dkrFtjtt||d}n"|d krhtjd tt|i|d}tj||d }t |||||d } t |||||d } t | | d d| j } | j d| dkst | j d| dkst dS)N1-1-2000 2-15-2000rr 4-15-2000 5-15-2000rindexrar)r rrFZ check_dtyper)pd date_rangeunionSeriesrangelen DataFramedd from_pandasrr divisionsr$AssertionError) rrrr rrr$psdsresultexpectedr2r r r test_series_resamplesr8ZnuniquesizesumcCsttjdddd}|tjdddd}tjtt||d}tj|dd }tt | d |t | d |dS) N 2000-01-01 2000-02-15rr r!r"r#rr&Z30min) r)r*r+r,r-r.r0r1rrr)rr$r4r5r r r $test_resample_has_correct_fill_value4sr=cCstjdddd}tjtt||d}tj|dd}t|d d |d d t|d d d g|d d d gdS) Nr;r<rr r#rr&Z10minrmin) r)r*r,r-r.r0r1rraggr$r4r5r r r test_resample_agg@s"rAcCstjdddd}tjtt||d}tj|dd}d d d }t|d j |dd |d j |dd |d j |dd dk st dS)Nr;r<rr r#rr&c_s|Srr )rbarargsr r r r fooQsz,test_resample_agg_passes_kwargs..fooZ2h)rC)rB) r)r*r,r-r.r0r1rrr?computeallr3)r$r4r5rEr r r test_resample_agg_passes_kwargsLs  *rHc Csztjdddd}|tjdddd}tjtt||d}tj|dd }tj t d d | d  W5QRXdS) NrrDr r!r"r#rr&z(Index is not contained within new index.)matchZ2M)r)r*r+r,r-r.r0r1pytestZraises ValueErrorrrrFr@r r r Ctest_resample_throws_error_when_parition_index_does_not_match_indexXs rMcCsddddddddd d d d d ddddddddddg}t|tt|d}tj|ddd|d<|dd}tj |ddd}|d }t ||dS)Nl@Brk+l@-I$ l+lD60l+l>Fn+lPln+lcn+lvn+lY/b;2.lpWta2.lO2.l2.l.o-2.lzl@2.lh #S 3.l}Fk}/lqXk}/l(]~;l}/l*bl}/l06l}/lXIl}/lLh\l}/lHJGm}/lxbOyn -)TimeZCountsrNT)utcZ1Qrr&) r)r/r-r.Z to_datetime set_indexrr9r0r1rFr)timesrr7ddfactualr r r 4test_resample_pads_last_division_to_avoid_off_by_oneas<rTcCsddl}tjdddd}|tjdddd}tjd|jt|i|d }tj|d d }| d  }| d  }t ||dS) Nrz 2012-01-02z 2012-02-02Hr z 2012-03-02z 2012-04-02pr#rr&Z2D) numpyr)r*r+r/randomr.r0r1rrrFr)npr$rrRr7r6r r r (test_resample_does_not_evenly_divide_daysrZcCsxtjdddd}|tjdddd}tjtt||d}tj|dd }|d  }|d  }t ||dS) Nz2012-01-02 00:00:00z2012-01-02 01:00:00Tr z2012-01-02 06:00:00z2012-01-02 08:00:00r#rr&Z57T) r)r*r+r,r-r.r0r1rrrFr)r$sr5r7r6r r r /test_series_resample_does_not_evenly_divide_daysr]c Csxtddddgi}tj|ddd}z|jdds>tWn2tk rr}zdt |ksbtW5d}~XYnXdS) NxrBrF)rsortZ1mr2) r)r/r0r1r^rrr3rLstr)rrRer r r test_unknown_divisions_errors rccCsddlm}m}ddl}|}tj|||ddd}|jjddt|d}t ||d }| d }t j |d d }| djjd kstdS) Nr)datetime timedeltarIr rBd)highr9)datevaluesrir&)rdrerWnowr)r*rXrandintr.r/rPr0r1rrheadr$namer3)rdrerYZ date_todaydaysdatarrRr r r test_resample_index_names rrcCstdtdtdg}tjdgdgdgg|d}|jd|_tj|dd }|d }|d }t ||ft dS) Nz2016-10-15 00:00:00z2016-10-16 10:00:00z2016-10-17 00:00:00rBrr_r#zAmerica/Sao_Paulor&Z1D) r)Z Timestampr/r$Z tz_localizer0r1rrr CHECK_FREQ)r$rrRr6r7r r r *test_series_resample_non_existent_datetimesrtr?Zquantilecsntjdddd}tjtt||d}tj|dd}fdd }||d }||d }t||d d dS) Nr;r<rr r#rr&cs t|Sr)r)rr?r r z"test_common_aggs..Z1dFr') r)r*r,r-r.r0r1rr)r?r$r4r5fresr7r rur test_common_aggss rz)r) itertoolsrZpandasr)rKZdask.dataframeZ dataframer0Zdask.dataframe.utilsrrs_compatZ PANDAS_GT_110rmarkZ parametrizelistr8r=rArHrMrTrZr]rcrrrtrzr r r r sF          %