
    Mh-                        S SK Jr  S SKrS SKrS SKJs  Jr  S SK	r
S SK	JrJr  S SKJr  S SKJr  S rS rS rS rS	 rS
 rS rS rS r\R6                  R9                  S/ SQ5      S 5       r\R6                  R9                  SSS/5      S 5       r\R6                  R9                  S/ SQ5      S 5       rS r \RB                  " S5      S 5       r"g)    )	timedeltaN)	DataFrameSeries)timedelta_rangec            	         [        SS/[        5       [        SS9/S9n U R                  S5      R                  5       n[        S[        R
                  [        R
                  S/[        SSSS9S9n[        R                  " X5        g )	N      )minutesdataindex1min0 day   periodsfreq)	r   r   resampleasfreqnpnanr   tmassert_frame_equal)dfresultexpecteds      V/var/www/html/env/lib/python3.13/site-packages/pandas/tests/resample/test_timedelta.pytest_asfreq_bugr      so    	Ay{Ia4H&I	JB[[ '')F#gqv>H &+    c            	      $   [         R                  " S[         R                  S/5      n [        S/ SQ0U 5      R	                  S5      R                  5       n[        SS[        R                  S/0[        SS	SS
9S9n[        R                  " X5        g )N0s2svalue)   r	      1sg      @g      @r   r	   r   r   )pdto_timedeltaNaTr   r   meanr   r   r   r   r   )r   r   r   s      r   test_resample_with_natr,      sy    OOT266401E+U3<<TBGGIF	3$%gqt<H &+r   c            	      J   [        SSSS9n [        S/ SQ0U S9nUR                  S5      R                  5       nSS	[        R
                  [        R
                  [        R
                  [        R
                  S
/0n[        U[        SSSS9S9n[        R                  " X$5        g )Nz00:00:00z00:10:005minr   r#   )r   r%   
   r   2minr   r0   )r   r   r   r   r   r   r   r   )r   r   r   expected_datar   s        r   $test_resample_as_freq_with_subperiodr3   &   s    J
@E	*-U	;B[[ '')Fq"&&"&&"&&"&&"EFM/*jv"VH &+r   c                  0   [        S[        R                  " S5      05      n U R                  U R                  S-  5      R                  5       n [        SSSS9U l        [        S[        R                  " S5      0[        R                  " [        R                  " S5      SS	9S
9nUR                  S5      R                  5       n[        R                  " X 5        US   nUR                  S5      R                  5       n[        R                  " X S   5        g )NAi     z0 days30min2   r   r   minunitr'   )r   r   arangegroupbyr   sumr   r(   r)   r   r   r   assert_series_equal)r   r   r   ss       r   test_resample_with_timedeltasrB   2   s    #ryy/0H" 4599;H$XGRHHN		biioboobiioE&R
B [[!%%'F&+
3AZZ $$&F6C=1r   c            	          [        [        [        S5      5      [        SSSS9S9n U R	                  S5      R                  5       n[        / SQ[        SSSS9S9n[        R                  " X5        g )	Nr%   1 dayrA   r9   r'   r"   )r   r%   r   r	   )r   listranger   r   r?   r   r@   )rA   r   r   s      r   %test_resample_single_period_timedeltarG   C   s[    tE!H~_W3PQ%RSAZZ!!#FiwTST'UVH6,r   c                      [        SSSS9n [        [        S5      U S9nUR                  S5      R	                  5       nUR                  [        5      n[        R                  " X#5        g )N0	   10msr   r'   )	r   r   rF   r   r+   astypefloatr   r@   )r   seriesr   r   s       r   #test_resample_timedelta_idempotencyrO   J   sS    C8EE!HE*F__V$))+F}}U#H6,r   c                     [        SSSS9n [        [        R                  R	                  S5      R                  [        U 5      5      U S9nUR                  SSS	9R                  5       nUR                  S5      R                  5       n[        SS
SS9n[        SSSS9n[        R                  " UR                  U5        [        R                  " UR                  U5        g )Nr!      rA   startr   r   r$   r'   r"   5s)offset25srS   endr   29s)r   r   r   randomdefault_rngstandard_normallenr   r+   r   assert_index_equalr   )rngts	with_basewithout_baseexp_without_baseexp_with_bases         r   (test_resample_offset_with_timedeltaindexre   S   s    
bs
;C			%%a(88SB#	NBD.335I;;t$))+L&Tu4H#$EEM,,,.>?)//=9r   c            
         [        SS0[        R                  " [        [	        S5      5      SS9S9n U S   R                  S5      U S'   U R                  S	5      R                  S
 5      n[        R                  " [        R                  " SS/SS9S	S9R                  S5      n[        SS/SS/S.US9nUR                  SS/SS9nUS   R                  S5      US'   [        R                  " X5        g )N	Group_objr5      rA   r;   r'   categoryGroup10sc                 <    U R                  5       R                  S   $ )Nr   )value_countsr   xs    r   <lambda>Dtest_resample_categorical_data_with_timedeltaindex.<locals>.<lambda>f   s    q~~/?/E/Ea/Hr   r   r0   zm8[s]dtyper/   ns)rg   rj   r   )axis)r   r(   r)   rE   rF   rL   r   aggTimedeltaIndexr   arrayas_unitreindexr   r   )r   r   exp_tdir   s       r   2test_resample_categorical_data_with_timedeltaindexr|   b   s    	K%R__T%)_SV-W	XB[/((4BwK[[##$IJF!R @uMUUG CjC:6H g 6Q?H -44Z@HW&+r   c                  v   [        SSSS9n [        SU 0U S9n[        SSSS9n[        X"SS9n[        R                  UR
                  S	'   UR                  S5      R                  5       S   n[        R                  " XC5        US   R                  S5      R                  5       n[        R                  " XC5        g )
NrD   z6 day4Dr/   timer'   2D)r   namer   )
r   r   r   r(   r*   ilocr   firstr   r@   )timesr   times2express        r   test_resample_timedelta_valuesr   s   s    
 GW48E	FE?%	0BWgD9F
F
3C&&CHHQK
++d

!
!
#F
+C3$
V*

d
#
)
)
+C3$r   zstart, end, freq, resample_freq))8h21h59min50srk   3h)r   22h1h5h)527D5006D3D10D)1Dr   r   r   )r   r   rk   2h)0hr   rk   r   )r   85DDr   c                    [        XUS9n[        [        R                  " [	        U5      5      US9nUR                  U5      R                  5       n[        X0US9n[        R                  " UR                  U5        UR                  R                  UR                  :X  d   e[        R                  " UR                  S   5      (       a   eg )NrW   r'   )r   rS   rX   )r   r   r   r=   r]   r   r:   r   r^   r   r   isnanr   )rS   rX   r   resample_freqidxrA   r   expected_indexs           r   !test_resample_timedelta_edge_caser      s      T
:CryyS"#.AZZ&**,F$-#NN&,,7<< 3 3333xxB(((((r   
duplicatesTFc           	         [        [        R                  R                  S5      R	                  SS9[        SSSS9S9nU (       a	  / S	QUl        UR                  S
S 2S S 24   R                  S5      R                  S 5      n[        S/S-  /S-  S/S-  /-   [        S
SSS9S9nUR                  Ul        [        R                  " X#5        g )Nr$   )'  r   )sizer!   r   	3906250nsrR   r'   )r5   Br5   Cr&   3sc                     [        U 5      $ )N)r]   rn   s    r   rp   Etest_resample_with_timedelta_yields_no_empty_groups.<locals>.<lambda>   s    SVr   i   r      i     )r   r   rZ   r[   normalr   columnslocr   applyr   r   )r   r   r   r   s       r   3test_resample_with_timedelta_yields_no_empty_groupsr      s     

		a ''Z'8D%kJ
B )
VVDE1H&&t,223CDF
bSEAI;&D"4@H zzH&+r   r<   )rA   msusrt   c           
         [         R                  " SU  S35      n[        S[        R                  " [         R
                  " S5      SS9R                  U5      0[        R                  " SSSS	9S
9nUR                  S5      R                  S5      n[        S[        R                  " S5      [        R                  " S5      /0[        R                  " SSSSS9S
9R                  U5      n[        R                  " X45        g )Nzm8[]r#   r   rA   r;   20200101UTC)r   tzr'   r   gGz?z0 days 00:00:00.990000z0 days 00:00:02.990000r$   )r   r   r   )r   rs   r   r(   r)   r=   rL   
date_ranger   quantile	Timedeltar   r   )r<   rs   r   r   r   s        r    test_resample_quantile_timedeltar      s     HHs4&]#E		"//"))A,S9@@GHmmJe<
B [[''-F5656	
 mmJe$G fUm  &+r   c                     [         R                  " [        S5       V s/ s H  n [         R                  " SU S-  -   S9PM     sn 5      n[	        [        S5      US9nUR                  SSSS9R                  5       n[	        / S	Q[         R                  " [        S
5       V s/ s H  n [         R                  " SU S-  -   S9PM     sn SS9S9n[        R                  " X45        g s  sn f s  sn f )Nr0   x   r6   )secondsr'   r:   right)closedlabel)r   r	            rJ      <   r/   )
r(   IndexrF   r   r   r   r?   rw   r   r@   )ir   serr   r   s        r   test_resample_closed_rightr      s    
((b	J	1BLLq2v6	J
KC
r#
&C\\%w\?CCEF9>qBAR\\#B,/B
H 6, K Cs   $C# $C(pyarrowc                      [         R                  " [        SSS9SS9n [        [        R
                  " S[        R                  S9U S9nUR                  S5      R                  5       n[        R                  " X!5        g )NrD   r%   )r   zduration[ns][pyarrow]rr   r'   r   )r(   r   r   r   r   r=   float64r   r+   r   r@   )r   r   r   s      r   test_arrow_duration_resampler      s^     ((?7A6>U
VCbii4C@Ht$))+F6,r   )#datetimer   numpyr   pytestpandas.util._test_decoratorsutil_test_decoratorstdpandasr(   r   r   pandas._testing_testingr   pandas.core.indexes.timedeltasr   r   r,   r3   rB   rG   rO   re   r|   r   markparametrizer   r   r   r   
skip_if_nor    r   r   <module>r      s       ) )   :,,	,2"--:,"%$ %		)	) e}5, 6,( !89, :,(- y- -r   