
    I j                     x    S SK rS SKrS SKJr  S SKJrJrJr  S SK	J
r  S SKJrJr   " S S5      r " S S5      rg)	    N)Pandas4Warning)	Timedeltatimedelta_rangeto_timedelta)DaySecondc                      \ rS rSrS rS r\R                  R                  SSS/5      S 5       r	\R                  R                  S/ S	Q5      S
 5       r
\R                  R                  S/ SQ5      S 5       rS rS rS r\R                  R                  S/ SQ5      S 5       rS r\R                  R                  S/ SQ5      S 5       rSrg)TestTimedeltas   c                     [        SSSSS9n[        R                  " SSS9S-  R                  S	5      R	                  S
5      n[
        R                  " UR                  5       U5        g )Nz0 Days
   100000Ds)periodsfrequniti8)dtypei zm8[D]zm8[s])r   nparangeviewastypetmassert_numpy_array_equalto_numpy)selftdiexp_arrs      څ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/indexes/timedeltas/test_timedelta_range.pytest_timedelta_range_unit(TestTimedeltas.test_timedelta_range_unit   sS    hM99Rt,w6<<WELLWU
##CLLNG<    c                 8   [        [        R                  " S5      SS9R                  S5      n[	        SSSS9n[
        R                  " X!5        [        [        R                  " S5      SS9R                  S5      n[	        SSSS	9n[
        R                  " X!5        [        [        R                  " S5      SS9R                  S5      [        S
5      -   [        5       -   n[	        SSSS	9n[
        R                  " X!5        [        / SQSS9R                  S5      [        S
5      -   n[	        SSSS9n[
        R                  " X!5        [        [        R                  " S5      SS9R                  S5      S-  n[	        SSSS9n[
        R                  " X!5        g )N   Dr   us0 days)r   r      z10 daysr      z1 days, 00:00:02z5 days, 00:00:02)      r$      	   2D2   min   30min)r   r   )	r   r   r   as_unitr   r   assert_index_equalr   r   )r   expectedresults      r   test_timedelta_range#TestTimedeltas.test_timedelta_range   sJ   		!37??E 13?
f/		"C8@@F 93?
f/ 1C088>JSUR 	 !!35GcR
f/c:BB4H6RS9T !3QTJ
f/		"E:BB4H2M D
f/r"   zdepr_unit, unit)Hhour)Ssecondc                    SU S3n[        [        R                  " S5      US9n[        R                  " [
        US9   [        [        R                  " S5      US9n[        R                  " XT5        S S S 5        g ! , (       d  f       g = f)N'z8' is deprecated and will be removed in a future version.r$   r&   match)r   r   r   r   assert_produces_warningr   r6   )r   	depr_unitr   depr_msgr7   r8   s         r   #test_timedelta_units_H_S_deprecated2TestTimedeltas.test_timedelta_units_H_S_deprecated1   sk     	{RS 	  		!48''hG!"))A,Y?F!!&3 HGGs   5A<<
B
r   )TtLlUuNnc                     SU 3n[         R                  " [        US9   [        [        R
                  " S5      US9  S S S 5        g ! , (       d  f       g = f)Nzinvalid unit abbreviation: rA   r$   r&   )pytestraises
ValueErrorr   r   r   )r   r   msgs      r   "test_timedelta_unit_T_L_U_N_raises1TestTimedeltas.test_timedelta_unit_T_L_U_N_raises<   s:    +D62]]:S11D1 211s   A
Azperiods, freq))r-   r0   )r$   r%   )   19h12min)r.   16h)r/   12hc                 \    [        SSUS9n[        SSUS9n[        R                  " X45        g )Nr(   4 days)startendr   r]   r^   r   )r   r   r6   )r   r   r   r8   r7   s        r   test_linspace_behavior%TestTimedeltas.test_linspace_behaviorC   s.    
 !xXwO"xdK
f/r"   c                     Sn[         R                  " [        US9   [        SSSS9  S S S 5        g ! , (       d  f       g = f)NzInvalid frequency: HrA   r(   r\   19H12minr_   rQ   rR   rS   r   r   rT   s     r   test_timedelta_range_H_raises,TestTimedeltas.test_timedelta_range_H_raisesL   s-    $]]:S1(zJ 211   1
?c                     Sn[         R                  " [        US9   [        SSSS9  S S S 5        g ! , (       d  f       g = f)NzInvalid frequency: TrA   r(   r\   19h12Tr_   rd   re   s     r   test_timedelta_range_T_raises,TestTimedeltas.test_timedelta_range_T_raisesS   s-    $]]:S1(xH 211rh   c                 h   Sn[         R                  " [        US9   [        SS9  S S S 5        [         R                  " [        US9   [        SS9  S S S 5        [         R                  " [        US9   [        SS9  S S S 5        [         R                  " [        US9   [        5         S S S 5        [         R                  " [        US9   [        SSS	S
S9  S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Np= f! , (       d  f       g = f)NzVOf the four parameters: start, end, periods, and freq, exactly three must be specifiedrA   r(   )r]   z5 days)r^   r+   r   r   h)r]   r^   r   r   rd   re   s     r   test_errorsTestTimedeltas.test_errorsY   s    . 	 ]]:S1(+ 2 ]]:S1) 2 ]]:S1A& 2 ]]:S1 2 ]]:S1("3O 21 21 21 21 21 21s;   
C
C02
DD	D#
C-0
C>
D
D #
D1z"start, end, freq, expected_periods))1D10Dr0   r$   )r0   30D3Dr   )2s50s5sr   )4D16Dru   r$   )8Drz   40siC  c                     [        XUS9n[        U5      US   :X  d   e[        U5      US   :  d   e[        U5      U:X  d   eg )Nr_   r   )r   r   len)r   r]   r^   r   expected_periodsress         r   $test_timedelta_range_freq_divide_end3TestTimedeltas.test_timedelta_range_freq_divide_endo   sP     E>3q6)))~R(((3x++++r"   c                 8    [        SSSS9nUR                  b   eg )N0s1s   rn   )r   r   )r   r8   s     r   test_timedelta_range_infer_freq.TestTimedeltas.test_timedelta_range_infer_freq   s      tR8{{"""r"   zfreq_depr, start, end))z3.5l05:03:0105:03:10)z2.5Tz5 hoursz5 hours 8 minutes)z3.5Sr   r   c                     SU 3n[         R                  " [        US9   [        X#US9  S S S 5        g ! , (       d  f       g = f)NzInvalid frequency: rA   r_   rd   )r   	freq_deprr]   r^   rT   s        r   !test_timedelta_range_removed_freq0TestTimedeltas.test_timedelta_range_removed_freq   s4    , $I;/]]:S1%yA 211s	   3
A N)__name__
__module____qualname____firstlineno__r    r9   rQ   markparametrizerF   rU   r`   rf   rk   rp   r   r   r   __static_attributes__r   r"   r   r
   r
      s    =0. [[.0PQ4 R4 [[V%MN2 O2 [[W00KIP, [[,	

,
,#
 [[	
(B)(Br"   r
   c                   &    \ rS rSrS rS rS rSrg)TestTimedeltaRangeUnitInference   c                     [        S5      R                  U5      n[        SS9R                  U5      n[        X#SS9nUR                  U:X  d   eg )Nr   r,   daysr%   r*   r   r5   r   r   r   r   r]   r^   r   s        r   1test_timedelta_range_unit_inference_matching_unitQTestTimedeltaRangeUnitInference.test_timedelta_range_unit_inference_matching_unit   sI    !$$T*Q''-es3xx4r"   c                     [        S5      R                  U5      n[        SS9R                  S5      n[        X#SS9nUR                  U:X  d   e[        X#R                  S5      SS9nUR                  S:X  d   eg )Nr   r,   r   r   r%   r*   nsr   r   s        r   3test_timedelta_range_unit_inference_mismatched_unitSTestTimedeltaRangeUnitInference.test_timedelta_range_unit_inference_mismatched_unit   sp    !$$T*Q'',es3xx4e[[%6SAxx4r"   c                     [        S5      R                  S5      n[        SS9R                  S5      n[        XSS9nUR                  S:X  d   e[        XR                  S	5      SS9nUR                  S	:X  d   eg )
Nr   msr,   r   r   	2000000usr*   r'   r   r   )r   r]   r^   r   s       r   (test_timedelta_range_unit_inference_tickHTestTimedeltaRangeUnitInference.test_timedelta_range_unit_inference_tick   sp    !$$T*Q'',e{;xx4e[[%6[Ixx4r"   r   N)r   r   r   r   r   r   r   r   r   r"   r   r   r      s       r"   r   )numpyr   rQ   pandas.errorsr   pandasr   r   r   pandas._testing_testingr   pandas.tseries.offsetsr   r   r
   r   r   r"   r   <module>r      s;      ( 
 KB KB\   r"   