
    m*i                        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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#\RH                  RK                  S\" S/5      S4\" / SQ\&" S5      S9S4/5      S 5       r'S r(\RH                  RK                  SSSS/\R                  " SS/5      \R                  " SSS/5      /5      S 5       r)\RH                  RK                  SSSS/\R                  " SS/5      \R                  " SSS/5      /5      S  5       r*\RH                  RK                  SS!S"/\R                  " S"/5      \R                  " S#S"/5      /5      S$ 5       r+\RH                  RK                  SS!S"/\R                  " S"/5      \R                  " S#S"/5      /5      S% 5       r,S& r-g)'ztest fancy indexing & misc     N)datetime)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S  rS! r S" r!S# r"S$ r#S% r$S& r%\R                  R                  S'\&RN                  \&RP                  /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        c/var/www/html/land-doc-ocr/venv/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.   r/   r0   r1   )r6   r7   r8   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   r4   setitemilocappendr3   r   
isinstancepdIntervalIndexTimedeltaIndexDatetimeIndexPeriodIndexlen
MultiIndextyper	   _valuesndarrayjoin
IndexErrorr1   NotImplementedErrorr/   r0   )r6   r   frame_or_seriesindexer_sliusing_array_managerobjidxrnd3msgsr8   potential_errorss              r;   test_getitem_ndarray_3d!TestFancy.test_getitem_ndarray_3dJ   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#   rB   rC   zCannot set values with ndim > rH   )rE   zCannot set values with ndim > 1rF   rG   z,Array conditional must be same shape as selfr   r   )r   r)   rI   rJ   rK   r4   rM   r1   ndimrZ   r/   r0   )	r6   r   r]   r^   r`   ra   rb   errr8   s	            r;   test_setitem_ndarray_3d!TestFancy.test_setitem_ndarray_3dx   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)   r2   r   r   r4   r5   )r6   keyr7   r9   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   r3   r)   infr   r	   r-   r4   assert_index_equalr6   r7   r9   r:   s       r;   test_inf_upcastTestFancy.test_inf_upcast   s     s#q	q	rvv vvbffai A%%%!Qrzz:
f/rr   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-   r4   assert_produces_warningFutureWarningr3   r   objectassert_frame_equalr6   r7   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   ru   r   r   r|   r   r   r%   r&   r#   r   r   )r   r)   r*   reshapelistcopyr4   r   r   r3   r   r   )r6   r   r7   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   r4   r   r   r3   r   r   )r6   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~   rt   r~   r|   )r~   r|   r|   )r   r)   eyeru   r	   r4   rw   rx   s       r;   test_dups_fancy_indexing"TestFancy.test_dups_fancy_indexing   sC     rvvay/:S#J'')
f/rr   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   aaaaaaart   r&   )r   r   ru   rM   r4   r   r6   r7   r9   s      r;   &test_dups_fancy_indexing_across_dtypes0TestFancy.test_dups_fancy_indexing_across_dtypes   s^    ;<d9oV?@Ai 	1AqD
b)rr   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   r3   r4   r   r	   r/   r0   KeyError)r6   r7   rowsr:   r9   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   z8"None of [Index(['E'], dtype='str')] are in the [index]"r   r   z;"None of [Index(['E'], dtype='object')] are in the [index]")r   r)   rI   rJ   standard_normalr   r/   r0   r   reescaper3   )r6   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P #  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   r/   r0   r   r3   )r6   r   r7   s      r;   &test_dups_fancy_indexing_missing_label0TestFancy.test_dups_fancy_indexing_missing_label4  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   r6   r7   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   rt   r   r   )r   r   r   )	r   r)   rI   rJ   r   r/   r0   r   r3   r   s     r;   test_dups_fancy_indexing2#TestFancy.test_dups_fancy_indexing2A  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)   rI   rJ   r   rM   r3   r4   r   )r6   r7   r:   r9   s       r;   test_dups_fancy_indexing3#TestFancy.test_dups_fancy_indexing3L  s    II!!!$44V<-#J
 771Q<A
f/C:&
f/771Q36?Ac
*+
f/rr   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   ranger4   r5   )r6   
indexer_slro   r:   r9   s        r;   test_duplicate_int_indexing%TestFancy.test_duplicate_int_indexing`  sD    U1XY/%(1a&1C!%
v0rr   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>n  s    AJS-EA-Err   r   r   c                     U S:X  a  S$ U $ )Nr   -----r   r   s    r;   r   r   r  s    1:0L10Lrr   )r   r"   r   )r   applyr3   rM   )r6   r7   idxtemps       r;   test_indexing_mixed_frame_bug'TestFancy.test_indexing_mixed_frame_bugg  s    %03337OP

 W]]#EF6
 jCvvc3h%%&LM"sF{wwt}'''rr   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|   )            rt   z'[26, -8] not in index'r   )r      ir   r   )
r   r)   rI   rJ   r/   r0   r   r   r   shaper   s     r;    test_multitype_list_index_access*TestFancy.test_multitype_list_index_accessv  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  rt   )	r   r)   r   reset_index	set_indexreindexru   r4   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)rr   c                 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   rM   r   r  isnar   r3   r4   r   values)r6   r7   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,rr   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   r.  r   )r   r   r3   r4   r   )r6   r7   r:   r(  cols        r;   "test_multi_assign_broadcasting_rhs,TestFancy.test_multi_assign_broadcasting_rhs  s    $&&$	
 779}!C&(gHLL#  ,.c7<<>r#w!|c3Z'(
b+rr   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   rM   r4   r   r   s      r;   test_setitem_listTestFancy.test_setitem_list  sf     aVaS1!A!Q!5FD
f)rr   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r8  r   )r   r	   rP   	Timestampr   r   _is_all_datesr/   r0   r   r3   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)Nr7  r   r8  z^0$r9  )r   r   r;  r/   r0   r   r3   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                    [        / SQ/[        S5      S9nU[        S5         R                  [        5      U[        S5      '   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[        S5         R                  [        5      U[        S5      '   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[        S5         R                  [        5      U[        S5      '   [        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[        S5         R                  [        5      U[        S5      '   [        R                  " X45        g )N)123.4r#         @r   ABCDEFGrt   ABCDGr   r"   )r   r"   rD  rE  r#   rF  r   CDGr   r   )r   rC  rD  rE  r#   rF  r   r   )rB  r"   r%   rE  r#   rF  r   )r   r   r$  r   r   rM   r)   r+   r4   r   r3   )r6   r   df_origr7   r:   s        r;   test_astype_assignment TestFancy.test_astype_assignment4  s   12DO
 ")g!7!>!>v!FW\\^ ''!QqS&/00:1Q3-.Y
 !)e 5 < <V De ,,V4 ,,V4
b+ \\^3..rxx8q#v/0$y/
 #+4="9"@"@"Hg
b+\\^ "q3*} 5 < <RXX Fq3*}-.Y
 #+4="9"@"@"Hg
b+rr   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+   rM   r4   r   r3   r   s      r;   (test_astype_assignment_full_replacements2TestFancy.test_astype_assignment_full_replacementsZ  s    123 3rxx01c#789
b+123C1q#v
b+rr   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	   tolistr4   rw   r   r)   r*   )r6   rP  ss2exps        r;   test_index_type_coercion"TestFancy.test_index_type_coercionh  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,,,, :rr   r   N)+__name__
__module____qualname____firstlineno____doc__r<   r?   r/   markfilterwarningsre   rj   rp   ry   r   parametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r+  r1  r4  r<  r?  rK  rN  r4   getitemr3   rW  __static_attributes__r   rr   r;   r   r   %   s[   ,1,
+ [[	(	(T [[		00$, [[UT6N3- 4-.+*0
*2 , [[Vie%=> ?$	'0(1(.L*\.-`,&
*
	$,L, [[YRVV(<=)- >)-rr   r   c            
       :   \ rS rSrS rS rS rS 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)   rI   rJ   r4   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+rr   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   r3   r   r4   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+rr   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}   ri  r   )r   r   r3   r   r4   r   r6   r7   r'  s      r;    test_loc_setitem_fullindex_views)TestMisc.test_loc_setitem_fullindex_views  sK    YY7Oggi66"((+rxx
b&rr   c                    S n[         R                  " S5      R                  SS5      n/ SQn[        X4[	        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" XVX5        XVX4 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'   U(       aN  [        R                  " [        SS9   [        R                  " [        SS9   U" XVX5        S S S 5        S S S 5        g [        R                  " [        SS9   U" XVX5        S S S 5        g ! , (       d  f       NC= f! , (       d  f       g = f! , (       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   r3   r4   r   rM   )r7   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*+!!$3rr   r   r#   r&   )jimrs  rt  jolineabcder+   )ru   r   r   r%   r   r"   )r      r   r$   r	  rs  )r"      r   r#   r
  rt  r   r-   c                     SU  3$ )N@r   r   s    r;   r   -TestMisc.test_rhs_alignment.<locals>.<lambda>  s
    AaS'rr   )r!   z@-28z@-20z@-12g      1@)z@2g      :g      2g      $z@18zInvalid valuer   zincompatible dtype)r)   r*   r   r   r   rM   r   r$  mapr/   r0   	TypeErrorr4   r   r   )
r6   r   r  xsr)  r7   rv  rx  rw  frames
             r;   test_rhs_alignmentTestMisc.test_rhs_alignment  s   	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y@//!)= by= A@ ++MAUV"99 WV  A@ WVs0   5F'
FF'
F8
F$	 F''
F58
G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   )rP   
IndexSlicer	   r   r)   r*   r4   !assert_indexing_slices_equivalent)r6   r   SLCro   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Vrr   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*   rU   r/   r0   r1   )r6   r   r   r]   r`   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#   zrm   )r   r"   r$   )r"   r"   r   )r   yr   r   c   r#   )r   r   g       @)r   r"   r   )r   r   g     X@)r	   r   r   r3   r4   r   )r6   r   r7   r:   rv  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+rr   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   rM   r3   r   r)   r+   r4   r5   rm  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?rr   rD   )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-   r3   r4   r5   )r6   rD   rT  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Krr   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   notnullr2   r   r4   r   )r6   r7   index_dfr9   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/rr   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}   )r3   rM   atiat)r   getattrweakrefref)r6   r7   r(   wrs       r;   test_no_reference_cycle TestMisc.test_no_reference_cycle(  sG    aV1a&120DB 1[[_t||rr   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_countsr3   )r6   nulls_fixturer7   vcresult1result2r:   s          r;   test_label_indexing_on_nan#TestMisc.test_label_indexing_on_nan0  sX    QM23__E_*&&'#""""""rr   r   N)rY  rZ  r[  r\  rf  rj  rn  r  r/   r^  r`  r   r)   r*   linspacer  r  r  r  r  r  r  r  rb  r   rr   r;   rd  rd    s    
,,'4:l [[sB2!4bkk#sB6OP
W
W!
,@ [[V%9:L ;L0	#rr   rd  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%   ri  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   r3   r4   r   r6   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Brr   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   r4   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Brr   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Brr   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   ri  r  r   r"   r%   r   )r|   r~   r   r   r   r   r   )Nr~   r   )r   r   rM   r)   r   r
   r4   r   )r6   r  rV  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3rr   r   )rY  rZ  r[  r\  r)   r   r   r   r
   EXPECTED_SINGLE_ROW_RESULTSr/   r^  r`  r  r  r  r  rb  r   rr   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4rr   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   rX   _validate_setitem_valuer   _ndarray)r6   tz_naive_fixturer^   r  dtiro   r&  newvals           r;   test_setitem_dt64_string_scalar8TestDatetimelikeCoercion.test_setitem_dt64_string_scalar  s    qR8SXX4X()&&v.$C:99		)));;''6??:::;;&(((rr   boxrn   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)rO   ru  r4   r3   r   r   r   rX   r  r   r  )
r6   r  r^   rn   r  r  r  ro   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??:::;;&(((rr   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   rX   r  r  )r6   r^   r  tdiro   r&  s         r;   test_setitem_td64_scalar1TestDatetimelikeCoercion.test_setitem_td64_scalar  sg     gq1SXX4X()&&v.$C{{##v666rr   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)
rO   ru  r4   r3   r   r   r   rX   r  r  )r6   r^   rn   r  r  ro   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6rr   r   N)rY  rZ  r[  r\  r  r/   r^  r`  r   r)   r2   rP   Categoricalr	   ru  r  r   Hourr  r  rb  r   rr   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7rr   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&   r/  r|   r~   r   r   r   r'   r   )r   rP   r2   r   r3   r4   r5   rM   )r7   r:   r9   s      r;   "test_extension_array_cross_sectionr    s    	1a&01a&0	
 Cj
B bhh1vW5c3ZcRHVVC[F6,WWQZF6,rr   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+   r/  r|   r~   r   r   r   )r   r   r(   r   )
r   rP   r2   r)   r3   r   r4   r5   rM   r   )r7   r9   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,rr   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   r/   r0   r4   rM   r[   )ro   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	   r4   r5   )r  ro   resrV  s       r;   (test_ser_list_indexer_exceeds_dimensionsr    sG     ",C
S/1a&
!C
"b1v
/C3$rr   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   rH   z"Must have equal len keys and value(setting an array element with a sequencer   r   r   r   r|   r~   r   r/  )	r   rZ   r/   r0   r1   r3   r)   r2   r   )r  r7   r8   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   r/   r0   r1   r   rO   r)   rY   r3   all)r  r^   ro   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  r/  r  r   )r   r3   r)   r   r4   r   r2   r   rO   rY   r  )r  r7   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~&&&rr   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)Nri  r   )r  r   r   r  r   r  r   )r   rO   r)   rY   r4   r5   r/   r0   r1   r   r3   r  )r  r^   ro   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   r3   r4   r5   rM   )s1rU  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0rr   ).r]  r2   r   r   r  numpyr)   r/   pandas.errorsr   pandas.core.dtypes.commonr   r   r   pandasrP   r   r	   r
   r   r   r   r   pandas._testing_testingr4   pandas.tests.indexing.commonr   !pandas.tests.indexing.test_floatsr   r   rd  r  r  r  r  r^  r`  r   r
  r  r  r  r  r  r  r   rr   r;   <module>r'     s   "   	    '      / 5m	- m	-`d# d#N@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1rr   