
    I j              
          S SK J r   S SKrS SKrS SKrS SKJrJrJrJ	r	J
r
JrJr  S SKJr   " S S5      r\R"                  R%                  SSS/\4\R(                  " S	S
SS9\R*                  " SS94/5      S 5       r\R"                  R%                  SSS/SS/4SS/SS/4/5      \R"                  R%                  SSS/5      S 5       5       rg)    )datetimeN)	DataFrameIndex
MultiIndex
RangeIndexSeries
date_rangeoption_contextc                   J    \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 rSrg)TestResetIndex   c                 ^   [        SSSSS9R                  S 5      n[        S[        R                  R                  S5      R	                  [        U5      5      0US9nUR                  5       nUR                  R                  S	   [        R                  " S
5      :X  d   eUR                  S5      n[        R                  " X$SS9  [        SSS5      n[        US//SS/S9nUR                  S5      nUR                  S	   U:X  d   eUR                  5       S   R                  S	   U:X  d   eg )Nz1/1/2001z6/1/2001Dns)startendfrequnitv   indexr   zM8[ns]r   F)check_namesi        g333333(@DateValuecolumns)r	   
_with_freqr   nprandomdefault_rnglenreset_indexdtypesilocdtype	set_indextmassert_frame_equalr   r   )selfdtid1d2d3stampdfs          }/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/series/methods/test_reset_index.pytest_reset_index_dti_round_trip.TestResetIndex.test_reset_index_dti_round_trip   s   *3T

*T
 	 RYY2215<<SXFGsS^^yy~~a BHHX$6666\\'"
b%8 r2&0AB\\&!xx{e###~~',,Q/5888    c                    [        S[        R                  " S5      R                  S5      -  [	        [        S5      [        S9[	        [        S5       Vs/ s H  nSU 3PM
     sn[        S9S9S S	 nUR                  5       nS
S/UR                  l
        SUl        UR                  5       nSU;   d   eUR                  SS9nSU;   d   eUR                  SS9nUnUR                  SSS9nUb   e[        R                  " XE5        [        S// SQSS/// SQ/ SQ/ SQ/S9n[!        [        R"                  R%                  S5      R'                  S5      US9nUR                  SS9n[)        UR*                  5      S:X  d   eUR                  SS/SS9n[        R,                  " UR                  [	        UR/                  S5      5      5        [1        U[         5      (       d   eg s  snf )Ng?x   )      ABCDr(   r9   zi-r   r      hashcategoryvaluevalue2nameTdrop)rF   inplacebar)onetwothreer      )r   r   r   r   r   r   )r   rL   r   r   rL   r   )r   rL   r   rL   r   rL   )levelscodesr      r   levelrQ   rF   )r   r!   arangereshaper   listobjectrangestackr   namesrD   r%   r*   assert_series_equalr   r   r"   r#   standard_normalr$   r   assert_index_equalget_level_values
isinstance)	r,   ir2   serss2return_valuer   rss	            r3   test_reset_indexTestResetIndex.test_reset_index&   s   "))C.((11$v,f55959a2aS95VD
 1	
 hhj!:.		__"}}__(_+2~~ OOO&~~4~>###
q% G4q!f=%'9;MN
 299((+;;A>eL]]]#2::!###]]!Qd]3
bhhe.D.DQ.G(HI"f%%%%= 6s   G&c                     [        / SQ[        [        S5      SS9S9nUR                  5       R                  R
                  b   eUR                  SS9R                  R
                  b   eg )NrL   r      ri   xrC   r   TrE   )r   r   rW   r%   r   rD   r,   ra   s     r3   test_reset_index_name$TestResetIndex.test_reset_index_nameJ   sZ    9E%($=>}}$$))111}}$}'--22:::r6   c                    [        / SQ/ SQ// SQS9nSS/SS/4 GH  nUR                  SS/5      S	   nUR                  US   S
9n[        R                  " XAR                  S5      5        UR                  US S S
9n[        R                  " XAR                  S5      5        UR                  US
9n[        R                  " XA5        UR                  SS/5      R                  USS9n[        R                  " XAS	/   5        [
        R                  " [        SS9   UR                  SS/S
9  S S S 5        UR                  S5      S   nUR                  US   S
9n[        R                  " XASS/   5        UR                  US S S
9n[        R                  " XASS/   5        UR                  US   SS9n[        R                  " XAS   5        [
        R                  " [        SS9   UR                  / SQS
9  S S S 5        GM     [        [        S5      5      R                  / SS9n[        [        S5      5      n[        R                  " XE5        g ! , (       d  f       GN.= f! , (       d  f       GM\  = f)Nrh   )r:   r>   rO   )ABCr   ro   rp   r   rL   rq   rP   TrR   zLevel E matchEzToo many levels)r   rL   r   r:   rE   )r   r)   r%   r*   r+   pytestraisesKeyErrorrZ   
IndexErrorr   rW   )r,   r2   rM   ra   resultexpecteds         r3   test_reset_index_level%TestResetIndex.test_reset_index_levelO   s   	9-GCj1a&(Fc3Z(-A]]]3F!!&,,s*;<]]]4F!!&,,s*;<]]]0F!!&-\\3*-99T9RF!!&cU)4xz:S#J/ ; S!#&A]]]3F!!&c3Z.9]]]4F!!&c3Z.9]]]>F""6c73z1BCI. DC? )F a!--bt-<%(#
v0+ ;: DCs   I)I'
I$	'
I7	c                     [        [        S5      SSS9nUR                  5       n[        UR                  [
        5      (       d   e[        SS/SS//SS/[        SS9S	9n[        R                  " X#5        g )
Nr   ro   int64)rD   r(   r   rL   r   )stopr=   )	r   rW   r%   r^   r   r   r   r*   r+   )r,   ra   series_resultseries_expecteds       r3   test_reset_index_range%TestResetIndex.test_reset_index_rangey   so    58#W5---z::::#VaVwnJA<N
 	m=r6   c                    [        [        S5      5      n[        R                  " [        SS9   UR                  SSS9  S S S 5        [        R                  " [        SS9   UR                  S5        S S S 5        [        [        S5      [        R                  " SS//S-  5      S	9n[        R                  " [        S
S9   UR                  SSS9  S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)Nr:   zdoes not match index namerr   wrongTrE   rL   r   r   z	not found)r   rW   ru   rv   rw   r%   r   from_productrk   s     r3   test_reset_index_drop_errors+TestResetIndex.test_reset_index_drop_errors   s     58]]8+FGMM'M- H]]8+FGMM'" H 58:#:#:QF8a<#HI]]8;7MM'M- 87 HGGG
 87s#   C C+ C<
C(+
C9<
D
c                    / SQ/ SQ/n[        USS06n[        R                  " U5      n[        R                  R                  S5      R                  S5      n[        XCS9n[        R                  UR                  S'   UR                  5       n[        U[        5      (       d   e[        UR                  5      [        UR                  R                   5      S	-   :X  d   eUR                  R"                  UR                  R"                  :X  d   eUR                  SS
9n[        U[        5      (       d   eUR                  R"                  UR                  R"                  :X  d   eg )N)rH   rH   bazr   quxr   foor   )rI   rJ   rI   rJ   rI   rJ   rI   rJ   strictTr      r   ri   rL   rE   )zipr   from_tuplesr!   r"   r#   r[   r   nanr'   r%   r^   r   r$   r   r   rM   rD   )r,   arraystuplesr   datar`   	deleveleds          r3   test_reset_index_with_drop)TestResetIndex.test_reset_index_with_drop   s   DD
 f*T*&&v.yy$$Q'77:T'ffOO%	)Y////9$$%SYY-=-=)>)BBBB##syy~~555OOO.	)V,,,,##syy~~555r6   c                     [        [        S5      SS9nUR                  SSSS9  [        [        S5      SS9n[        R                  " X5        g )Nr   oldrC   newT)rD   rF   rG   )r   rW   r%   r*   rZ   )r,   r`   rz   s      r3   -test_reset_index_inplace_and_drop_ignore_name<TestResetIndex.test_reset_index_inplace_and_drop_ignore_name   sB    U1XE*Ut<%(/
s-r6   c                     [         R                  " S5        [        / SQ[        S9n[	        SS5         UR                  SS9nS S S 5        [        R                  " WU5        g ! , (       d  f       N&= f)Npyarrow)abcr<   zfuture.infer_stringTrE   )ru   importorskipr   rV   r
   r%   r*   rZ   )r,   r`   ry   s      r3   "test_reset_index_drop_infer_string1TestResetIndex.test_reset_index_drop_infer_string   sU    I&_F3148__$_/F 9
vs+ 98s   A##
A1 N)__name__
__module____qualname____firstlineno__r4   re   rl   r{   r   r   r   r   r   __static_attributes__r   r6   r3   r   r      s1    9$"&H;
(1T>.6(.,r6   r   zarray, dtyper   r   z	12-1-2000r   zQ-DEC)periodsr   )r   c                 R   [         R                  " SS/SS/U /5      n[        [        US9S S R	                  5       R
                  nU(       a  SO[        n[        [        R                  [        R                  U[        :X  a  UOU[        S.5      n[        R                  " XF5        g )Nr   rL   g      ?g      ?)r(   r   str)level_0level_1level_2r   )r   r   r   rV   r%   r&   r!   r~   float64r*   rZ   )arrayr(   using_infer_stringidxry   exprz   s          r3   7test_reset_index_dtypes_on_empty_series_with_multiindexr      s     
!
!Aq6C:u"=
>C&,Ra0<<>EEF%%6Cxxzz#vos5		
H 6,r6   znames, expected_namesro   r   allow_duplicatesFTc                 J   [        S/[        R                  " S/S//U S9S9nU(       a7  UR                  SS9n[	        / SQ// UQSPS9n[
        R                  " XE5        g [        R                  " [        S	S
9   UR                  5         S S S 5        g ! , (       d  f       g = f)NrL   )rY   r   T)r   )rL   rL   rL   r   r   zcannot insertrr   )
r   r   from_arraysr%   r   r*   r+   ru   rv   
ValueError)rY   expected_namesr   ra   ry   rz   s         r3   test_column_name_duplicatesr      s     	s*001#s5IJA5i[2FN2FA2FG
f/]]:_=MMO >==s   :B
B")r   numpyr!   ru   pandaspdr   r   r   r   r   r	   r
   pandas._testing_testingr*   r   markparametrizerV   period_rangePeriodDtyper   r   r   r6   r3   <module>r      s          `, `,F 
sVOOKANN(	
	-	-$ 
sc3Z 
T	Y	23 +eT];	 <	r6   