a S/b @s*ddlmZddlZddlZddlmZddlm Z ddl m Z iZ ej jrTde d<d1dd Zejgd eed d ggd ddggdddgddgddZejdgdddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zejd+gd,d-d.Zd/d0ZdS)2)productN)getnanos) assert_eqFZ check_freqmeancKst|j|fi||SN)getattrresample)dffreqhowkwargsr Ilib/python3.9/site-packages/dask/dataframe/tseries/tests/test_resample.pyrsr)objmethod npartitionsr closedlabelseriesframe)countrZohlc)Z30ThdwMrightleftc Cstjdddd}|tjdddd}|dkrFtjtt||d}n"|d krhtjd tt|i|d}tj||d }t |||||d } t |||||d } t | | d d| j } | j d| dksJ| j d| dksJdS)N1-1-2000 2-15-2000rr 4-15-2000 5-15-2000rindexrar)r rrFZ check_dtyper) pd date_rangeunionSeriesrangelen DataFramedd from_pandasrr divisionsr%) rrrr rrr%psdsresultexpectedr3r r rtest_series_resamplesr8r)rnuniquesizesumcCsttjdddd}|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/r1r2rrr)rr%r4r5r r r$test_resample_has_correct_fill_value5s r>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/r1r2rraggr%r4r5r r rtest_resample_aggAs"rBcCstjdddd}tjtt||d}tj|dd}d d d }t|d j |dd |d j |dd |d j |dd dk sJdS)Nr<r=rr!r$rr'c_s|Srr )rbarargsr r r rfooRsz,test_resample_agg_passes_kwargs..fooZ2h)rD)rC) r*r+r-r.r/r1r2rrr@computeall)r%r4r5rFr r rtest_resample_agg_passes_kwargsMs  *rIcCstjdddd}|tjdddd}tjtt||d}tj|dd }tj t d d "| d  Wdn1s0YdS) Nrr Dr!r"r#r$rr'z(Index is not contained within new index.matchZ2M)r*r+r,r-r.r/r1r2pytestZraises ValueErrorrrrGrAr r rCtest_resample_throws_error_when_parition_index_does_not_match_indexYs rOcCs~gd}t|tt|d}tj|ddd|d<|dd}tj |ddd}|d }t ||dS) N)l@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 -)TimeZCountsrPT)ZutcZ1Qrr') r*r0r.r/Z to_datetime set_indexrr:r1r2rGr)timesr r7ddfactualr r r4test_resample_pads_last_division_to_avoid_off_by_onebsrUcCsddl}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,r0randomr/r1r2rrrGr)npr%r rSr7r6r r r(test_resample_does_not_evenly_divide_daysr[cCsxtjdddd}|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/r1r2rrrGr)r%sr5r7r6r r r/test_series_resample_does_not_evenly_divide_daysr^c Csxtdgdi}tj|ddd}z|jddsz"test_common_aggs..Z1dFr() r*r+r-r.r/r1r2rr)r@r%r4r5fresr7r rrrtest_common_aggss rwcCs:tjtddtdWdn1s,0YdS)Nzgetnanos was deprecatedrK)rMZwarns FutureWarningrr r r rtest_getnanos_deprecatedsry)r) itertoolsrZpandasr*rMZdask.dataframeZ dataframer1Zdask.dataframe.tseries.resamplerZdask.dataframe.utilsrrpZ_compatZ PANDAS_GT_110rZmarkZ parametrizelistr8r>rBrIrOrUr[r^rdrorqrwryr r r rsJ          %