
    Mh              
          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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)startendfreq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          ^/var/www/html/env/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	   zzDOOPTU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                  S
S9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&   r7   zi-r   r      T)future_stackhashcategoryvaluevalue2namedrop)rE   inplacebar)onetwothreer      )r   r   r   r   r   r   )r   rK   r   r   rK   r   )r   rK   r   rK   r   rK   )levelscodesr      r   levelrP   rE   )r   r   arangereshaper   listobjectrangestackr   namesrC   r#   r(   assert_series_equalr   r   r    r!   standard_normalr"   r   assert_index_equalget_level_values
isinstance)	r*   ir0   serss2return_valuer   rss	            r1   test_reset_indexTestResetIndex.test_reset_index$   s   "))C.((11$v,f55959a2aS95VD
 1	
 hhDh)!:.		__"}}__(_+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rK   r      rh   xrB   r   TrD   )r   r   rV   r#   r   rC   r*   r`   s     r1   test_reset_index_name$TestResetIndex.test_reset_index_nameH   sZ    9E%($=>}}$$))111}}$}'--22:::r4   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)Nrg   )r8   r<   rN   )ABCr   rn   ro   r   rK   rp   rO   TrQ   zLevel E matchEzToo many levels)r   rK   r   r8   rD   )r   r'   r#   r(   r)   pytestraisesKeyErrorrY   
IndexErrorr   rV   )r*   r0   rL   r`   resultexpecteds         r1   test_reset_index_level%TestResetIndex.test_reset_index_levelM   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   rn   int64)rC   r&   r   rK   r   )stopr;   )	r   rV   r#   r]   r   r   r   r(   r)   )r*   r`   series_resultseries_expecteds       r1   test_reset_index_range%TestResetIndex.test_reset_index_rangew   so    58#W5---z::::#VaVwnJA<N
 	m=r4   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)Nr8   zdoes not match index namerq   wrongTrD   rK   r   r   z	not found)r   rV   rt   ru   rv   r#   r   from_productrj   s     r1   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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)rG   rG   bazr   quxr   foor   )rH   rI   rH   rI   rH   rI   rH   rI   r      r   rh   rK   TrD   )zipr   from_tuplesr   r    r!   rZ   r   nanr%   r#   r]   r   r"   r   r   rL   rC   )r*   arraystuplesr   datar_   	deleveleds          r1   test_reset_index_with_drop)TestResetIndex.test_reset_index_with_drop   s   DD
 f&&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5r4   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   oldrB   newT)rC   rE   rF   )r   rV   r#   r(   rY   )r*   r_   ry   s      r1   -test_reset_index_inplace_and_drop_ignore_name<TestResetIndex.test_reset_index_inplace_and_drop_ignore_name   sB    U1XE*Ut<%(/
s-r4   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_stringTrD   )rt   importorskipr   rU   r
   r#   r(   rY   )r*   r_   rx   s      r1   "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__r2   rd   rk   rz   r   r   r   r   r   __static_attributes__r   r4   r1   r   r      s1    9 "&H;
(1T>.6(.,r4   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   rK   g      ?g      ?)r&   r   string)level_0level_1level_2r   )r   r   r   rU   r#   r$   r   r}   float64r(   rY   )arrayr&   using_infer_stringidxrx   expry   s          r1   7test_reset_index_dtypes_on_empty_series_with_multiindexr      s     
!
!Aq6C:u"=
>C&,Ra0<<>EEF((fCxxzz#vos5		
H 6,r4   znames, expected_namesrn   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S/-   S9n[
        R                  " XE5        g [        R                  " [        S	S
9   UR                  5         S S S 5        g ! , (       d  f       g = f)NrK   )rX   r   T)r   )rK   rK   rK   r   r   zcannot insertrq   )
r   r   from_arraysr#   r   r(   r)   rt   ru   
ValueError)rX   expected_namesr   r`   rx   ry   s         r1   test_column_name_duplicatesr      s     	s*001#s5IJA5i[.A32FG
f/]]:_=MMO >==s   :B
B")r   numpyr   rt   pandaspdr   r   r   r   r   r	   r
   pandas._testing_testingr(   r   markparametrizerU   period_rangePeriodDtyper   r   r   r4   r1   <module>r      s          ^, ^,B 
sVOOKANN(	
	-	-$ 
sc3Z 
T	Y	23 +eT];	 <	r4   