
    Mhj                        S r SSKrSSKJr  SSKrSSKrSSKrSSKrSSKJ	r	  SSK
Jr  SSKJrJrJ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KJr  SS	KJr   " S
 S5      r  " S S5      r! " S S5      r" " S S5      r#S r$S r%\RL                  RO                  S\" S/5      S4\" / SQ\(" S5      S9S4/5      S 5       r)S r*\RL                  RO                  SSSS/\R                  " SS/5      \R                  " SSS/5      /5      S  5       r+\RL                  RO                  SSSS/\R                  " SS/5      \R                  " SSS/5      /5      S! 5       r,\RL                  RO                  SS"S#/\R                  " S#/5      \R                  " S$S#/5      /5      S% 5       r-\RL                  RO                  SS"S#/\R                  " S#/5      \R                  " S$S#/5      /5      S& 5       r.S' r/g)(ztest fancy indexing & misc     N)datetime)using_pyarrow_string_dtype)IndexingError)is_float_dtypeis_integer_dtypeis_object_dtype)	DataFrameIndexNaTSeries
date_rangeoffsetstimedelta_range)_mklbl)gen_objc                      \ rS rSrSrS rS r\R                  R                  S5      S 5       r
\R                  R                  S5      S 5       rS	 rS
 rS r\R                  R                  SSS/5      S 5       rS rS rS rS rS r\R                  R                  S/ SQ\" S5      /5      S 5       rS rS rS rS rS rS rS r\R                  R?                  \ " 5       S S!9S" 5       r!S# r"S$ r#S% r$S& r%S' r&S( r'\R                  R                  S)\(RR                  \(RT                  /5      S* 5       r+S+r,g,)-	TestFancy'   z"pure get/set item & fancy indexingc                    [        [        [        R                  " SS5      [        R                  S9S9n[        R
                  " S[        R                  S9US'   [        R
                  " S[        S9US'   Sn[        R                  " [        US	9   [        R                  " / S
Q5      UR                  UR                  SS S4'   S S S 5        [        R                  " / S
Q5      UR                  UR                  SS S4'   UR                  UR                  SS S4   n[        / S
Q/ SQSS9n[        R                   " X45        g ! , (       d  f       N= f)N      dtypeindex
   foobar@Must have equal len keys and value when setting with an iterablematch)y        p=
ף@yGz??g@      ?         )      r$   r%   r   name)r	   r
   nparangeint64zerosfloat64complexpytestraises
ValueErrorarraylocr   r   tmassert_series_equal)selfdfmsgresultexpecteds        U/var/www/html/env/lib/python3.13/site-packages/pandas/tests/indexing/test_indexing.pytest_setitem_ndarray_1d!TestFancy.test_setitem_ndarray_1d*   s    U299Q#3288DEHHRrzz25	HHRw/5	 Q]]:S1+-884R+SBFF288Aa=%'( 2 (*xx0N'Orxx!}e#$1u,-*,U
 	v0 21s   5E
Ec                 z   [        [        [        R                  " SS5      5      S9n[        R                  " S[        R
                  S9US'   [        R                  " S[        S9US'   Sn[        R                  " [        US	9   [        R                  " SS
5      S-  USS& S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   r   r   r   r   r    r'   y              ?r#   r$   )
r	   r
   r*   r+   r-   r.   r/   r0   r1   r2   )r7   r8   r9   s      r<   test_setitem_ndarray_1d_2#TestFancy.test_setitem_ndarray_1d_2@   s     U299Q#345HHRrzz25	HHRw/5	P]]:S1ii1o*BqG 211s   B,,
B:zPignore:Series.__getitem__ treating keys as positions is deprecated:FutureWarningc                    [        X!5      nU" U5      n[        R                  R                  S5      R	                  SSS9n/ nU[
        L aM  U[        R                  [        R                  4;   a)  UR                  S5        U(       a  UR                  S5        U[
        L d  U[        R                  L a)  UR                  S5        U(       a  UR                  S5        U[        R                  L d  U[
        L a$  U[        R                  L a  UR                  S	5        U[        L a$  U[        R                  L a  UR                  S
5        [        U[        R                  5      (       a$  U[        R                  L a  UR                  S
5        [        U[        R                  [        R                   [        R"                  45      (       a  UR                  S5        [%        U5      S:X  d  [        U[        R&                  5      (       a  UR                  S5        [)        U5      [*        L aK  [        UR,                  [        R.                  5      (       d"  UR                  S5        UR                  S5        SR1                  U5      n	[2        [4        [6        4n
[8        R:                  " XS9   Xg     S S S 5        g ! , (       d  f       g = f)Nr#   r$   r#   r#   r#   sizez8Wrong number of dimensions. values.ndim > ndim \[3 > 1\]z$Passed array should be 1-dimensional;Buffer has wrong number of dimensions \(expected 1, got 3\)zindexer should be 1-dimensionalz&Cannot index with multidimensional key Index data must be 1-dimensionalData must be 1-dimensionalr   z%positional indexers are out-of-boundszvalues must be a 1D arrayz only handle 1-dimensional arrays|r    )r   r*   randomdefault_rngintegersr   r5   setitemilocappendr4   r	   
isinstancepdIntervalIndexTimedeltaIndexDatetimeIndexPeriodIndexlen
MultiIndextyper
   _valuesndarrayjoin
IndexErrorr2   NotImplementedErrorr0   r1   )r7   r   frame_or_seriesindexer_sliusing_array_managerobjidxrnd3msgsr9   potential_errorss              r<   test_getitem_ndarray_3d!TestFancy.test_getitem_ndarray_3dL   s    o-3ii##A&//	/Bf$RWW8M)MKKST"BCf$rww(>KKVW"=>"&& v%+*CKK@Ai'K2::,EKK:;eR--..;"''3IKK:;eb//1A1A2>>RSSKK45u:?j>>KK?@;%
5=="**(M(MKK34 KK:;hhtn&
4GH]]+7I 877s   /J==
KzPignore:Series.__setitem__ treating keys as positions is deprecated:FutureWarningc                 l   [        X!5      nU" U5      n[        R                  R                  S5      R	                  SSS9nU[
        R                  L a  [        nSUR                   3nO[        nSR                  / SQ5      n[        R                  " XxS9   S	XV'   S S S 5        g ! , (       d  f       g = f)
Nr#   r$   rC   rD   zCannot set values with ndim > rI   )rF   zCannot set values with ndim > 1rG   rH   z,Array conditional must be same shape as selfr    r   )r   r*   rJ   rK   rL   r5   rN   r2   ndimr[   r0   r1   )	r7   r   r^   r_   ra   rb   rc   errr9   s	            r<   test_setitem_ndarray_3d!TestFancy.test_setitem_ndarray_3dz   s     o-3ii##A&//	/B"''!C2388*=CC((C ]]3*DI +**s   B%%
B3c                     [         R                  " S5      n[        SS/SS//5      nX!   n[        SS/SS9n[        R
                  " X45        [        SS/5      nXQ   nUS:X  d   eg )Nr   r   r#   r&   r'   r)   )r*   r3   r	   r   r5   r6   )r7   keyr8   r:   r;   sers         r<   test_getitem_ndarray_0d!TestFancy.test_getitem_ndarray_0d   so    hhqk AA'(1a&q)
v0 aVn{{    c                 n   [        S/S9nSUR                  S'   SUR                  S'   SUR                  [        R                  '   UR                  [        R                  S4   S:X  d   eUR                  n[        SS[        R                  /[        R                  S9n[        R                  " X#5        g )Nr   columnsr   r#   r&   r   )	r	   r4   r*   infr   r
   r.   r5   assert_index_equalr7   r8   r:   r;   s       r<   test_inf_upcastTestFancy.test_inf_upcast   s     s#q	q	rvv vvbffai A%%%!Qrzz:
f/rs   c           	         [        SS0SSS./5      n[        R                  US'   US   R                  [        R                  :X  d   e[
        R                  " [        SS9   S	UR                  S
'   S S S 5        [        SS/[        R                  S/[        S	[        R                  /[        S9S.5      n[
        R                  " X5        g ! , (       d  f       Nb= f)Nar   r&   r#   r}   bcitem of incompatible dtyper    r   )r   r   r   r}   r   r   )r	   r*   nanr   r.   r5   assert_produces_warningFutureWarningr4   r   objectassert_frame_equalr7   r8   r;   s      r<   test_setitem_dtype_upcast#TestFancy.test_setitem_dtype_upcast   s    a"234&&3#w}}

***''!=
 #BFF6N
 a&{v1VW
 	b+
 
s   "C
CvalgQ	@wxyzc                    [        [        R                  " SSS9R                  SS5      [	        S5      / SQS9nUR                  5       n[        R                  " [        S	S
9   XR                  S'   S S S 5        [        SUS// SQ/[	        S5      / SQS9n[        R                  " X45        [        US   5      (       d   e[        US   5      (       d   eg ! , (       d  f       Nl= f)Nr%   r,   r   r#   r&   abr   r   bazr   rv   r   r    r}   r   r   r&   r'   r$   r   r   )r	   r*   r+   reshapelistcopyr5   r   r   r4   r   r   )r7   r   r8   leftrights        r<   test_setitem_dtype_upcast2$TestFancy.test_setitem_dtype_upcast2   s     IIaw'//15t*)
 wwy''!=
 $'HHZ 
 a[)$t*)
 	d*U,,,,U,,,,
 
s   #C
C&c                    [        [        R                  " SSS9R                  SS5      S-  [	        S5      / SQS	9n[
        R                  " [        S
S9   SUR                  S'   S S S 5        [        / SQ/ SQ/[	        S5      / SQS	9n[
        R                  " X5        [        US   5      (       d   e[        US   5      (       d   eg ! , (       d  f       Nk= f)Nr%   r,   r   r#   r&   g      $@r   r   r   r   r    r   r   )r   r   g?)g333333?g?g      ?r   r   )r	   r*   r+   r   r   r5   r   r   r4   r   r   )r7   r   r   s      r<   test_setitem_dtype_upcast3$TestFancy.test_setitem_dtype_upcast3   s    IIaw'//15<t*)

 ''!=
 $*DHHZ 

 /t*)
 	d*d5k****d5k****
 
s   C
Cc                     [        [        R                  " S5      / SQS9nUSS/   R                  n[	        / SQ5      n[
        R                  " X#5        g )Nr&   )r}   r}   r   ru   r   r}   )r   r}   r}   )r	   r*   eyerv   r
   r5   rx   ry   s       r<   test_dups_fancy_indexing"TestFancy.test_dups_fancy_indexing   sC     rvvay/:S#J'')
f/rs   c                     [        / SQ/[        S5      S9n[        / SQ/5      n[        S5      Ul        UR                  S S 2S4     UR                  S S 2S4     [        R
                  " X5        g )N)r   r#   r"          @      @r   r   aaaaaaaru   r'   )r	   r   rv   rN   r5   r   r7   r8   r:   s      r<   &test_dups_fancy_indexing_across_dtypes0TestFancy.test_dups_fancy_indexing_across_dtypes   s^    ;<d9oV?@Ai 	1AqD
b)rs   c                 ,   [        / SQ/ SQ[        S5      S./ SQS9nSS/n[        S	S
/SS/SS/S.US9nUR                  U   n[        R                  " XC5        UR                  [        U5         n[        R                  " XC5        / SQn[        R                  " [        SS9   UR                  U     S S S 5        / SQn[        R                  " [        SS9   UR                  U     S S S 5        g ! , (       d  f       ND= f! , (       d  f       g = f)Nr$      	   r   )      @r$   r%   r   abcd)testtest1otherAr   BCr   r   r   r   r   g      @r%   dr   )r   r   Enot in indexr    )FGHr   r   r   )	r	   r   r4   r5   r   r
   r0   r1   KeyError)r7   r8   rowsr;   r:   s        r<   %test_dups_fancy_indexing_not_in_order/TestFancy.test_dups_fancy_indexing_not_in_order  s    "^d6lS&
 Sz!WQx3*ET
 
f/d$
f/]]8>:FF4L ; .]]8>:FF4L ;: ;:
 ;:s   &C4D4
D
Dc                    [        [        R                  R                  S5      R	                  S5      [        S5      S9nU(       aG  [        R                  " [        [        R                  " S5      S9   UR                  S/     S S S 5        g [        R                  " [        [        R                  " S5      S9   UR                  S/     S S S 5        g ! , (       d  f       g = f! , (       d  f       g = f)	Nr#   )r$   r&   AABCDr   z;"None of [Index(['E'], dtype='string')] are in the [index]"r    r   z;"None of [Index(['E'], dtype='object')] are in the [index]")r	   r*   rJ   rK   standard_normalr   r0   r1   r   reescaper4   )r7   using_infer_stringdfnus      r<   +test_dups_fancy_indexing_only_missing_label5TestFancy.test_dups_fancy_indexing_only_missing_label   s    II!!!$44V<DM
 iiS #  iiS #   s   4C;C&
C#&
C4vals)r   r   r#   abcc                     [        SU05      n[        R                  " [        SS9   UR                  / SQ     S S S 5        g ! , (       d  f       g = f)Nr   r   r    )r      r   r	   r0   r1   r   r4   )r7   r   r8   s      r<   &test_dups_fancy_indexing_missing_label0TestFancy.test_dups_fancy_indexing_missing_label6  s8     T{#]]8>:FF9 ;::s   A
Ac                     [        S/ SQ0/ SQS9n[        R                  " [        SS9   UR                  / SQ     S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   r   r    )r   r   r   r   r7   r8   s     r<   #test_dups_fancy_indexing_non_unique-TestFancy.test_dups_fancy_indexing_non_unique=  s:    .6JK]]8>:FF?# ;::s   A
Ac                    [        [        R                  R                  S5      R	                  S5      / SQS9n[
        R                  " [        SS9   UR                  S S 2/ SQ4     S S S 5        g ! , (       d  f       g = f)Nr#   )r$   r$   )r   r   r   r   r   ru   r   r    )r   r   r   )	r	   r*   rJ   rK   r   r0   r1   r   r4   r   s     r<   test_dups_fancy_indexing2#TestFancy.test_dups_fancy_indexing2C  s[     II!!!$44V<-

 ]]8>:FF1o%& ;::s   A11
A?c                    [        [        R                  R                  S5      R	                  S5      / SQSS/S9nUR
                  SS nUR                  S	S/   n[        R                  " X25        UnUR                  S S 2SS/4   n[        R                  " X25        UR
                  SS2S S 24   nUR                  S	S/SS/4   n[        R                  " X25        g )
Nr#   )r   r#   )	r   r   r   r#   r#   r#   r&   r&   r&   r}   r   r   r   r%   r   )	r	   r*   rJ   rK   r   rN   r4   r5   r   )r7   r8   r;   r:   s       r<   test_dups_fancy_indexing3#TestFancy.test_dups_fancy_indexing3N  s    II!!!$44V<-#J
 771Q<A
f/C:&
f/771Q36?Ac
*+
f/rs   c                     [        [        S5      / SQS9n[        [        S5      SS/S9nU" U5      S/   n[        R                  " XC5        g )Nr&   r   r   r&   r   r#   r   )r   ranger5   r6   )r7   
indexer_slrp   r;   r:   s        r<   test_duplicate_int_indexing%TestFancy.test_duplicate_int_indexingb  sD    U1XY/%(1a&1C!%
v0rs   c                     [        SSSS.SSSS.S.5      nUS	   R                  S
 5      US'   US   S:H  nUR                  US	4   R                  S 5      nX1R                  US4'   UR                  S   S:X  d   eg )Naaabbbcccr   r#   r&   o      iM  r~   r}   c                     U S:X  a  S$ U $ )Nr   _ xs    r<   <lambda>9TestFancy.test_indexing_mixed_frame_bug.<locals>.<lambda>p  s    AJS-EA-Ers   r   r   c                     U S:X  a  S$ U $ )Nr   -----r   r   s    r<   r   r   t  s    1:0L10Lrs   )r   r#   r   )r	   applyr4   rN   )r7   r8   idxtemps       r<   test_indexing_mixed_frame_bug'TestFancy.test_indexing_mixed_frame_bugi  s    %03337OP

 W]]#EF6
 jCvvc3h%%&LM"sF{wwt}'''rs   c                 b   [        [        R                  R                  S5      R                  S5      S// SQ-   S9n[        R
                  " [        [        R                  " S5      S9   U/ SQ     S S S 5        US	   R                  S
   UR                  S
   :X  d   eg ! , (       d  f       N4= f)Nr#   )r   r$   r}   )            ru   z'[26, -8] not in index'r    )r      ir   r   )
r	   r*   rJ   rK   r0   r1   r   r   r   shaper   s     r<    test_multitype_list_index_access*TestFancy.test_multitype_list_index_accessx  s    II!!!$++G4seFV>V
 ]]82995N+OP| Q"v||A"((1+--- QPs   *B  
B.c                 l   [        SSSSSSSSSSS	S
SSS.SSS[        R                  [        R                  [        R                  [        R                  S[        R                  [        R                  [        R                  [        R                  [        R                  [        R                  S.SSSSSSSSSSSSSSS.SSSS S S S S S S S S S S!S.S".5      R                  5       nUR	                  / S#Q5      R                  5       R                  UR                  S$9n[        R                  " X!5        g )%NnonQC10111213243546474859)         r   r   r   r         r                       r"   goe@gm77M @g,o@g2?g@?gufn?g%?gTFa?gI"?g+ew7?gy7Z?g9?gGp#?gC?i  i  i  i  i  )PRuidQCdatayear)r  r  r  ru   )	r	   r*   r   reset_index	set_indexreindexrv   r5   r   r   s      r<   test_set_index_nanTestFancy.test_set_index_nan  sJ     " " +**++*++++++++" cAC
F +-G 	L LL01[]WRZZW( 	
 	f)rs   zcan't multiply arrow stringsreasonc                 z   [        / SQ/ SQ[        [        S5      5      [        [        SS5      5      S.5      R                  SS05      n[        R
                  UR                  S'   UR                  5       nUR                  R                  5       ) nS	S/nUS
-  n[        R
                  UR                  S'   [        S[        R
                  SSSS// SQ[        / SQ5      SSS[        R
                  SS/S.5      nUR                  X44   UR                  X44'   [        R                  " X&5        [        S[        R
                  SSSS// SQ/ SQSSS[        R
                  SS/S.5      nUR                  5       nUR                  X44   R                  UR                  X44'   [        R                  " X&5        g )N)r}   r   r}   r   r}   r   )r   r   r   r   r   r   r%      )FCPFcol1col2r$  r.   r   r   r#  r#   )r&   r&   r}   r   )r   r   r'   r%   r   r   r      r   r   )r	   r   r   astyper*   r   rN   r   r!  isnar   r4   r5   r   values)r7   r8   df2maskcolsdftr;   s          r<   test_multi_assignTestFancy.test_multi_assign  s    4(U1XU1b\*	
 &&)$
% 	 ggi~AgBFFCc37(23QBFFB3	
 "ggdj1

c,
 BFFCc37(+QBFFB3	
 ggi!ggdj188

c,rs   c                    [        / SQ/ SQ/ SQ/ SQS.5      nUR                  5       nUS   S:H  nS H  nUS   UR                  X44'   M     US   R                  5       UR                  US   S:H  SS	/4'   [        R                  " X5        g )
N)r   r#   r   r   r   )r   r   r   r   r   )r&   r'   r$   r%   r   )r   r   r   Dr   r   r   r   r1  r   )r	   r   r4   r5   r   )r7   r8   r;   r+  cols        r<   "test_multi_assign_broadcasting_rhs,TestFancy.test_multi_assign_broadcasting_rhs  s    $&&$	
 779}!C&(gHLL#  ,.c7<<>r#w!|c3Z'(
b+rs   c                     [        SS/S/S9n/ SQUR                  S'   SS/UR                  S'   [        SS/S/S9nSS/UR                  S'   [        R                  " X!5        g )Nr   r   r   r   r%  r#   )r	   rN   r5   r   r   s      r<   test_setitem_listTestFancy.test_setitem_list  sf     aVaS1!A!Q!5FD
f)rs   c                    [        S/[        [        R                  " S5      /[        S95      nUR
                  R                  (       d   e[        R                  " [        SS9   US     S S S 5        [        R                  " [        SS9   UR                  S     S S S 5        g ! , (       d  f       N@= f! , (       d  f       g = f)Nr   z
2011-01-01r   '2011'r    2011r;  r   )r	   r
   rQ   	Timestampr   r   _is_all_datesr0   r1   r   r4   r   s     r<   test_string_sliceTestFancy.test_string_slice"  s     sE2<<#=">fMNxx%%%%]]884vJ 5 ]]884FF9 54 54 54s   %B%B6%
B36
Cc                 F   [        5       nUR                  R                  (       a   e[        R                  " [
        SS9   US     S S S 5        [        R                  " [
        SS9   UR                  S     S S S 5        g ! , (       d  f       N@= f! , (       d  f       g = f)Nr:  r    r;  z^0$r<  )r	   r   r>  r0   r1   r   r4   r   s     r<   test_string_slice_empty!TestFancy.test_string_slice_empty.  sn     [88))))]]884vJ 5 ]]851FF9 21 54 21s   B(B
B
B c                 f   [        / SQ/[        S5      S9nUR                  5       nUR                  S S 2SS24   R	                  [
        R                  5      UR                  S S 2SS24'   [        / SQ/[        S5      S9nU(       d6  US   R	                  [        5      US'   US   R	                  [        5      US'   [        R                  " X45        UR                  5       nUR                  S S 2S4   R	                  [
        R                  5      UR                  S S 2S4'   [        / S	Q/[        S5      S9nU(       d  US   R	                  [        5      US'   [        R                  " X45        UR                  5       nUR                  S S 2SS
/4   R	                  [
        R                  5      UR                  S S 2SS
/4'   [        / SQ/[        S5      S9nU(       d6  US   R	                  [        5      US'   US
   R	                  [        5      US
'   [        R                  " X45        g )N)123.4r$         @r   ABCDEFGru   r   r#   )r   r#   rG  rH  r$   rI  r   r   r   )r   rF  rG  rH  r$   rI  r   r   )rE  r#   r&   rH  r$   rI  r   )r	   r   r   rN   r'  r*   r,   r   r5   r   r4   )r7   r   df_origr8   r;   s        r<   test_astype_assignment TestFancy.test_astype_assignment9  s   12DO
 \\^ ''!QqS&/00:1Q3-.Y
 "$SM008HSM$SM008HSM
b+ \\^3..rxx8q#v/0$y/
 "$SM008HSM
b+\\^ "q3*} 5 < <RXX Fq3*}-.Y
 "$SM008HSM$SM008HSM
b+rs   c                    [        S/ SQ05      nUS   R                  [        R                  5      UR                  S S 2S4'   [        S/ SQ05      n[
        R                  " X5        [        S/ SQ05      nUS   R                  [        R                  5      UR                  S S 2S4'   [
        R                  " X5        g )Nr   )r"   r   r   r   r   )r	   r'  r*   r,   rN   r5   r   r4   r   s      r<   (test_astype_assignment_full_replacements2TestFancy.test_astype_assignment_full_replacements`  s    123 3rxx01c#789
b+123C1q#v
b+rs   indexerc                    [        [        S5      5      [        [        S5      [        SS5      S94 GH  n[        UR                  5      (       d   eUR	                  5       nSU" U5      S'   [        UR                  5      (       d   eU" U5      S   S:X  d   eUR	                  5       nSU" U5      S'   UR                  nSU;  a'  [        UR                  R                  5       S/-   5      n[        R                  " UR                  U5        UR	                  5       nSU" U5      S'   [        UR                  5      (       a  GM   e   [        [        S5      [        R                  " S	5      S94 H  n[        UR                  5      (       d   eUR	                  5       nSU" U5      S'   [        UR                  5      (       d   eU" U5      S   S:X  d   eUR	                  5       nSU" U5      S'   [        R                  " UR                  UR                  5        UR	                  5       nSU" U5      S'   [        UR                  5      (       a  M   e   g )
Nr$   r   r%   r   r   g?r  0g      @)r   r   r   r   r   r   r
   tolistr5   rx   r   r*   r+   )r7   rQ  ss2exps        r<   test_index_type_coercion"TestFancy.test_index_type_coercionn  s    q"F585A;$GHA#AGG,,,,B GBK!"((++++2;s#q(((B GBK''CzAGGNN,s23!!"((C0B GBK"288,,,,# I& q389A!!''****B GBK!"((++++2;s#q(((B GBK!!"((AGG4B GBK"288,,,, :rs   r   N)-__name__
__module____qualname____firstlineno____doc__r=   r@   r0   markfilterwarningsrf   rk   rq   rz   r   parametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r   r  xfailr   r.  r4  r7  r?  rB  rL  rO  r5   getitemr4   rX  __static_attributes__r   rs   r<   r   r   '   s   ,1,
+ [[	(	(T [[		00$, [[UT6N3- 4-.+*0
*2 , [[Vie%=> ?$	'0(1(.L*\ [["$-K  .-.-`,&
*
	%,N, [[YRVV(<=)- >)-rs   r   c            
       ~   \ rS rSrS rS rS r\R                  R                  \
" 5       SS9S 5       r\R                  R                  S\" S	S
5      \R                  " S
5      S-   \R                   " SSS
5      /5      S 5       rS rS rS r\R                  R                  S/ SQ5      S 5       rS rS rS rSrg)TestMisci  c                 8   [        [        R                  R                  S5      R                  S5      [        R                  R                  S5      R                  S5      S.5      nSUS'   [        US   US   S/S-  S.5      n[        R
                  " X!5        g )Nr#   r   )r  r"   r}   r  r"   )r  r"   r}   )r	   r*   rJ   rK   r5   r   r   s      r<   test_float_index_to_mixed"TestMisc.test_float_index_to_mixed  s    YY**1-44R8YY**1-44R8
 32c7CtbyIJ
h+rs   c                     [        / SQ/ SQS./ SQS9nSUR                  UR                  S S '   [        / SQ/ S	QS.UR                  S9n[        R                  " X!5        g )
Nr   r   r~   r"   r   r   r   r   r#   r   )r   r   r$   )r	   r4   r   r5   r   r   s      r<   &test_float_index_non_scalar_assignment/TestMisc.test_float_index_non_scalar_assignment  sN    YY7O rxx|99=RXXN
h+rs   c                     [        / SQ/ SQS./ SQS9nUR                  5       nUR                  UR                     UR                  UR                  '   [        R
                  " X5        g )Nr   r   r~   rk  r   )r	   r   r4   r   r5   r   r7   r8   r*  s      r<    test_loc_setitem_fullindex_views)TestMisc.test_loc_setitem_fullindex_views  sK    YY7Oggi66"((+rxx
b&rs   zcan't set int into stringr  c                    S n[         R                  " S5      R                  SS5      n/ SQn[        X#[	        S5      SS9nS	UR
                  S
SS2SSS24   -  nUR                  5       n/ SQUS'   / SQUS'   UR
                  SS2SS
24==   S	-  ss'   UR                  5       nUR
                  SS2SS
24==   S	-  ss'   U" XEXv5        XEXv4 H2  nUS   R                  S5      US'   US   R                  S 5      US'   M4     / SQUS'   / SQUS'   [        R                  " [        SS9   U" XEXv5        S S S 5        g ! , (       d  f       g = f)Nc                    [        S5      / SQ[        SS5      penSS/SS/[        SS5      pnU R                  5       n
XR                  XG4'   [        R
                  " X5        U R                  5       n
XR                  XX4'   [        R
                  " X5        U R                  5       n
XR                  Xi4'   [        R
                  " X5        g )	Nbcdr   r   r'   joejolier#   r&   )r   slicer   r4   r5   r   rN   )r8   rhs	right_loc
right_iloclbl_oneidx_one	slice_onelbl_twoidx_two	slice_twor   s              r<   	run_tests.TestMisc.test_rhs_alignment.<locals>.run_tests  s    *.u+y%1+iG+0'*:QFE!QKiG779D),HHW%&!!$2779D*-IIg&'!!$3779D.1IIi*+!!$3rs   r   r$   r'   )jimru  rv  jolineabcder,   )rv   r   r   r&   r   r#   )r      r   r%   r
  ru  )r#      r   r$   r  rv  r   r.   c                     SU  3$ )N@r   r   s    r<   r   -TestMisc.test_rhs_alignment.<locals>.<lambda>  s
    AaS'rs   )r"   z@-28z@-20z@-12g      1@)z@2g      :g      2g      $z@18zincompatible dtyper    )r*   r+   r   r	   r   rN   r   r'  mapr5   r   r   )	r7   r  xsr,  r8   rx  rz  ry  frames	            r<   test_rhs_alignmentTestMisc.test_rhs_alignment  sX   	4" YYr]""1a(0rtG}GL 2771Qr61Qr6>** WWY
.
5/
7!QqS!R'!GGI	qsAaCx B&  	"91 y5E <..y9E%L"7^//0ABE'N 6 @
5@
7''=QRby5 SRRs   .
E
Er   r   r   d      c                    [         R                  n[        U5      n[        [        R
                  " S5      US9n[        R                  " X2US   S S2   USS S2   5        [        R                  " X2S US   S2   US SS2   5        [        R                  " X2US   US   S2   USSS2   5        [        R                  " X2US   US   S2   US S 5        g )Nr   r   r   r  r   r  r   )rQ   
IndexSlicer
   r   r*   r+   r5   !assert_indexing_slices_equivalent)r7   r   SLCrp   s       r<   )test_str_label_slicing_with_negative_step2TestMisc.test_str_label_slicing_with_negative_step  s     mmCjRYYr]#.
,,Sc!fll2CSBZP
,,SmAm2Dc%1R%jQ
,,SWs1v*+SAb\	
 	,,Sc!fs2w6K2LcRTSTgVrs   c                     U" [         R                  " [        U5      5      US9n[        R                  " [
        SS9   U" U5      S S S2     S S S 5        g ! , (       d  f       g = f)Nr   zslice step cannot be zeror    r   )r*   r+   rV   r0   r1   r2   )r7   r   r   r^   ra   s        r<    test_slice_with_zero_step_raises)TestMisc.test_slice_with_zero_step_raises  sG    biiE
35A]]:-HIsOCaC  JIIs   A
A$c                 ^   [        / SQSS9n[        / SQ/ SQS.US9nUR                  5       nSS	S.nXBR                  S
'   SS	/UR                  S
'   [        R
                  " X#5        [        / SQ/ SQS.US9nXBR                  S
'   [        / SQ/ SQS.US9n[        R
                  " X#5        g )N)r   r$   zrn   )r   r#   r%   )r#   r#   r   )r   yr   r   c   r$   )r   r   g       @)r   r#   r   )r   r   g     X@)r
   r	   r   r4   r5   r   )r7   r   r8   r;   rx  s        r<   8test_loc_setitem_indexing_assignment_dict_already_existsATestMisc.test_loc_setitem_indexing_assignment_dict_already_exists  s    js+YY7uE779Bq	b'Q
b+ Y_=UKq	93CDER
b+rs   c                 "   [        / SQ/ SQS.5      nUR                  / S S 24   nUR                  S S 2S4   R                  [        R
                  :X  d   e[        R                  " UR                  S S 2S4   UR                  S S 2S4   5        g )Nr   )r   b2b3r~   r}   r   )r	   rN   r4   r   r*   r,   r5   r6   ro  s      r<   *test_iloc_getitem_indexing_dtypes_on_empty3TestMisc.test_iloc_getitem_indexing_dtypes_on_empty  so    Y->?@ggb!enwwq#v$$000
swwq#vA?rs   rE   )r$   i?B i@B c                    [        [        U5      [        R                  S9nSUR                  [        S5      '   [
        R                  " UR                  [        S5         [        SS/S95        SUR                  [        S5      '   [
        R                  " UR                  [        S5         [        S	SS/S95        g )
N)r   r   *   r   g      E@r   r   +   r#   g     E@)r   r   r*   r.   r4   r5   r6   )r7   rE   rU  s      r<   !test_loc_range_in_series_indexing*TestMisc.test_loc_range_in_series_indexing  s     tBJJ7eAh
quuU1XtA30GHeAh
quuU1XtAq60JKrs   c                    [        [        R                  " S5      R                  SS5      [	        S5      [	        S5      S9n[        S[	        S5      [	        S5      S9nXR                  5          n[        [        R                  " S	S
[        R                  /SS[        R                  /[        R                  /S-  /5      [	        S5      [	        S5      S9n[        R                  " X45        g )Ng      "@r&   r   ABCr   r   r   ABr  r"   r   r   )
r	   r*   r+   r   r   notnullr3   r   r5   r   )r7   r8   index_dfr:   r;   s        r<   #test_partial_boolean_frame_indexing,TestMisc.test_partial_boolean_frame_indexing  s    IIcN""1a(UT%[
 Qd4j$t*E$$&'HHsC(3RVV*<rvvhlKLu+K

 	f/rs   c                     [        SS/SS/S.5      nS H  n[        X5        M     [        R                  " U5      nAU" 5       b   eg )Nr   r   r#   r&   r~   )r4   rN   atiat)r	   getattrweakrefref)r7   r8   r)   wrs       r<   test_no_reference_cycle TestMisc.test_no_reference_cycle(  sG    aV1a&120DB 1[[_t||rs   c                     [        SSSU/5      nUR                  SS9nUR                  U   nX1   nSnXF:X  d   eXV:X  d   eg )Nr   z{1,2}F)dropna)r   value_countsr4   )r7   nulls_fixturer8   vcresult1result2r;   s          r<   test_label_indexing_on_nan#TestMisc.test_label_indexing_on_nan0  sX    QM23__E_*&&'#""""""rs   r   N)rZ  r[  r\  r]  rh  rl  rp  r0   r_  rb  r   r  ra  r   r*   r+   linspacer  r  r  r  r  r  r  r  rd  r   rs   r<   rf  rf    s    
,,' [[13<WX-6 Y-6^ [[sB2!4bkk#sB6OP
W
W!
,@ [[V%9:L ;L0	#rs   rf  c            
          \ rS rSr/ SQ\R
                  SS/\4/ SQ\R
                  SS/S4\" S	S
S
5      \" S	S
S5      \" S	S
S5      /\\" S	S
S5      \" S	S
S5      /S4/ SQ/ SQS4/r	\
R                  R                  S\	5      S 5       r\
R                  R                  S\	5      S 5       r\
R                  R                  S\	5      S 5       rS rSrg)TestDataframeNoneCoercioni<  r   r#   r&   rk  r   r   N  r   r   )Nr   r   r;   c                     Uu  p#n[        SU05      nS UR                  SS/4'   [        SU05      n[        R                  " XV5        g Nr   r   r	   r4   r5   r   r7   r;   
start_dataexpected_resultwarnstart_dataframeexpected_dataframes          r<   test_coercion_with_loc0TestDataframeNoneCoercion.test_coercion_with_locK  sO    ,4)
T#UJ$78*.AwJ'&'?@
oBrs   c                     Uu  p#n[        SU05      nS XUS   US   S   :H  '   [        SU05      n[        R                  " XV5        g r  )r	   r5   r   r  s          r<   (test_coercion_with_setitem_and_dataframeBTestDataframeNoneCoercion.test_coercion_with_setitem_and_dataframeU  sU    ,4)
T#UJ$78OS./%2H2KKL&'?@
oBrs   c                     Uu  p#n[        SU05      nS UR                  US   US   S   :H  '   [        SU05      n[        R                  " XV5        g r  r  r  s          r<   $test_none_coercion_loc_and_dataframe>TestDataframeNoneCoercion.test_none_coercion_loc_and_dataframe_  s]    ,4)
T#UJ$78SWOE2oe6LQ6OOP&'?@
oBrs   c                 d   [        / SQ/ SQ[        SSS5      [        SSS5      [        SSS5      // SQS.5      nS UR                  S	'   [        [        R                  SS/[        R                  S
S/[
        [        SSS5      [        SSS5      // SQS.5      n[        R                  " X5        g )Nr   rk  r  r   r#   r&   r   )r}   r   r   r   r   r   r   )Nr   r   )r	   r   rN   r*   r   r   r5   r   )r7   r  rW  s      r<   test_none_coercion_mixed_dtypes9TestDataframeNoneCoercion.test_none_coercion_mixed_dtypesi  s    #$tQ*HT1a,@(4QRTUBVW$	
 #'Qffa^ffc3'8D!Q/$11EF%	
 	o3rs   r   )rZ  r[  r\  r]  r*   r   r   r   r   EXPECTED_SINGLE_ROW_RESULTSr0   r_  ra  r  r  r  r  rd  r   rs   r<   r  r  <  s    
RVVQNM2	2663,d3 dAq!8D!Q#7$19MN(4A&q!(<=	
 
 4d;# [[Z)DEC FC [[Z)DEC FC [[Z)DEC FC4rs   r  c            
          \ rS rSrS r\R                  R                  S\\	R                  \R                  \R                  \/5      \R                  R                  SSS/\" SS5      \	R                  " / SQ5      /5      S	 5       5       r\R                  R                  S
S\R"                  " S5      /5      S 5       r\R                  R                  S\\	R                  \R                  \R                  \/5      \R                  R                  SSS/\" SS5      \	R                  " / SQ5      /5      S 5       5       rSrg)TestDatetimelikeCoercioni  c                 L   Un[        SSUS9n[        UR                  SS95      nUR                  nSnUR	                  U5        Xr" U5      S'   UcB  UR
                  UR
                  :X  d   eUR                  R                  UR                  L d   eg UR                  UL d   eg )N
2016-01-01r&   periodstzTdeepz
2018-01-01r   )r   r   r   rY   _validate_setitem_valuer   _ndarray)r7   tz_naive_fixturer_   r  dtirp   r)  newvals           r<   test_setitem_dt64_string_scalar8TestDatetimelikeCoercion.test_setitem_dt64_string_scalar  s    qR8SXX4X()&&v.$C:99		)));;''6??:::;;&(((rs   boxro   r   r   r#   )TTFc                    Un[        U[        5      (       a  U[        R                  L a  [        SS5      n[	        SSUS9n[        UR                  SS95      nUR                  nU" SS	/5      n	UR                  U	5        X" U5      U'   UcB  UR                  UR                  :X  d   eUR                  R                  UR                  L d   eg UR                  UL d   eg )
Nr   r   r  r&   r  Tr  z
2019-01-01z
2010-01-02)rP   rw  r5   r4   r   r   r   rY   r  r   r  )
r7   r  r_   ro   r  r  r  rp   r)  newvalss
             r<   test_setitem_dt64_string_values8TestDatetimelikeCoercion.test_setitem_dt64_string_values  s     c5!!kRVV&;1+CqR8SXX4X()|\23&&w/ 'C:99		)));;''6??:::;;&(((rs   scalarz3 Daysr'   c                     [        SSS9n[        UR                  SS95      nUR                  nUR	                  U5        X!" U5      S'   UR                  R
                  UR
                  L d   eg )N1 Dayr&   r  Tr  r   )r   r   r   rY   r  r  )r7   r_   r  tdirp   r)  s         r<   test_setitem_td64_scalar1TestDatetimelikeCoercion.test_setitem_td64_scalar  sg     gq1SXX4X()&&v.$C{{##v666rs   c                 \   [        U[        5      (       a  U[        R                  L a  [        SS5      n[	        SSS9n[        UR                  SS95      nUR                  nU" SS	/5      nUR                  U5        Xq" U5      U'   UR                  R                  UR                  L d   eg )
Nr   r   r  r&   r  Tr  z10 Daysz44 hours)
rP   rw  r5   r4   r   r   r   rY   r  r  )r7   r_   ro   r  r  rp   r)  r  s           r<   test_setitem_td64_string_values8TestDatetimelikeCoercion.test_setitem_td64_string_values  s     c5!!kRVV&;1+Cgq1SXX4X()y*-.&&w/ 'C{{##v666rs   r   N)rZ  r[  r\  r]  r  r0   r_  ra  r   r*   r3   rQ   Categoricalr
   rw  r  r   Hourr  r  rd  r   rs   r<   r  r    s"   )* [[UT288RXXr~~u$UV[[AaRXX.A%BC) W)0 [[X',,q/'BC	7 D	7 [[UT288RXXr~~u$UV[[AaRXX.A%BC7 W7rs   r  c                  N   [        [        R                  " SS/SS9[        R                  " SS/SS9S.SS	/S
9n [        [        R                  " SS/SS9SS/SS9nU R                  S   n[
        R                  " X!5        U R                  S   n[
        R                  " X!5        g )Nr   r#   Int64r   r&   r'   r2  r}   r   r   r   r   r(   r   )r	   rQ   r3   r   r4   r5   r6   rN   )r8   r;   r:   s      r<   "test_extension_array_cross_sectionr    s    	1a&01a&0	
 Cj
B bhh1vW5c3ZcRHVVC[F6,WWQZF6,rs   c                  Z   [        [        R                  " SS/SS9[        R                  " SS/SS9S.SS/S	9n U R                  S   n[        SS/SS
S/SS9n[        R                  " X5        U R                  S   n[        R                  " X5        [        [        R                  " SS/SS9[        R                  " SS/5      S.SS/S	9n U R                  S   n[        SS/[        S
S/SS9n[        R                  " X5        U R                  S   n[        R                  " X5        g )Nr   r#   r  r   r,   r2  r}   r   r   r   r   )r   r   r)   r   )
r	   rQ   r3   r*   r4   r   r5   r6   rN   r   )r8   r:   r;   s      r<   +test_extension_array_cross_section_convertsr    s   	1a&01a&0	
 Cj
B VVC[Fq!fGC:CHH6,WWQZF6, 
hh1vW-BHHc3Z4HICj
B VVC[Fq#hfS#JSIH6,WWQZF6,rs   z	ser, keysr   )r   r   r   r   r   )r   r   c                 ,   [         SpC[        R                  " X4S9   U" U 5      U     S S S 5        U[        R                  :X  a  [
        SpC[        R                  " X4S9   SU" U 5      U'   S S S 5        g ! , (       d  f       NR= f! , (       d  f       g = f)NzToo many indexersr    ztoo many indices for arrayr   )r   r0   r1   r5   rN   r\   )rp   keys
indexer_liexp_errexp_msgs        r<   'test_ser_tup_indexer_exceeds_dimensionsr     ss     %&9W	w	.3 
/ RWW%'C	w	. !
3 
/	. 
/	. 
/	.s   A4B4
B
Bc                     [        S/5      nU " U5      SS/   n[        SS/[        SS/5      S9n[        R                  " X#5        g )Nr   r   r   )r   r
   r5   r6   )r  rp   resrW  s       r<   (test_ser_list_indexer_exceeds_dimensionsr    sG     ",C
S/1a&
!C
"b1v
/C3$rs   valuer   r   c                    [        S/ SQ05      nSR                  SS/5      n[        R                  " [        US9   XR
                  S'   S S S 5        [        / SQ[        R                  " / SQ[        S	9S
.5      n[        R                  " [        SS9   XR
                  S'   S S S 5        g ! , (       d  f       Nh= f! , (       d  f       g = f)Nr   r   rI   z"Must have equal len keys and value(setting an array element with a sequencer    r   r   r   r}   r   r   r2  )	r	   r[   r0   r1   r2   r4   r*   r3   r   )r  r8   r9   s      r<   %test_scalar_setitem_with_nested_valuer    s    
 
C#	$B
((06	
C 
z	-v 
.
 
-v)NO	PB	z)M	Nv 
O	N 
.	- 
O	Ns   B-B>-
B;>
Cc                    [        / SQ5      n[        R                  " [        SS9   X" U5      S'   S S S 5        [        / SQ[        S9nX" U5      S'   [        U [        R                  5      (       a(  UR                  S   U :H  R                  5       (       d   eg UR                  S   U :X  d   eg ! , (       d  f       N= f)Nr   r  r    r   r  r   )
r   r0   r1   r2   r   rP   r*   rZ   r4   all)r  r_   rp   s      r<   ,test_scalar_setitem_series_with_nested_valuer  6  s    
 
C	z)S	T#C 
U f
-CKQ%$$
e#((****wwqzU""" 
U	Ts   B11
B?)r  r  r   c                    [        S/ SQ05      nXR                  S'   [        / SQS[        R                  [        R                  /S.5      n[        R
                  " X5        [        / SQ[        R                  " / SQ[        S9S.5      nXR                  S'   [        U [        R                  5      (       a(  UR                  S   U :H  R                  5       (       d   eg UR                  S   U :X  d   eg )Nr   r   r  r  r2  r  r   )r	   r4   r*   r   r5   r   r3   r   rP   rZ   r  )r  r8   r;   s      r<   -test_scalar_setitem_with_nested_value_length1r  H  s     
C#	$BFF6NyRVVRVV/DEFH"' 
-v)NO	PBFF6N%$$v%',,....vvf~&&&rs   c                    [        / SQ5      n[        U [        R                  5      (       a.  X" U5      S'   [        / SQ5      n[        R
                  " X#5        O,[        R                  " [        SS9   X" U5      S'   S S S 5        [        / SQ[        S9nX" U5      S'   [        U [        R                  5      (       a(  UR                  S   U :H  R                  5       (       d   eg UR                  S   U :X  d   eg ! , (       d  f       N= f)Nrk  r   )r  r   r   r  r    r  r   )r   rP   r*   rZ   r5   r6   r0   r1   r2   r   r4   r  )r  r_   rp   r;   s       r<   4test_scalar_setitem_series_with_nested_value_length1r  ]  s     
!C%$$#C/*
s-]]H
 #(KQ
 f
-CKQ%$$
e#((****wwqzU"""
 
s   4C>>
Dc                     [        SSS/S9n [        5       U S'   [        5       U R                  S'   [        R                  " U R                  S   [        5       5        [        R                  " U R                  S   [        5       5        [        SSS/S9n[        5       UR                  S'   [        R                  " UR                  S   [        5       5        g )NOr}   r   )r   r   r   )r   r4   r5   r6   rN   )s1rV  s     r<   +test_object_dtype_series_set_series_elementr  x  s    	c#s	,BhBsG(BFF3K266#;1266#;1	c#s	,BBGGAJ2771:vx0rs   )0r^  r3   r   r   r  numpyr*   r0   pandas._configr   pandas.errorsr   pandas.core.dtypes.commonr   r   r   pandasrQ   r	   r
   r   r   r   r   r   pandas._testing_testingr5   pandas.tests.indexing.commonr   !pandas.tests.indexing.test_floatsr   r   rf  r  r  r  r  r_  ra  r   r  r  r  r  r  r  r  r   rs   r<   <module>r)     s   "   	    5 '      / 5q	- q	-h^# ^#B@4 @4FP7 P7f-"-: bTlFfYd5kBFKL"	"% fq!fbhh1v.C!Q0HI. fq!fbhh1v.C!Q0HI## fseRXXse_ekk#u.EF''$ fseRXXse_ekk#u.EF##01rs   