
    Mh'A                     P   S r SSKJr  SSKrSSKrSSKrSSKJr  SSK	J
r
JrJrJrJrJrJrJrJrJrJrJrJrJr  SSKJr  S rS rS rS	 rS
 rS r \RB                  RE                  S\" S/ SQS.5      \" SS05      \" SS0\#S9/\" / SQ/ SQS.5      \" S/ SQ05      \" S/ SQ05      //5      S 5       r$S r%S r&S r'S r(S r)S r*S r+S r,S r-S r.S r/S  r0S! r1S" r2\RB                  RE                  S#\" S$S%S&S'9\" S(S%S)S'9\" S*S%S+S'9/5      S, 5       r3S- r4S. r5S/ r6S0 r7S1 r8S2 r9S3 r:S4 r;\RB                  RE                  S50 S6S70/5      \RB                  RE                  S8/ S9S9/\<" SS:5      \Rz                  " / 5      /5      S; 5       5       r> " S< S=5      r? " S> S?5      r@g)@ztest get/set & misc     )	timedeltaN)IndexingError)NA	DataFrameIndex
IndexSlice
MultiIndexNaTSeries	Timedelta	Timestampconcat
date_rangeisnaperiod_rangetimedelta_rangec                  p   [        [        R                  R                  S5      R	                  S5      / SQS9n SnSn[
        R                  " [        US9   [        R                  " [        US9   U S     S S S 5        S S S 5        [
        R                  " [        US9   [        R                  " [        US9   SU S'   S S S 5        S S S 5        [
        R                  " [        S	S9   U S
     S S S 5        U R                  5       n [
        R                  " [        US9   [        R                  " [        US9   U S     S S S 5        S S S 5        Sn[
        R                  " [        US9   [        R                  " [        US9   SU S'   S S S 5        S S S 5        g ! , (       d  f       GN,= f! , (       d  f       GN6= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)N      )abr   r   r   indexz>Series.__[sg]etitem__ treating keys as positions is deprecatedz/index 5 is out of bounds for axis 0 with size 5matchr   z^'c'$cz7index 5 is out of bounds for axis (0|1) with size 5|^5$)r   nprandomdefault_rngstandard_normalpytestraises
IndexErrortmassert_produces_warningFutureWarningKeyError
sort_index)swarn_msgmsgs      \/var/www/html/env/lib/python3.13/site-packages/pandas/tests/series/indexing/test_indexing.pytest_basic_indexingr-      s   
		a 003;T	A PH
;C	z	-''XFaD G 
. 
z	-''XFAaD G 
. 
xx	0	# 
1 	
A	z	-''XFaD G 
. EC	z	-''XFAaD G 
.	- GF 
.	- GF 
.	- 
1	0 GF 
.	-
 GF 
.	-s   F-0F6F-G9F??G(G#H9G4?H*H'H
H'
F*	%F--
F<?
G		G
G #
G14
H	>H
H
H$	 H''
H5c                     U n[        / SQUS9n[        [        S5      US9nUS   n[        SS/[        SS/US9S9n[        R                  " XESS	9  g )
N   r   r0   dtype   r   r0   r   r   Tcheck_exactr   r   ranger$   assert_series_equal)any_numeric_dtyper2   idxserresultexpecteds         r,   6test_getitem_numeric_should_not_fallback_to_positionalr>   9   sW    E
	
'C
q
%CVFq!fE1a&$>?H6>    c                     U n[        / SQUS9n[        [        S5      US9nSUS'   [        / SQUS9n[        R                  " X4SS	9  g )
Nr/   r1   r3   r   
   r0   )rA   r0   rA   Tr4   r6   )r9   r2   r:   r;   r=   s        r,   6test_setitem_numeric_should_not_fallback_to_positionalrB   C   sH    E
	
'C
q
%CCFk-H3d;r?   c                     U R                   / SQ   nX   nU R                  U5      n[        R                  " X#5        XS   US    nU R                  US   US    n[        R                  " X#5        g )Nr   rA      r   r   )r   reindexr$   r8   loc)datetime_seriesindicesr<   r=   s       r,   test_basic_getitem_with_labelsrJ   M   sr    ##K0G%F&&w/H6,QZ'!*5F""71:
;H6,r?   c                      [        [        SSSS9/ SQS9n [        SSS9nU R                  S   nX!:X  d   eU R                  S	   nX!:X  d   eU S   nX!:X  d   eg )
Nz
2011-01-01r3   z
US/Eastern)periodstzr   r   r   r   )rM   r   r   )r   r   r   rG   iloc)r;   r=   r<   s      r,    test_basic_getitem_dt64tz_valuesrP   Y   sq     <|<OC ,7HWWS\FXXa[FXFr?   c                 V   [        [        R                  R                  S5      R	                  S5      5      nUS   n[
        R                  " X25        [
        R                  " U5         SUS'   S S S 5        U (       d  US:H  R                  5       (       d   eg g ! , (       d  f       N1= f)Nr   rA   .r   )	r   r   r   r   r    r$   r8   assert_cow_warningall)using_copy_on_writewarn_copy_on_writer)   r<   s       r,   test_getitem_setitem_ellipsisrV   h   s    ryy$$Q'77;<AsVF6%			1	2# 
3!  """"  
3	2s   )B
B(z$result_1, duplicate_item, expected_1   )r0   r   r   r3   r0   r   r0   i9  r1   r0   r   r3   c                     U R                  U5      nUR                  U5      n[        R                  " US   U5        US   U S   :X  d   eg )Nr0   r   )_appendr$   r8   )result_1duplicate_item
expected_1r<   r=   s        r,   $test_getitem_with_duplicates_indicesr_   t   sN    " n-F!!.1H6!9h/!9###r?   c                      [        / SQ/ SQ5      n U R                  S   U S   :X  d   eSU R                  S'   [        R                  " U S   S5        g )NrY   rN   r   r   r   )r   rO   r$   assert_almost_equalr)   s    r,   test_getitem_setitem_integersrc      sI    y/*A66!9#AFF1I1S61%r?   c                  b   [        SSSS9n [        U 5      n[        US   [        5      (       d   e[        UR                  S   [        5      (       d   e[        UR
                  S   [        5      (       d   e[        UR                  S   [        5      (       d   e[        UR                  S	   [        5      (       d   e[        X S
9nSn[        R                  " [        US9   [        US   [        5      (       d   e S S S 5        [        UR                  U S      [        5      (       d   e[        UR
                  S   [        5      (       d   e[        UR                  U S      [        5      (       d   e[        UR                  S	   [        5      (       d   eg ! , (       d  f       N= f)N2009041520090519B)freqr   r0   r   r3      r   z;Series.__getitem__ treating keys as positions is deprecatedr   )r   r   
isinstancer   atiatrG   rO   r$   r%   r&   )rngr;   r+   s      r,   test_series_box_timestamprn      sT   
Z#
6C
+Cc!fi((((cffQi++++cggaj),,,,cggaj),,,,chhqk9----

 C
GC		#	#M	=#a&),,,, 
>cffSVni0000cggaj),,,,cggc!foy1111chhqk9---- 
>	=s   %F  
F.c                     [        SSSS9n [        U 5      n[        US   [        5      (       d   e[        UR                  S   [        5      (       d   e[        UR
                  S   [        5      (       d   e[        UR                  S   [        5      (       d   e[        UR                  S	   [        5      (       d   eg )
Nz	1 day 1 sr   hrL   rh   r   r0   r   r3   ri   )r   r   rj   r   rk   rl   rG   rO   )rm   r;   s     r,   test_series_box_timedeltarr      s    
+qs
;C
+Cc!fi((((cffQi++++cggaj),,,,cggaj),,,,chhqk9----r?   c           
          [        [        S5      [        [        SSS5      5      S9n[        R                  " [
        SS9   U " U5      S     S S S 5        g ! , (       d  f       g = f)	NrA   r      r   r   z^1$r   r0   )r   r7   listr!   r"   r'   
indexer_slr;   s     r,   test_getitem_ambiguous_keyerrorrx      sE    
r$uQA"7
8C	xv	.3 
/	.	.s   A
A$c                     [        / SQ/ SQ5      n[        R                  " [        [        R
                  " S5      S9   U " U5      / SQ     S S S 5        g ! , (       d  f       g = f)N)r0   r   r3   ri   )foobarrz   bahz['bam'] not in indexr   )rz   r{   r|   bam)r   r!   r"   r'   reescaperv   s     r,   test_getitem_dups_with_missingr      sC     ;
<C	xryy1G'H	I345 
J	I	Is   A
A#c           
          [        [        S5      [        [        SSS5      5      S9nUR                  5       nSU " U5      S'   [	        U[        S/S/S9/5      n[
        R                  " X#5        g )NrA   r   rt   r   r   r   r0   )r   r7   ru   copyr   r$   r8   )rw   r)   s2r=   s       r,   test_setitem_ambiguous_keyerrorr      sb    uRyU1b!_ 56A 
BJrN1q&!QC012H2(r?   c                    [         R                  X R                  S   '   [         R                  U R                  / SQ'   [         R                  U R                  S'   [         R                  " U R                  S   5      (       d   e[         R                  " U R                  S   5      (       d   eSU [         R                  " U 5      '   [         R                  " U R                  S   5      (       a   eg )Nr   )r0   r         r   )r   nanr   rO   isnan)rH   s    r,   test_setitemr      s    02O))!,-')vvO$ ffO88O((+,,,,88O((+,,,,12OBHH_-.xx,,Q/00000r?   c                     U SS n[        U5      [        UR                  5      :X  d   eUR                  R                  SL d   eg )Nr   rt   T)lenr   	is_unique)rH   sls     r,   test_setslicer      s@    	2	Br7c"((m###88%%%r?   c                    Sn[         R                  " [        US9   U S S 2S4     S S S 5        [         R                  " [        US9   SU S S 2S4'   S S S 5        Sn[         R                  " [        US9   U [	        S S5      /     S S S 5        U [	        S S5      4   nU S S n[
        R                  " X#5        Sn[         R                  " [        US9   U SS S //     S S S 5        [         R                  " [        US9   SU SS S //'   S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Nm= f! , (       d  f       g = f)Nz0key of type tuple not found and not a MultiIndexr   r   z Indexing with a single-item listr   zunhashable type(: 'slice')?)r!   r"   r'   
ValueErrorslicer$   r8   	TypeError)rH   r+   r<   r=   s       r,   !test_basic_getitem_setitem_cornerr      s   
<C	xs	+1 
,	xs	+ !1 
, -C	z	-tQ() 
.
 eD!n./Fr"H6, )C	y	,T4L)* 
-	y	,-.T4L)* 
-	,) 
,	+	+	+
 
.	- 
-	,	,	,s;   
D
D"4D3
E>
E
D"
D03
E
E
E#c                    U R                  5       nU SS nU SS  nUSS nU R                  S   UR                  ;  d   eUR                  S   UR                  ;  d   e[        U5      [        UR                  5      :X  d   eXR                  S      XUR                  S      :X  d   eUR                  S   U R                  S   :X  d   e[        R                  " [
        R                  " U5      [
        R                  " U 5      SS  5        U SS n[        R                  " U5         SUS S & S S S 5        U(       a  [        R                  " X5        g U SS S:H  R                  5       (       d   eg ! , (       d  f       NJ= f)NrA   rt   i	   r   r0      )
r   r   r   r$   assert_numpy_array_equalr   arrayrR   r8   rS   )	string_seriesobject_seriesrT   rU   originalnumSlicenumSliceEndobjSlicer   s	            r,   
test_slicer      sb   !!#HR#H%KR#Hq!777q!777x=C////*+xq8I/JJJJ>>! 3 3B 7777 5rxx7Nst7TU 
r"	B			1	21 
3 
}7b$)..0000 
3	2s   'E11
E?c            	         [        / [        S9n [        S5      U R                  S'   [        R
                  " U [        [        S5      S/S95        U R                  U R                  R                  SS5      5      n [        R
                  " U [        [        R                  [        S5      /SS/S95        [        S5      U R                  S'   [        [        S5      SS/S9n[        R
                  " X5        g )Nr1   r0   rg   z1 daysr   r   A)r   objectr   rG   r$   r8   r   rF   r   insertr   r   )r)   r=   s     r,   test_timedelta_assignmentr     s    r A1AEE#J1fYx%8FG			!''..C()A1fbffi.A%B3PS*UV1AEE#Ji)#s<H1'r?   c                 p   [        S Vs0 s H  o/ SQ_M	     sn5      nUR                  / SQSS9nUb   e[        S/S/S9nSUS	'   UR                  5       nU  U (       a5  [        R
                  " 5          US	   R                  U5        S S S 5        UnOe[        R                  " [        S
S9   US	   R                  U5        S S S 5        [        / SQ/ SQ/ SQ/ SQS.5      nUR                  / SQSS9nUb   e[        R                  " X&5        g s  snf ! , (       d  f       N= f! , (       d  f       Nj= f)NrN   rY   T)inplacer0   )r   r   r   r   r   valzinplace methodr   )r   r0   r   )r   r   r   r   )
r   	set_indexr   r   r$   raises_chained_assignment_errorupdater%   r&   assert_frame_equal)rT   r   dfreturn_valuer)   df_originalr=   s          r,   test_underlying_data_conversionr     s   	/:/QyL/:	;B<<<>Ls9+&ABuI'')K//1uIQ 2''=MNuIQ O))IN
  ))/4)H###"') ; 21 ONs   D5D.D'
D$'
D5c                     U R                   / SQ   n[        R                  UR                   S'   [        R                  " U R                   S   5      (       a   eg )NrD   r0   rA   )rO   r   r   r   )rH   seqs     r,   test_preserve_refsr   6  sF    


{
+C&&CHHQKxx,,R011111r?   c                     U n[        [        R                  R                  S5      R	                  [        U5      5      USS9nU" U5      S   nUR                  UR                  :X  d   eg )Nr   sth)r   namerz   )r   r   r   r   r    r   r   )%lexsorted_two_level_string_multiindexrw   r   r;   r<   s        r,   test_multilevel_preserve_namer   <  s]    1E

		a 00U<EPUC _U#F;;#(("""r?   r   z
2014-01-01rt   MSrq   z2014-01M0rp   c           
         [        U S   5      n[        U S   5      n[        [        R                  " S5      U 5      n[        nXS   4 H  n[
        R                  " X4US S2   USS S2   5        [
        R                  " X4S US2   US SS2   5        X S   4 HG  n[
        R                  " X4XeS2   USSS2   5        [
        R                  " X4XVS2   USSS2   5        MI     M     g )Nr      rt      r   )strr   r   aranger   r$   !assert_indexing_slices_equivalent)r   keystr1keystr2r;   SLCkeykey2s          r,   test_slice_with_negative_stepr   I  s     %(mG%)nG
2
&C
Cq"
,,Scg2g,AErE
K
,,Sgs2g,EQrE
KBi(D00$2+6FBqQSGU00#2+6FAaPRFT )	 #r?   c                  l    [        SS/SS/S9n U S   S:X  d   eU S   S:X  d   eSU S'   U S   S:X  d   eg )Nr0   r   )r   )r   r   r3   )r   rb   s    r,   test_tuple_indexr   a  sP    1vff-.AV9>>V9>>AfIV9>>r?   c                      [        S5      [        S5      p[        SS/X/S9nX    S:X  d   eX!   S:X  d   eSX!'   X!   S:X  d   eg )Nr   r   r0   r   r   r3   )	frozensetr   )idx0idx1r)   s      r,   test_frozenset_indexr   j  sT    33$1vd\*A7a<<7a<<AG7a<<r?   c                      [        SS/SS/S9n U R                  5       n[        SS/SS/S9nX R                  U S:  '   [        R                  " X5        g )	Nr0   r   r   r   r   r      d   )r   r   rG   r$   r8   )r;   r=   rhss      r,   "test_loc_setitem_all_false_indexerr   t  sS    
!QSz
*CxxzH
!QSz
*CGGC#I3)r?   c                      [        S/5      n U R                  [        [        S/SS9   n[        / SS9n[        R                  " X5        g )Nr0   Fbooleanr1   int64)r   rG   r   r$   r8   r;   r<   r=   s      r,   +test_loc_boolean_indexer_non_matching_indexr   }  s@    
!+CWWVRKy9:Fb(H6,r?   c                      [        S/5      n [        [        S/SSS/S9n[        R                  " [        SS9   U R
                  U     S S S 5        g ! , (       d  f       g = f)Nr0   Fr   r   )r2   r   Unalignabler   )r   r   r!   r"   r   rG   )r;   indexers     r,   ,test_loc_boolean_indexer_miss_matching_indexr     sI    
!+Cb%[	!Q@G	}M	: 
;	:	:s   A
A c                      [        SS/05      n [        SU 05      nXR                  S'   [        X S.5      n[        R                  " X5        g )Nr   r0   label	new_label)r   r   )r   r   rG   r$   r8   )r   r;   r=   s      r,   (test_loc_setitem_nested_data_enlargementr     sG    	C!:	B
'2
CGGK45H3)r?   c                      [        S[        / SQSS9S9n U R                  SS n[        S[        S/SS9S9n[        R                  " X5        g )Nr0   )r   r0   r   Int64r1   r   r   r3   )r   r   rG   r$   r8   r   s      r,   'test_loc_ea_numeric_index_oob_slice_endr     sG    
%	9
:CWWQq\FauaS89H6,r?   c                      [        SSS.5      n [        R                  " [        SS9   U R                  S     S S S 5        g ! , (       d  f       g = f)Nr0   r   )TFr   r   )r   r!   r"   r'   rG   )r;   s    r,   test_getitem_bool_int_keyr     s6    
!$
%C	xs	+
 
,	+	+s   A
Ar   r   xr   Fr   c                     [        SSS/0UE5      nUR                  5       nSUR                  U '   [        R                  " X#5        g )Nr   r0   r         ?)r   r   rG   r$   r   )r   r   r   r=   s       r,   test_setitem_empty_indexerr     sB     
C!Q'3'	(BwwyHBFF7O"'r?   c                   p   \ rS rSr\R
                  R                  SS1SS0/5      S 5       r\R
                  R                  SS1SS0S1S4SS0S4/5      S 5       r\R
                  R                  SS1SS0/5      S 5       r	\R
                  R                  SS1SS0S1S4SS0S4/5      S 5       r
S	rg
)TestDeprecatedIndexersi  r   r0   c                     [        SS/5      n[        R                  " [        SS9   UR                  U     S S S 5        g ! , (       d  f       g = fNr0   r   as an indexer is not supportedr   r   r!   r"   r   rG   selfr   r;   s      r,   $test_getitem_dict_and_set_deprecated;TestDeprecatedIndexers.test_getitem_dict_and_set_deprecated  s8     aVn]]9,LMGGCL NMM   A  
Ar   c                     [        SS/[        R                  " SS/5      S9n[        R                  " [
        SS9   UR                  U     S S S 5        g ! , (       d  f       g = fNr0   r   rX   )r3   ri   r   r   r   r   r	   from_tuplesr!   r"   r   rG   r   s      r,   /test_getitem_dict_and_set_deprecated_multiindexFTestDeprecatedIndexers.test_getitem_dict_and_set_deprecated_multiindex  sK     aV:#9#966:J#KL]]9,LMGGCL NMM   A
A#c                     [        SS/5      n[        R                  " [        SS9   SUR                  U'   S S S 5        g ! , (       d  f       g = fr   r   r   s      r,   $test_setitem_dict_and_set_disallowed;TestDeprecatedIndexers.test_setitem_dict_and_set_disallowed  s:     aVn]]9,LMCGGCL NMMr   c                     [        SS/[        R                  " SS/5      S9n[        R                  " [
        SS9   SUR                  U'   S S S 5        g ! , (       d  f       g = fr   r   r   s      r,   /test_setitem_dict_and_set_disallowed_multiindexFTestDeprecatedIndexers.test_setitem_dict_and_set_disallowed_multiindex  sM     aV:#9#966:J#KL]]9,LMCGGCL NMMr    N)__name__
__module____qualname____firstlineno__r!   markparametrizer   r   r   r  __static_attributes__r  r?   r,   r   r     s    [[UaS1a&M2 3 [[UaS1a&A3(aVQK$HI J [[UaS1a&M2 3 [[UaS1a&A3(aVQK$HI Jr?   r   c            
          \ rS rSrS rSSSS\\R                  " S5      \R                  " S5      /r	SS/\
" SS	5      / S
Q\
" SSS5      /r\R                  R                  S\	S	S\R                  " S	5      \R                   " S	5      /-   5      \R                  R                  S\5      S 5       5       r\R                  R                  S\	SS\R                   " S5      /-   5      \R                  R                  S\5      S 5       5       r\R                  R                  S\	S/-   5      \R                  R                  S\5      S 5       5       rSrg)TestSetitemValidationi  c                 l   Sn[         R                  " U5      nUR                  5       n[        R                  " XES9   X!U'   UR                  5       nS S S 5        [        R                  " XES9   X!R
                  U'   UR                  5       nS S S 5        [        R                  " XES9   X!R                  U'   UR                  5       nS S S 5        [        R                  " XES9   X!S S & S S S 5        g ! , (       d  f       N= f! , (       d  f       N|= f! , (       d  f       NR= f! , (       d  f       g = f)Nz3Setting an item of incompatible dtype is deprecatedr   )r~   r   r   r$   r%   rO   rG   )r   r;   invalidr   warnr+   orig_sers          r,   _check_setitem_invalid,TestSetitemValidation._check_setitem_invalid  s    Ciin88:''8"L--/C 9 ''8 'HHW--/C 9 ''8&GGG--/C 9 ''8F 98 98 98 98 98s/   C2.D)D$D%2
D 
D
D"%
D3y      ?       @True1z1.0r
   r   r0   TFFNr  g      ?r   c                 J    [        / SQSS9nU R                  X1U[        5        g )Nr  boolr1   r   r  r&   )r   r  r   r;   s       r,   #test_setitem_validation_scalar_bool9TestSetitemValidation.test_setitem_validation_scalar_bool  s#    
 )8##C'=Ir?   Tr   c                     [        / SQUS9n[        U5      (       a'  U[        La  [        R                  " U5      (       d  S nO[
        nU R                  XAX55        g )NrY   r1   )r   r   r
   r   isnatr&   r  )r   r  any_int_numpy_dtyper   r;   r  s         r,   "test_setitem_validation_scalar_int8TestSetitemValidation.test_setitem_validation_scalar_int  sI     Y&9:==WC/8I8ID D##C'@r?   c                 J    [        / SQUS9nU R                  XAU[        5        g )N)r0   r   Nr1   r  )r   r  float_numpy_dtyper   r;   s        r,   $test_setitem_validation_scalar_float:TestSetitemValidation.test_setitem_validation_scalar_float  s#     \):;##C'=Ir?   r  )r  r  r  r  r  r
   r   
datetime64timedelta64_invalid_scalarsr   	_indexersr!   r  r	  r   float64r  r  r"  r
  r  r?   r,   r  r    s^   , 	
e
u QCq!&:E$d<STI[[#q#rxx{BJJqM&JJ [[Y	2J 3J [[Y(8D#rzzRU;W(WX[[Y	2A 3 YA [[Y(8D6(AB[[Y	2J 3 CJr?   r  )A__doc__datetimer   r~   numpyr   r!   pandas.errorsr   pandasr   r   r   r   r	   r
   r   r   r   r   r   r   r   r   pandas._testing_testingr$   r-   r>   rB   rJ   rP   rV   r  r	  r   r_   rc   rn   rr   rx   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r?   r,   <module>r0     s     	   '     8?<	-	# * rl+,As8Ar7&)	
 y\23Ay>"Ay>"	
$$&.&.6)1&/616((22# <$7Y5Rc2UU *-*- c3Z 01R%q"rxxPR|$TU( V 2( <9J 9Jr?   