+
    xȇi)                     ^   ^ RI H t  ^ RIt^ RIt^ RIHt ^ RIHt ^ RI	H
t
HtHtHt ^ RIt^ RIHtHtHtHtHtHtHtHtHtHtHtHtHtHtHtHt ^ RI H!t" ^ RI#H$t$ ^ RI%H&t&  ! R	 R
4      t' ! R R4      t( ! R R4      t) ! R R4      t* ! R R4      t+ ! R R4      t, ! R R4      t-R t.R t/R# )    )datetimeN)	_registry)is_object_dtype)CategoricalDtypeDatetimeTZDtypeIntervalDtypePeriodDtype)Categorical	DataFrameDatetimeIndexIndexIntervalIntervalIndex
MultiIndexNaTPeriodPeriodIndexSeries	Timestampcut
date_rangenotnaperiod_range)SparseArray)BDayc                     a ] tR t^(t oR t]P                  P                  R. RSO4      R 4       tR t	R t
R tR tR	 tR
 tR tR tR tR tR tR tR t]P                  P                  R]! R4      ]! R4      3]! ^ ^R7      ]! RR4      3]! RRR7      P7                  R4      ]! RRR7      3.4      R 4       t]P                  P                  R]P>                   U u. uF1  p \A        V PB                  \D        4      '       d   K%  V PB                  NK3  	  up RR.,           4      R  4       t#R! t$]P                  P                  R". RTO4      R$ 4       t%]P                  P                  R". RTO4      R% 4       t&R& t'R' t(R( t)R) t*]P                  P                  R. RUO4      R* 4       t+]P                  P                  R+R,R,..4      R- 4       t,R. t-R/ t.]P                  P                  R0. RVO. RWO.4      R2 4       t/R3 t0R4 t1R5 t2R6 t3R7 t4]P                  P                  R8. RXO. RYO^3. RZO. RYO^3. R[O. RYO^3. R\O. R]O^3. R^O. R_O^3. R`O. R_O^3.4      R: 4       t5R; t6]P                  P                  R<. RaO^]7! . RbO. RbO. RbO.. RaOR>7      3R9R#.^^.]7! . RcO. RdO. ReO.. RaOR>7      3. RfO]8Pr                  ! . RgO]8Pt                  R?7      ]7! . RgO. RgO. RgO.. RfOR>7      3. RhO. RgO. RiO. RjO.]7! . RkO. RlO. RmO.. RaOR>7      3. RnO]8Pr                  ! . RgO. RiO. RjO.]8Pt                  R?7      ]7! . RoO. RpO. RqO.. RaOR>7      3R=R9.]7! ^^.^	^
.^^..R=R9.R>7      ]7! . RrO. RsO. RtO.. RfOR>7      3.4      R@ 4       t;RA t<RB t=RC t>RD t?RE t@RF tARG tB]P                  P                  RH/ RIR1/.4      RJ 4       tCRK tDRL tERM tFRN tGRO tHRP tIRQtJVtKRR# u up i )uTestDataFrameSetItemc           	          ! R  R\         4      pR.p\        V4      p\        R^./VR7      p^VR&   ^WA! R4      &   \        R^.R^.V! R4      ^./VR7      p\        P                  ! WE4       R# )c                       ] tR t^+tRtRtR# )@TestDataFrameSetItem.test_setitem_str_subclass.<locals>.mystring N)__name__
__module____qualname____firstlineno__	__slots____static_attributes__r!       x/Users/max/.openclaw/workspace/postharvest/venv/lib/python3.14/site-packages/pandas/tests/frame/indexing/test_setitem.pymystringr    +   s    Ir(   r*   z2020-10-22 01:21:00+00:00aindexbcN)strr   r   tmassert_equal)selfr*   datar-   dfexpecteds   &     r)   test_setitem_str_subclass.TestDataFrameSetItem.test_setitem_str_subclass)   sz    	s 	 ,,d#aSz/38C=cA3aS(3-!EUS
%r(   dtypeint64c                    \         P                  P                  ^4      P                  ^^
\	        V4      4      p\         P
                  ! W1R7      W!&   W!,          P                  P                  V8X  g   Q hR# )   r9   N)nprandomdefault_rngintegerslenarrayr9   name)r3   r9   float_framearrs   &&& r)   test_setitem_dtype'TestDataFrameSetItem.test_setitem_dtype6   sZ    
 ii##A&//2s;7GHXXc7!'',,555r(   c                    \         P                  P                  ^4      P                  \	        V4      ^34      pW!RR.&   \
        P                  ! VRR.,          P                  V4       R# )r<   ABN)r>   r?   r@   standard_normalrB   r1   assert_almost_equalvalues)r3   rE   r4   s   && r)   test_setitem_list_not_dataframe4TestDataFrameSetItem.test_setitem_list_not_dataframe@   sV    yy$$Q'77[9I18MN"&S#J
{C:6==tDr(   c                   \        R . ROR. RO/\        . RORR7      R7      p\        . RO\        . RORR7      RR7      pR	p\        P                  ! \
        VR
7      ;_uu_ 4        W!R&   RRR4       \        \        P                  P                  ^4      P                  ^ ^R4      . ROR7      pRp\        P                  ! \
        VR
7      ;_uu_ 4        VP                  RR.4      P                  4       VR&   RRR4       Rp\        P                  ! \
        VR
7      ;_uu_ 4        \        4       VR&   RRR4       R#   + '       g   i     L; i  + '       g   i     La; i  + '       g   i     R# ; i)barbazr.   r/   foorD   r,   fizr-   rD   z/cannot reindex on an axis with duplicate labelsmatchnewcolNcolumnszDCannot set a DataFrame with multiple columns to the single column grgrz7Cannot set a DataFrame without columns to the column gr   r<      )defr+   r.   r/   )ghij)r+   r.   r/   r+   )   ri   )r+   r.   r/   ra   )r   r   r   pytestraises
ValueErrorr>   r?   r@   rA   groupbycount)r3   r5   sermsgs   &   r)   test_setitem_error_msmgs-TestDataFrameSetItem.test_setitem_error_msmgsE   s   Iuo6e4
  ,59

 @]]:S11xL 2 II!!!$--aF;(

 U]]:S11zz3*-335BtH 2 H]]:S11 {BtH 21 21 21
 211s$   &E%E)E'E	E$	'E8	c                ^   ^
p^p\        \        V4      R7      p\        P                  P	                  ^4      P                  V4      p\        V4       F  pWCV&   K	  	  \        \        P                  ! WB4      P                  W4      \        V4      R7      p\        P                  ! W64       R# )
   r,   N)
r   ranger>   r?   r@   rL   repeatreshaper1   assert_frame_equal)r3   NKr5   new_colrg   r6   s   &      r)   test_setitem_benchmark+TestDataFrameSetItem.test_setitem_benchmarkc   s    U1X&))''*::1=qAqE RYYw2::1@aQ
b+r(   c                z   \        \        P                  P                  ^4      P	                  R4      \        P
                  ! ^4      . ROR7      pVP                  ^ RVR,          4       VP                  ^RVR,          4       VR,          P                  R4      VR&   VP                  p\        \        P                  ! R4      .^,          \        P                  ! R4      .,           . ROR	7      p\        P                  ! W#4       VR,          P                  R4      VR&   VP                  p\        \        P                  ! R4      .^,          \        P                  ! R4      .^,          ,           . ROR	7      p\        P                  ! W#4       VR,          P                  R
4      VR&   VP                  p\        \        P                  ! R4      .^,          \        P                  ! R4      .^,          ,           \        P                  ! R
4      .,           . ROR	7      p\        P                  ! W#4       R# )r<   r/   r+   r-   r\   rT   rR   float32xfloat64r,   int32yN   r`   )r/   r.   r+   )rT   r/   rR   r.   r+   r   )rT   r/   rR   r.   r+   r   r   )r   r>   r?   r@   rL   arangeinsertastypedtypesr   r9   r1   assert_series_equalr3   r5   resultr6   s   &   r)   test_setitem_different_dtype1TestDataFrameSetItem.test_setitem_different_dtypen   s   II!!!$44V<))A,#

 			!UBsG$
		!UBsG$
 S'..+3XXi !A%))<(==4
 	v0 S'..+3XXi !A%))<(=(AA4
 	v0S'..)3XXi !A%))<(=(AARXXgEVDWW9
 	v0r(   c                    \        . ROR7      pVP                  VR&   . ROVR&   \        R. RO/. ROR.R7      p\        P                  ! W4       R# )rJ   r,   X)r4   r-   r\   NrJ   rK   Cr   r   z)r   r-   r1   rx   r3   r5   exps   &  r)   test_setitem_overwrite_index1TestDataFrameSetItem.test_setitem_overwrite_index   sL     _-((3!3'
 	b&r(   c                    \        4       p. ROVR&   \        R. RO/4      p\        P                  ! W4       \        \        . 4      R7      p. ROVR&   \        R. RO/4      p\        P                  ! W4       R# )r_   rT   r[   Nr^   )r   r1   rx   r   r3   r5   r6   s   &  r)   test_setitem_empty_columns/TestDataFrameSetItem.test_setitem_empty_columns   se    
 [5	eY/0
b+uRy)5	eY/0
b+r(   c                    \        R RRRR7      p\        \        P                  ! \	        V4      4      R7      pWR&   VR,          P
                  \        P
                  ! R4      8X  g   Q hR# )	z1/1/2000 00:00:00z1/1/2000 1:59:5010sns)frequnitr,   rJ   M8[ns]N)r   r   r>   r   rB   r9   )r3   rngr5   s   &  r)   %test_setitem_dt64_index_empty_columns:TestDataFrameSetItem.test_setitem_dt64_index_empty_columns   sR    ,.@uSWXRYYs3x013#w}} 2222r(   c                    \        \        ^4      R7      p\        RRR7      VR&   \        \        RRR7      ..^,          \        ^4      R.R7      p\        P                  ! W4       R# )r`   r,   20130101UTCtznowr   N)r   ru   r   r1   rx   r   s   &  r)   $test_setitem_timestamp_empty_columns9TestDataFrameSetItem.test_setitem_timestamp_empty_columns   sY    U1X&jU35	
u-./!358eW
 	b+r(   c                *   \         P                  ! . R	O. R
O4      p\        \        ^
4      R.R7      pR\	        V4       R\	        V4       R2p\
        P                  ! \        VR7      ;_uu_ 4        WR&   RRR4       R#   + '       g   i     R# ; i)r   rR   r[   zLength of values \(z$\) does not match length of index \(z\)rX   rT   N)r   r_   r_   r   r_   r<   rd   )r
   
from_codesr   ru   rB   rj   rk   rl   )r3   catr5   rp   s   &   r)   2test_setitem_wrong_length_categorical_dtype_raisesGTestDataFrameSetItem.test_setitem_wrong_length_categorical_dtype_raises   sw    $$%7IuRy5'2 #3s8* -114R	= 	 ]]:S11uI 2111s   2BB	c                    \        R . ROR. RO/4      p\        . RO4      pW!R&   \        VRR7      p\        P                  ! VR,          V4       R# )c_1n_1
new_columnrU   Nrd         ?       @      @r   r   r_   )r   r   r   r1   r   )r3   r5   sp_arrayr6   s   &   r)   test_setitem_with_sparse_value3TestDataFrameSetItem.test_setitem_with_sparse_value   sH    GHy)#<(6
r,/:r(   c                    \        R . ROR. RO/4      p\        \        . RO4      . R	OR7      pW!R&   \        \        . R
O4      RR7      p\        P                  ! VR,          V4       R# )r   r   r,   r   rU   Nrd   r   r   )r<   r_   r   )r_   r   r   )r   r   r   r1   r   )r3   r5   	sp_seriesr6   s   &   r)   (test_setitem_with_unaligned_sparse_value=TestDataFrameSetItem.test_setitem_with_unaligned_sparse_value   sT    GH;y1C	$<+i0|D
r,/:r(   c                    \        R R4      .p\        . 4      pWR&   \        RV/R.R7      p\        P                  ! W#4       R# )z2003-12Dr+   r[   N)r   r   r1   rx   )r3   r4   r   r6   s   &   r)   #test_setitem_period_preserves_dtype8TestDataFrameSetItem.test_setitem_period_preserves_dtype   sA    y#&'2sc4[3%8
f/r(   c                8   \        R \        . RORR7      R\        . R	O\        R7      R\        . R	O\        R7      R\        . R	ORR7      /4      p\        R \        . RR7      R\        . \        R7      R\        . \        R7      R\        . RR7      /4      p\        . R	O4       F\  w  r4R \	        V4      R\        V4      R\        V4      R\
        P                  ! V4      /VP                  VP                  ^ ,          &   K^  	  \        P                  ! W!4       R# )
r+   r:   r=   r.   r/   ra   uint32N)r   r_   r<   r^   )r   r   float	enumerateintr>   r   locshaper1   rx   )r3   r6   r5   idxr.   s   &    r)   "test_setitem_dict_preserves_dtypes7TestDataFrameSetItem.test_setitem_dict_preserves_dtypes   s    VIW5VIU3VIU3VIX6	
 VBg.VBe,VBe,VBh/	
  	*FCSXU1XU1XRYYq\	#BFF288A; + 	b+r(   z	obj,dtypez2020-01M)leftrightr   z
2011-01-01z
US/Easternr   s)r   r   c           
         \        R . ROR\        V.^,          VR7      /4      p\        R . RO/4      pWR&   \        P                  ! WC4       R# )r   objr=   Nr^   r   r   r1   rx   )r3   r   r9   r6   r5   s   &&&  r)   test_setitem_extension_types1TestDataFrameSetItem.test_setitem_extension_types   sH     eYvseaiu7UVWy)*5	
b+r(   ea_namezdatetime64[ns, UTC]	period[D]c                v    \        ^ .4      p^.W!&   \        ^ ^ .V^./4      p\        P                  ! W#4       R# )r   Nr   r1   rx   )r3   r   r   r6   s   &&  r)   test_setitem_with_ea_name.TestDataFrameSetItem.test_setitem_with_ea_name  s=     A3#a!gs34
f/r(   c                   \         P                  ! ^R.RR7      p\        V4      P                  4       pWR&   \	        ^ ^R.R^R./RR7      p\
        P                  ! W#4       \         P                  ! ^R.RR7      pWBR&   \
        P                  ! V^ ,          V^ ,          4       \
        P                  ! VR,          P                  4       V4       R# )r_   natdatetime64[ns]r=   newNdatetime64[s])
r>   rC   r   to_framer   r1   rx   r   assert_numpy_array_equalto_numpy)r3   data_nsr   r6   data_ss   &    r)   6test_setitem_dt64_ndarray_with_NaT_and_diff_time_unitsKTestDataFrameSetItem.test_setitem_dt64_ndarray_with_NaT_and_diff_time_units%  s    ((Au:-=>))+ua!TEAt9=EUV
f/ 1e*O<u
vay(1+6
##F5M$:$:$<fEr(   r   r   c                   ^dp\         P                  ! RV R24      p\         P                  ! V\         P                  R7      P	                  V4      pVR9   d   TpMVP                  R4      p\        R\         P                  ! V4      /\         P                  ! V4      R7      pWFV&   Wa,          P                  VP                  8X  g   Q hWa,          P                  V8H  P                  4       '       g   Q hR# )	d   M8[]r=   r   intsr,   N)r   ms	r>   r9   r   r:   viewr   r   rN   allr3   r   nr9   valsex_valsr5   s   &&     r)   -test_frame_setitem_datetime64_col_other_unitsBTestDataFrameSetItem.test_frame_setitem_datetime64_col_other_units3  s     3tfA'yy"((+007;G kk/2G		!-RYYq\B4x~~...7*//1111r(   c                   ^dp\         P                  ! RV R24      p\         P                  ! V\         P                  R7      P	                  V4      pVP                  R4      p\        R\         P                  ! V4      /\         P                  ! V4      R7      p\         P                  ! V\         P                  R7      P	                  R4      VR&   WFR&   VR,          P                  V8H  P                  4       '       g   Q hR	# )
r   r   r   r=   r   r   r,   r   datesNr   r   s   &&     r)   6test_frame_setitem_existing_datetime64_col_other_unitsKTestDataFrameSetItem.test_frame_setitem_existing_datetime64_col_other_unitsH  s     3tfA'yy"((+007++./		!-RYYq\Bii277A7 77""g-224444r(   c                   TpVR ,          P                  R4      pW2R&   \        P                  ! VR,          \        VRR7      4       RVR&   W2R&   \        P                  ! VR,          \        VRR7      4       VR VP                  P
                  ^,          P                  pVP                  P
                  ^,          P                  p\        P                  ! WE4       VP                  P                  pVP                  P                  p\        V4      \        V4      8X  g   Q hVP                  4       p\        VP                  R&   \        VP                  R&   VR ,          p	\        P                  ! \        V	4      \        . ROR R7      4       \        P                  ! VP                  VP                  4       R# )	rK   rT   r   rU   r   N)r_   r_   r_   r<   )TFT)renamer1   r   r   _mgrblocksrN   assert_extension_array_equal_ndarraybaseidcopyr   ilocr   r   )
r3   timezone_framer5   r   v1v2v1basev2basedf2r   s
   &&        r)   test_setitem_dt64tz(TestDataFrameSetItem.test_setitem_dt64tzY  sG   gnnU# 3
r#ws(=>33
r#ws(=>sG
 WW^^A%%WW^^A%%
''/!!!!&zRZ''' ggiS
uV}f5Hs.ST
szz2995r(   c                    \        R ^RR7      p\        \        P                  P	                  ^4      P                  R4      VR7      pWR&   \        VR,          4      p\        P                  ! W1RR7       VP                  R8X  g   Q hVP                  R8X  g   Q hVP                  4       P                  R4      p\        VP                  \        4      '       g   Q h\        P                  ! VP                  V4       R# )	z1/1/2000r-   )periodsrD   r,   r   Fcheck_namesNr   )r   r   r>   r?   r@   rL   r   r1   assert_index_equalrD   reset_index	set_index
isinstancer-   r   )r3   r   r5   rss   &   r)   test_setitem_periodindex-TestDataFrameSetItem.test_setitem_periodindexx  s    :qw?ryy,,Q/??GsS72g;
b59ww'!!!xx7"""^^''0"((K0000
bhh,r(   c                   \        R . ROR. RO/4      p\        P                  ! ^^.^^.^^..4      pW!RR.&   \        R . ROR. ROR. ROR. RO/4      pVR,          P                  VP                  4      VR&   VR,          P                  VP                  4      VR&   VR,          P                  VP                  8X  g   Q hVR,          P                  VP                  8X  g   Q h\
        P                  ! W4       R# )	r+   r.   r/   ra   N)onetwothreer^   r_   r`   r   r_   r_   r_   )r   r>   rC   r   r9   r1   rx   r3   r5   rF   r6   s   &   r)   'test_setitem_complete_column_with_array<TestDataFrameSetItem.test_setitem_complete_column_with_array  s    4c9EFhhAAA/0C:,YYY	
 !,,SYY7 ,,SYY7}""cii///}""cii///
b+r(   c                    \        R ^	RRR7      p\        V4      p\        R. RO/RR7      pVP                  VP                  R&   \        P
                  ! W#4       R# )	z
2016-01-01r   rJ   )r  r   rD   r   r=   NNNN)	r   r   r   r   r   r   r   r   r   )r   r   	_na_valuer
  r1   rx   )r3   r   r   r6   s   &   r)   test_setitem_period_d_dtype0TestDataFrameSetItem.test_setitem_period_d_dtype  sR    <3G3QR
 A
f/r(   c                @   \        . ROVR7      p\        \        P                  P	                  ^4      P                  R4      VR7      p. ROVR&   \        . R	O\        R7      pVR8X  d   \        . R
O\        R7      p\        P                  ! VP                  V4       R# )r_   r=   r[   Ff8Nr^   )r`   r`   rd   )r_   r<   r`   F)r   r   r   F)
r   r   r>   r?   r@   rL   objectr1   r  r\   )r3   r9   colsr5   expected_colss   &&   r)   $test_setitem_bool_with_numeric_index9TestDataFrameSetItem.test_setitem_bool_with_numeric_index  su     Ye,ryy,,Q/??GQUV#5	.f=D=!"8GM
bjj-8r(   indexerrK   c                    \        R R.R7      p\        R^^./4      pW2V&   \        R \        P                  .^,          R^^./4      pVR ,          P                  R4      VR &   \        P
                  ! W$4       R# )rJ   rK   r[   r.  N)r   r>   nanr   r1   rx   )r3   r3  r5   otherr6   s   &&   r)   #test_setitem_frame_length_0_str_key8TestDataFrameSetItem.test_setitem_frame_length_0_str_key  sr     Sz*3A-(7cBFF8a<q!f=> ,,X6
b+r(   c           
        . RO^,          p\        \        ^4      VR7      pRVP                  R&   R	VP                  R
&   RVR&   \        . RO\        P                  ^^\        P                  ^^.\        P                  ^^\        P                  ^^..RR7      pV^,          P                  \        P                  4      V^&   V^,          P                  \        P                  4      V^&   Wn        \        P                  ! W#4       R# )rJ   r   r   r.  r=   Nr   )r   r`   )r   rJ   )r_   ri   )r(  rK   r<   r   )r   r_   r<   r`   ri   r   )
r   ru   r   r>   r5  r   r:   r\   r1   rx   )r3   r/  r5   r6   s   &   r)   $test_setitem_frame_duplicate_columns9TestDataFrameSetItem.test_setitem_frame_duplicate_columns  s    "U1Xt4vv3"Arvvq!,Arvvq!,
 
 qk((2qk((2
b+r(   c                   . RO^,          p\        \        ^4      VR7      p\        P                  ! \        RR7      ;_uu_ 4        RVR .&   RRR4       VP
                  R,          p\        P                  ! \        RR7      ;_uu_ 4        RVR .&   RRR4       R#   + '       g   i     LX; i  + '       g   i     R# ; i)rJ   r   "Columns must be same length as keyrX   Nr   )r   r`   r   )r(  :Nr`   N)r   ru   rj   rk   rl   r
  )r3   r/  r5   r  s   &   r)   2test_setitem_frame_duplicate_columns_size_mismatchGTestDataFrameSetItem.test_setitem_frame_duplicate_columns_size_mismatch  s    "U1Xt4]]:-QRR!BuI S ggen]]:-QRR"CJ SR	 SR SRRs   BB0B-	0C	r/  r+   c                    \        . RO.VR7      p\        ^
^..RR.R7      pRp\        P                  ! \        VR7      ;_uu_ 4        W2R&   RRR4       R#   + '       g   i     R# ; i)	r_   r[   ra   rb   r>  rX   r+   Nr^   r   rj   rk   rl   )r3   r/  r5   rhsrp   s   &&   r)   #test_setitem_df_wrong_column_number8TestDataFrameSetItem.test_setitem_df_wrong_column_number  sT     	{D1"b
S#J72]]:S11sG 2111s   	AA)	c                   \        . RO.. ROR7      p\        . RO.. ROR7      pW!RR.&   \        . RO.. ROR7      p\        P                  ! W4       W!RR.&   \        . R	O.. R
OR7      p\        P                  ! W4       R# )r_   r+   r.   r[   r/   Nr^   r+   r.   r.   rt         )rt   rI  rJ  rt   )r+   r.   r.   r/   r   r3   r5   rC  r6   s   &   r)   /test_setitem_listlike_indexer_duplicate_columnsDTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns  so    	{O<@C:l^_E
b+C:./9MN
b+r(   c                    \        . RO.. ROR7      p\        ^
^..RR.R7      pRp\        P                  ! \        VR7      ;_uu_ 4        W!RR.&   RRR4       R#   + '       g   i     R# ; i)	r_   r+   r.   r[   r>  rX   Nr^   rG  rB  )r3   r5   rC  rp   s   &   r)   @test_setitem_listlike_indexer_duplicate_columns_not_equal_lengthUTestDataFrameSetItem.test_setitem_listlike_indexer_duplicate_columns_not_equal_length  sV    	{O<"b
S#J72]]:S11 SzN 2111s   AA-	c                   \        R \        ^
4      /4      p\        VR ,          ^4      p\        VP                  P
                  \        4      '       g   Q hW!R&   \        P                  ! V4      VR&   VP                  VR&   \        P                  ! VP                  4      VR&   VP                  \        4      VR&   \        VR,          P                  \        4      '       g   Q h\        VR,          P                  P
                  P                  \        4      '       g   Q h\        VR,          P                  \        4      '       g   Q h\        VR,          P                  P
                  P                  \        4      '       g   Q h\        VR,          P                  \        4      '       g   Q h\        VR,          P                  \        4      '       g   Q h\        VR,          4      '       g   Q hR p\         P"                  ! V! VP$                  4      V! VP$                  4      4       \         P"                  ! V! VP$                  4      V! VP&                  4      RR7       \         P"                  ! V! VP$                  4      V! VP(                  4      RR7       \         P"                  ! V! VP&                  4      V! VP(                  4      RR7       \         P*                  ! VR,          VR,          4       \         P*                  ! VR,          VR,          RR7       \         P*                  ! VR,          VR,          4       \         P*                  ! VR,          VR,          RR7       R	# )
rJ   rK   r   r   EFc                 @    \        \        P                  ! V 4      4      # N)r   r>   rC   r   s   &r)   <lambda>=TestDataFrameSetItem.test_setitem_intervals.<locals>.<lambda>  s    eBHHQK(r(   Fr  N)r   ru   r   r  r   
categoriesr   r>   rC   rN   r   r.  r9   r   r   r   r1   r  rK   r   r   r   )r3   r5   ro   r/   s   &   r)   test_setitem_intervals+TestDataFrameSetItem.test_setitem_intervals  sZ   U2Y'("S'1o#'',,m<<<<
 3((3-3**3((3::&3**V$3"S'--)9::::"S'++0066FFFF"S'--)9::::"S'++0066FFFF "S'--7777"S'--7777 r#w'''' )
agqw/
agqwEB
agqwEB
agqwEB 	r#w30
r#w3UC 	r#w30
r#w3UCr(   c                8   \        R \        \        R4      4      P                  R4      /4      pVP                  P
                  P                  RR.4      Vn        \        R \        RR.4      P                  RR.4      /4      p\        P                  ! W4       R# )rf   mncategoryr   mN)
r   r   listr   rf   r   reorder_categoriesr
   r1   rx   r   s   &  r)   test_setitem_categorical-TestDataFrameSetItem.test_setitem_categorical'  s}    VDJ/66zBCDttxx**C:6+sCj)<<c3ZHI
 	b+r(   c                    \        . R R7      p\        R.VR7      p. VR&   \        R.VR7      p\        P                  ! VP                  VP                  4       R# )r   rU   rJ   r\   r-   N)r   r   r1   r  r-   )r3   r-   r   r6   s   &   r)    test_setitem_with_empty_listlike5TestDataFrameSetItem.test_setitem_with_empty_listlike0  sK    bu%C56scU%8
fllHNN;r(   zcols, values, expectedr   c                ~    \        V.VR 7      pVR,          VR&   VR,          P                  ^ ,          pWS8X  g   Q hR# )r[   r+   N)r   rN   )r3   r/  rN   r6   r5   r   s   &&&&  r)   test_setitem_same_column-TestDataFrameSetItem.test_setitem_same_column8  s=     x.S'3C"!!!r(   c                   . R
ORR.. RO3p\         P                  ! V4      p\        R^R7      p\        P                  P                  ^4      P                  ^R\        V4      \        V4      34      p\        WBVR7      pVP                  P                  P                  4       VR,          R,          rv\        P                  P                  ^4      P                  V4       VR,          P                  VR	R	R13,          VR &   \        P                  ! VR ,          VR,          4       \        P                  P                  ^4      P                  V4       VR,          P                  Wg3,          VR&   \        P                  ! VR,          VR,          4       \        P                  P                  ^4      P                  V4       VR,          P                  Wg3,          VR&   \        P                  ! VR,          VR,          4       R	# )jimjoliefirstlast20141006)r  i  re  r(  N)rl  joerm  )r   centerr   )rm  ro  )rq  rn  )rm  rn  )rq  ro  )r   from_productr   r>   r?   r@   rA   rB   r   r-   rN   r	  shuffler   r1   rx   )r3   itr/  r-   r   r5   rg   rh   s   &       r)   test_setitem_multi_index-TestDataFrameSetItem.test_setitem_multi_indexJ  s    %w&79TT&&r*:r2yy$$Q'00D3u:s4y:QRt7xx##%r"vay1
		a ((+wKOOAttG,5	
biG5
		a ((+!"3488>
b!12B7H4IJ
		a ((+ !3488>?
b126H3IJr(   zcolumns,box,expectedrJ   r[   r=   c                j    \        ^^.^^.^^..RR.R7      pW$V&   \        P                  ! WC4       R# )r_   rJ   rK   r[   Nr   )r3   r\   boxr6   r5   s   &&&& r)   !test_setitem_list_missing_columns6TestDataFrameSetItem.test_setitem_list_missing_columnsb  s=    f AAA/#sD7
b+r(   c                    \        \        VR ,          VR,          4      4      pW!R&   VR,          p\        W!P                  RR7      p\        P
                  ! W44       R# )rJ   rK   tuplesrW   N)r`  zipr   r-   r1   r   )r3   rE   r~  r   r6   s   &&   r)   test_setitem_list_of_tuples0TestDataFrameSetItem.test_setitem_list_of_tuples  sO    c+c*K,<=> &HX&&(9(9I
v0r(   c                    \        R . ROR. RO/4      pR R 4       p^VP                  V&   \        R . ROR. RO/4      p\        P                  ! W4       R# )	r+   r.   c              3   $   "   T F  qx  K  	  R # 5irU  r!   .0r   s   & r)   	<genexpr>CTestDataFrameSetItem.test_setitem_iloc_generator.<locals>.<genexpr>       %f1f   Nr^   ri   r      r  r#  ri   r_   r_   r   r
  r1   rx   r3   r5   r3  r6   s   &   r)   test_setitem_iloc_generator0TestDataFrameSetItem.test_setitem_iloc_generator  sN    YY78%f%c9c9=>
b+r(   c                    \        R . ROR. RO/4      pR R 4       p^VP                  V^3&   \        R . ROR. RO/4      p\        P                  ! W4       R# )r+   r.   c              3   $   "   T F  qx  K  	  R # 5irU  r!   r  s   & r)   r  STestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator.<locals>.<genexpr>  r  r  Nr^   r  r  r  r  r  s   &   r)   +test_setitem_iloc_two_dimensional_generator@TestDataFrameSetItem.test_setitem_iloc_two_dimensional_generator  sR    YY78%f%
c9c9=>
b+r(   c                   \        R RR7      p\        V\        . RR7      R7      p\        RV\        P                  R7      VR&   \        RVR	R7      VR&   \        R
VR	R7      VR
&   \        RV\        P
                  R7      VR&   VP                  p\        \        P                  \        \        \        P
                  .\        R4      R7      p\        P                  ! W44       R# )r  S24)rD   r9   r0   r=   r   r+   )rD   r-   r9   r.   S64r/   ra   abcdr,   N)r   r   r   r>   r   uint8r   r.  r`  r1   r   )r3   r-   r5   r   r6   s   &    r)   (test_setitem_dtypes_bytes_type_to_object=TestDataFrameSetItem.test_setitem_dtypes_bytes_type_to_object  s    D.UE"E,BCcbii@3ce<3ce<3cbhh?3299ffbhh?tF|T
v0r(   c                X   \        R ^^.R^^./4      P                  R RRR/4      p\        RVP                  R7      pVR ,          VP                  VR 3&   VR,          VP                  VR3&   \        R ^^.R^^./4      P                  R RRR/4      p\
        P                  ! W4       R# )r+   r.   r:   Int64Fr,   N)r   r   r   r-   r   r1   rx   )r3   r   maskr6   s   &   r)    test_boolean_mask_nullable_int645TestDataFrameSetItem.test_boolean_mask_nullable_int64  s    C!Qq!f56=='3(
 e6<<0 &s

49 &s

49cAq63A78??'3(
 	f/r(   c                    \        R ^^./4      p\        ^^.RR7      VR &   \        R ^^./RR7      p\        P                  ! W4       R# )r+   r  r=   Nr   r   s   &  r)    test_setitem_ea_dtype_rhs_series5TestDataFrameSetItem.test_setitem_ea_dtype_rhs_series  sJ    aV}%!Qw/3cAq6]':
b+r(   c                
   \        R \        P                  ! ^
4      R\        P                  ! ^
4      /\        P                  ! ^
4      R7      p\        P                  ! R	4      p\        \        P                  ! ^
4      \	        . RR7      R7      pW#R &   \
        P                  ! \        RR7      ;_uu_ 4        \        P                  ! V4      VR&   RRR4       \
        P                  ! W14       R#   + '       g   i     L(; i)
znp-arrayz	np-matrixr,   r0   r=   r   z@matrix subclass is not the recommended way to represent matricesrX   N)rt   r_   )
r   r>   onesr   r   r1   assert_produces_warningPendingDeprecationWarningmatrixrx   )r3   r6   r+   r5   s   &   r)   test_setitem_npmatrix_2d-TestDataFrameSetItem.test_setitem_npmatrix_2d  s     k2772;?ryyQS}
 GGGRYYr]E"E4JK: ''%T
 
 !iilB{O	
 	b+
 
s   8C22D	r   ra   c                $   \        R ^^.R^^./VC4      p^^d^ ^/VP                  R&   ^ ^^^/VP                  R&   ^^/VP                  R&   \        R ^^d.R^^./VCR^^.R\        P                  ^./C4      p\        P
                  ! W#4       R# )r+   r.   r/   rb   Nr(  r+   )r(  r/   )r(  rb   )r   r   r>   r5  r1   rx   )r3   r   r5   r6   s   &&  r)   %test_setitem_aligning_dict_with_index:TestDataFrameSetItem.test_setitem_aligning_dict_with_index  s     aVS1a&9D9:S!S)vQ1vQv3*cAq6QTQ3Abffa[Q
 	b+r(   c                F   \        R ^^./4      p\        R ^
^./^^.R7      VR &   \        R \        P                  ^
./4      p\        P                  ! W4       \        R ^^./4      pVP                  ^ \        R ^
^./^^.R7      4       \        P                  ! W4       R# )r+   r,   N)r   r>   r5  r1   rx   isetitemr   s   &  r)   test_setitem_rhs_dataframe/TestDataFrameSetItem.test_setitem_rhs_dataframe  s    aV}%S2r(OAq6:3cBFFB<01
b+aV}%
Ay#BxA?@
b+r(   c                    \        R R.^^.^^..R7      p\        R ^
^./VR7      VR &   \        R \        ^
^.VR7      R^^./4      p\        P                  ! W#4       R# )r+   r.   r\   r4   r=   Nr   r3   any_numeric_ea_dtyper5   r6   s   &&  r)   *test_setitem_frame_overwrite_with_ea_dtype?TestDataFrameSetItem.test_setitem_frame_overwrite_with_ea_dtype  sq    Sz!Q!Q0@AS2r(O3GH3VRH,@AaV
 	b+r(   c           
     b   \         P                  ! R 4       \        R^^./4      p\        P                  ! RR4      ;_uu_ 4        \        RR.\        R7      VR&   RRR4       \        R^^.R\        RR.\        R7      /4      p\        P                  ! W4       R#   + '       g   i     LJ; i)pyarrowr+   zfuture.infer_stringTr.   r=   N)
rj   importorskipr   pdoption_contextr   r.  r   r1   rx   r   s   &  r)   'test_setitem_string_option_object_index<TestDataFrameSetItem.test_setitem_string_option_object_index  s    I&aV}%4d;;S#Jf5BsG <cAq63Sz0PQR
b+ <;s   BB.	c                    \        R^
./4      pVP                  4       pRpW.,          W&   \        P                  ! W4       R# )r+   Nr+   r.   r   r	  r1   rx   )r3   r5   r6   col_names   &   r)   test_setitem_frame_midx_columns4TestDataFrameSetItem.test_setitem_frame_midx_columns  s<    
RD)*779*~
b+r(   c                   \        R \        P                  ! ^
.RR7      /4      p\        ^.RR7      VP                  R&   \        R \        P                  ! ^.RR7      /4      p\
        P                  ! W4       \        R \        P                  ! ^
.RR7      /4      p\        ^.RR7      VP                  R&   \
        P                  ! W4       R# )r+   i8r=   r  Nr  )r(  r   )r   r>   rC   r   r   r1   rx   r
  r   s   &  r)   test_loc_setitem_ea_dtype.TestDataFrameSetItem.test_loc_setitem_ea_dtype  s    RXXrd$789G4vc288RD#=>?
b+RXXrd$789t73
b+r(   c                    \        \        R 4      .\        R7      p\        R^./4      pWP                  R&   WR&   \        R^.RVRV/4      p\
        P                  ! W#4       R# )z
2019-12-31r=   r+   r.   r/   N)r(  r.   )r   r   r.  r   r   r1   rx   )r3   r   r5   r6   s   &   r)   -test_setitem_index_object_dtype_not_inferringBTestDataFrameSetItem.test_setitem_index_object_dtype_not_inferring#  sh    Y|,-V<aSz"v3aSSS
 	b+r(   r!   N)r   r:   r   uint64r   r   )rf   r_  r   r   r   r   Y)r-  r  u8rd   )r+   r+   r+   )r   r   r   r+   r_   r<   r`   ri   )r   r   r   r+   )r   rK   rK   r+   )r   rK   r+   r^   )rK   r   r+   )r`   r<   r_   )r   r+   rK   )rJ   rK   r   r   )   r  r  r  )r_   r<   r     )r`   ri   r  r  r   r  r  r  r   )r  r  	   rK   r   r   rH  )         )r_   r  r  r  )r`   rt   rI  rJ  )r   r  r  r  )r   rJ   r   )r  r<   r  r  )rI  ri   rt   rJ  )r  r  r  r  )r  r<   r  )r  ri   rt   )rI  r  rJ  )Lr"   r#   r$   r%   r7   rj   markparametrizerG   rO   rq   r|   r   r   r   r   r   r   r   r   r   r   r   r	   r   r   r   as_unitr   r   ea_registryr   r  rD   propertyr   r   r   r   r  r  r%  r*  r1  r7  r;  r?  rD  rL  rO  rZ  rb  rf  ri  rw  r   r>   rC   r:   r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r'   __classdictcell__)r9   __classdict__s   0@r)   r   r   (   s    & [[M66E
#<	,#1J	',3,
;;0,6 [[IC 011A&gw(GH,<8@@ES\:	

,
, [[ %++	
+ejj(3 EJJ+	
 !+
.	/	0	0F [[V%IJ2 K2( [[V%IJ5 K5 6>-,&	0 [[W&899 :9 [[Yse5, 6,,.	# [[Vo%GH I
,!*DX,< [[ !<3!<3!<3i+i+i+	

"
"K0 [[ %!<>0 c
A!<>0  "((39i;_U  L,7!?OD0  )\<@Q!?OD0 c
Aq6Ar7RH5SzJ
K8/M-	
0b,c0b,1,,
10,,( [[Vb3*%56	, 7	,	,
,,,	,, ,[	
s   !"O
O
r   c                   z   a  ] tR tRt o ]P
                  R 4       t]P
                  R 4       tR tR t	R t
RtV tR# )	TestSetitemTZAwareValuesi4  c                F    \        R R.RR7      pVP                  R4      pV# )z2013-1-1 13:00z2013-1-2 14:00rK   rU   z
US/Pacific)r   tz_localize)r3   naiver   s   &  r)   r   TestSetitemTZAwareValues.idx5  s*    /1ABM-
r(   c                    \        \        P                  ! VP                  4       R R7      RR7      pVP                  VP                  8X  g   Q hV# )r.  r=   rK   rU   )r   r>   rC   tolistr9   )r3   r   r6   s   && r)   r6   !TestSetitemTZAwareValues.expected;  s:    "((3::<x@sK~~***r(   c                v   \        \        P                  P                  ^4      P	                  R4      R.R7      pWR&   VP                  ^ ^.R7      P                  P                  R4      VR&   VR,          p\        VP                  R4      P                  R4      RR7      p\        P                  ! WE4       R# )	r<   rJ   r[   rK   r,   Nr   rU   r<   r_   )r   r>   r?   r@   rL   	to_seriesdt
tz_convertr   r  r1   r   )r3   r   r6   r5   r   comps   &&&   r)   test_setitem_dt64series0TestSetitemTZAwareValues.test_setitem_dt64seriesA  s    ryy,,Q/??GRUQVW3--q!f--00;;DA3CcnnU+77=CH
v,r(   c                    \        \        P                  P                  ^4      P	                  R4      R.R7      pWR&   VR,          p\
        P                  ! WB4       R# )r<   rJ   r[   rK   Nr  )r   r>   r?   r@   rL   r1   r   r3   r   r6   r5   r   s   &&&  r)   test_setitem_datetimeindex3TestSetitemTZAwareValues.test_setitem_datetimeindexK  sL    ryy,,Q/??GRUQVW 3C
v0r(   c                   \        \        P                  P                  ^4      P	                  R4      R.R7      pVP                  4       VR&   VR,          pVP                  P                  R4      p\        P                  ! WB4       R# )r<   rJ   r[   rK   usNr  )
r   r>   r?   r@   rL   to_pydatetimer  r  r1   r   r  s   &&&  r)   .test_setitem_object_array_of_tzaware_datetimesGTestSetitemTZAwareValues.test_setitem_object_array_of_tzaware_datetimesT  sk    ryy,,Q/??GRUQVW ##%3C;;&&t,
v0r(   r!   N)r"   r#   r$   r%   rj   fixturer   r6   r  r  r  r'   r  r  s   @r)   r  r  4  sG     ^^ 
 ^^ 
-11 1r(   r  c                   b   a  ] tR tRt o R tR tR tR tR tR t	R t
R	 tR
 tR tR tRtV tR# )!TestDataFrameSetItemWithExpansioni_  c                    \        R . ROR. RO/4      pVR ,          p\        P                  ! RR.RR.RR..4      VRR.&   ^dVP                  R&   \	        . ROR R	7      p\
        P                  ! W#4       R
# )r+   r.   g?g?g333333?g?      ?r/   ra   rU   Nr^   )ri   ri   r  r   r   )r   r>   rC   r
  r   r1   r   )r3   r5   ro   r6   s   &   r)   test_setitem_listlike_views=TestDataFrameSetItemWithExpansion.test_setitem_listlike_views`  st    YY78 g C:SzC:"FGC: )#.
s-r(   c                    \        ^^.^^..4      p^^.VR&   \        . RO. RO.. ROR7      p\        P                  ! W4       R# )r_   0 - Namer[   N)r_   r<   r   )r`   ri   r  )r   r_   r  r   r   s   &  r)   .test_setitem_string_column_numpy_dtype_raisingPTestDataFrameSetItemWithExpansion.test_setitem_string_column_numpy_dtype_raisingp  sF    AA'(Q:i3=OP
b+r(   c                0   \        . RORR7      p\        \        ^4      4      VP                  R&   \        ^ \        P
                  \        P
                  .^\        P
                  \        P
                  ... ROR7      p\        P                  ! W4       R# )r+   r   r\   r9   r[   NrG  r  )r   r`  ru   r   r>   r5  r1   rx   r   s   &  r)   'test_setitem_empty_df_duplicate_columnsITestDataFrameSetItemWithExpansion.test_setitem_empty_df_duplicate_columnsw  se    i@eAhv 1bffbff"56
 	b+r(   c           
        \        R \        P                  ! \        P                  P	                  ^4      P                  ^ R^d4      RR7      /4      p\        \        ^ RR4       Uu. uF  q" RVR,            2NK  	  up4      pVP                  R .RR7      p\        VP                  \        ^ R	R4      R
VR7      pVP                  pWQR&   VP                  p\        \        P                  ! R4      \        VR
R7      .R R.R7      p\         P"                  ! Wg4       WAR&   VP                  p\        \        P                  ! R4      \        VR
R7      \        VR
R7      .. ROR7      p\         P"                  ! Wg4       VR,          pVR,          p	\         P$                  ! VP&                  P                  V4       RVn        \         P"                  ! V	P+                  4       VP+                  4       4       R# u upi )valuei'  r   r=   i  z - i  T)by	ascendingi)  F)r   labelsr   )rY  orderedr,   rR  N)r  r   rR  )r   r>   rC   r?   r@   rA   r
   ru   sort_valuesr   r  rN   r   r   r9   r   r1   r   assert_categorical_equalr  rD   
sort_index)
r3   r5   rg   r  ro   r   r   r6   result1result2s
   &         r)   -test_setitem_with_expansion_categorical_dtypeOTestDataFrameSetItemWithExpansion.test_setitem_with_expansion_categorical_dtype  s   II))!,55aDG
 %5#:NO:NQ3q3wi0:NOP^^yD^9"((E!UC0fMjj 3XXg 0FE RSC.
 	v0 3! FEB FEB
 &
 	v0S'S'
##GLL$6$6< 
w113S^^5EFE Ps   &G8c                    \        4       p^VR&   \        R.R7      P                  \        P                  4      p\        P
                  ! W4       R# )r_   rT   r[   N)r   r   r>   r:   r1   rx   r   s   &  r)   test_setitem_scalars_no_index?TestDataFrameSetItemWithExpansion.test_setitem_scalars_no_index  s;    [5	eW-44RXX>
b+r(   c                    RVP                   9  g   Q hVR ,          VR&   RVP                   9   g   Q hVR,          pVR ,          p\        P                  ! W#RR7       R# )rJ   Fr  N)rJ   rK   )r\   r1   r   )r3   rE   r   r6   s   &&  r)   test_setitem_newcol_tuple_key?TestDataFrameSetItemWithExpansion.test_setitem_newcol_tuple_key  sj    
 $$% 	% % !,C 0H[00000X&s#
vUCr(   c                   \        R R\        4       R7      p\        V\        ^
4      R7      p\	        R^^4      p\        V4      p\        P                  W$&   \        P                  ! W$,          4      P                  4       '       g   Q hR# )z1/1/2012z2/1/2012)startendr   re  i  N)
r   r   r   ru   r   r   r>   r5  isnanr   )r3   r\   r4   ttss   &    r)   #test_frame_setitem_newcol_timestampETestDataFrameSetItemWithExpansion.test_frame_setitem_newcol_timestamp  sd    ::DFKb	:T2q!q\66xx!%%''''r(   c                    \        R R R./4      pVP                  VR&   ^dVP                  RR.R3&   VP                  ^.R3,          p\        R R.R^d./^.R7      p\        P                  ! W#4       R# )r+   r.   FTr(  r,   N)r   r-   r   r1   rx   r   s   &   r)   *test_frame_setitem_rangeindex_into_new_colLTestDataFrameSetItemWithExpansion.test_frame_setitem_rangeindex_into_new_col  su    c3Z()((3%(t}c!"QcC5#u5aSA
f/r(   c                    \        R R.^^.^^..R7      p\        R ^
^./VR7      VR&   \        R ^^.R^^.R\        ^
^.VR7      /4      p\        P                  ! W#4       R# )r+   r.   r  r=   r/   Nr   r  s   &&  r)    test_setitem_frame_keep_ea_dtypeBTestDataFrameSetItemWithExpansion.test_setitem_frame_keep_ea_dtype  sz    Sz!Q!Q0@AS2r(O3GH3aVaVVRH,@A
 	b+r(   c                   \        \        R 4      R7      pR\        P                  ! ^RR7      R\        P                  ! ^HRR7      RR/VP                  ^ &   \        \        P
                  ! R4      \        P
                  ! R	4      R..\        ^ .4      . ROR
7      pVR,          P                  R4      VR&   VR,          P                  R4      VR&   \        P                  ! W4       R# )abcr[   r+   r   )r   r.   r/   23z0 days 00:00:05z0 days 00:01:12r   zm8[s]Nrd   )
r   r`  r  to_timedeltar   	Timedeltar   r   r1   rx   )r3   r   r6   s   &  r)   &test_loc_expansion_with_timedelta_typeHTestDataFrameSetItemWithExpansion.test_loc_expansion_with_timedelta_type  s    4;/-#.


1
 ll,-r||<M/NPTUV*$

 !,,W5 ,,W5
f/r(   c                    \        4       p. ROVR&   \        R.RR7      p\        R. RO/VR7      p\        P                  ! W4       R# )r_   F)tupleize_colsr[   Nr^   r  )r   r   r1   rx   )r3   r5   r/  r6   s   &   r)   %test_setitem_tuple_key_in_empty_frameGTestDataFrameSetItemWithExpansion.test_setitem_tuple_key_in_empty_frame  s?    [6
fXU3fi0$?
b+r(   r!   N)r"   r#   r$   r%   r  r  r
  r  r  r  r%  r(  r+  r2  r6  r'   r  r  s   @r)   r  r  _  sD     . ,,+GZ,
D(0,0 , ,r(   r  c                   H  a  ] tR tRt o R t]P                  P                  R]]	P                  ]]P                  .4      ]P                  P                  R. R
O4      R 4       4       t]P                  P                  R]]	P                  ]]P                  .4      ]P                  P                  R. R
O4      R 4       4       t]P                  P                  R]]	P                  ]]P                  .4      ]P                  P                  R. R
O4      R 4       4       tRtV tR	# )TestDataFrameSetItemSlicingi  c                    \        \        P                  ! R4      4      p^VRR% \        P                  ! R4      p^VRR% \        V4      p\        P                  ! W4       R# )r   N)r   r_   )r   r>   zerosr1   rx   r$  s   &   r)   test_setitem_slice_position7TestDataFrameSetItemSlicing.test_setitem_slice_position  sP    rxx)*23hhx BCS>
b+r(   rz  r   c                    \        . RO.. RO.V,          ,           . ROR7      pV! . RO4      V! V4      R&   \        . RO.. RO.V,          ,           . ROR7      p\        P                  ! WE4       R# )r_   r[   r_   NNNr"  r<   ri   r  rd   rH  r   r3   r   rz  
indexer_sir5   r6   s   &&&&  r)   +test_setitem_slice_indexer_broadcasting_rhsGTestDataFrameSetItemSlicing.test_setitem_slice_indexer_broadcasting_rhs  sX     	{i[1_4oN .
2ri[L>A+==W
b+r(   c           	         \        . RO.. RO.V,          ,           . ROR7      pV! . RO4      VP                  \        \        ^V^,           4      4      &   \        . RO.. RO.V,          ,           . ROR7      p\        P
                  ! W44       R# )r_   r[   Nr"  rA  rd   rH  )r   r
  r`  ru   r1   rx   )r3   r   rz  r5   r6   s   &&&  r)   *test_setitem_list_indexer_broadcasting_rhsFTestDataFrameSetItemSlicing.test_setitem_list_indexer_broadcasting_rhs  sg     	{i[1_4oN),\):U1a!e_%&i[L>A+==W
b+r(   c                    \        . RO. RO.. RO.V,          ,           . R	OR7      pV! . R
O4      V! V4      R&   \        . RO.. R
O.V^,           ,          ,           . R	ORR7      p\        P                  ! WE4       R# )r_   r[   r@  r.  r	  Nr"  r   rA  rd   rH  r   rB  s   &&&&  r)   0test_setitem_slice_broadcasting_rhs_mixed_dtypesLTestDataFrameSetItemSlicing.test_setitem_slice_broadcasting_rhs_mixed_dtypes  sm     (I;?:O
 !.
2rK<.AE22#

 	b+r(   r!   Nr^   )r"   r#   r$   r%   r=  rj   r  r  r   r>   rC   r`  r  rD  rG  rJ  r'   r  r  s   @r)   r9  r9    s     , [[UVRXXtRXX$FG[[S),, - H, [[UVRXXtRXX$FG[[S),, - H, [[UVRXXtRXX$FG[[S),, - H,r(   r9  c                   ,   a  ] tR tRt o R tR tRtV tR# )TestDataFrameSetItemCallablei%  c                    \        R . ROR. RO/4      p. ROVR &   \        R . ROR. RO/4      p\        P                  ! W4       R# )rJ   rK   c                     R # )rJ   r!   rV  s   &r)   rW  DTestDataFrameSetItemCallable.test_setitem_callable.<locals>.<lambda>)  s    Sr(   Nr  r  )rI  rJ  r  r  r   r   s   &  r)   test_setitem_callable2TestDataFrameSetItemCallable.test_setitem_callable&  sA    \3=>,=.\BC
b&r(   c                    R  p\        R^.^R..\        R7      pWV^ 8  &   \        RV.VR..4      p\        P                  ! W#4       R# )c                     V ^,           # )r_   r!   rV  s   &r)   incETestDataFrameSetItemCallable.test_setitem_other_callable.<locals>.inc0  s    q5Lr(   r=   Nrs  )r   r.  r1   rx   )r3   rU  r5   r6   s   &   r)   test_setitem_other_callable8TestDataFrameSetItemCallable.test_setitem_other_callable.  sR    	 Q!R)826
r3i#r34
b+r(   r!   N)r"   r#   r$   r%   rQ  rW  r'   r  r  s   @r)   rM  rM  %  s     '
, 
,r(   rM  c                   $  a  ] tR tRt o ]P
                  P                  RR R .RR.R7      R 4       t]P
                  P                  R	R
7      ]P
                  P                  R]	]
P                  ].4      R 4       4       t]P
                  P                  R]	]
P                  ].4      R 4       tR tR t]P
                  P                  RRR.4      ]P
                  P                  R/ R^./RR./.4      R 4       4       tR tR tR tRtV tR# )TestDataFrameSetItemBooleanMaski;  	mask_typec                B    V \         P                  ! V 4      ^,          8  # r<   )r>   absr5   s   &r)   rW  (TestDataFrameSetItemBooleanMask.<lambda>>  s    Ba'r(   c                V    V \         P                  ! V 4      ^,          8  P                  # r]  )r>   r^  rN   r_  s   &r)   rW  r`  >  s    R"&&*q.5H4P4Pr(   	dataframerC   )idsc                j   VP                  4       pV! V4      pVP                  4       p\        P                  WT&   VP                  P                  4       p\        P                  V\        P                  ! V4      &   \        WcP                  VP                  R 7      p\        P                  ! WV4       R# )r   N)
r	  r>   r5  rN   rC   r   r-   r\   r1   rx   )r3   r[  rE   r5   r  r   r6   s   &&&    r)   test_setitem_boolean_mask9TestDataFrameSetItemBooleanMask.test_setitem_boolean_mask<  s{     } vv99>>##%66$ XXXrzzJ
f/r(   z1Currently empty indexers are treated as all False)reasonrz  c                   \        R R .R^.R^./4      pV\        8X  d   V! . RR7      pMV! . 4      pRp\        P                  ! \        VR7      ;_uu_ 4        ^.VP
                  VR.3&   RRR4       R#   + '       g   i     R# ; i)r+   r.   r/   r.  r=   z@Must have equal len keys and value when setting with an iterablerX   N)r   r   rj   rk   rl   r   )r3   rz  r5   r3  rp   s   &&   r)   :test_setitem_loc_empty_indexer_raises_with_non_empty_valueZTestDataFrameSetItemBooleanMask.test_setitem_loc_empty_indexer_raises_with_non_empty_valueO  sx     cUC!cA378&="H-G"gGP]]:S11&'SBFF7SE>" 2111s   A77B	c                $   \        R R .R^.R^./4      pV! R.4      p^
VR,          ,
          VP                  VR.3&   \        R R .R^.R^./4      p\        P                  ! W$4       ^	VP                  VR.3&   \        P                  ! W$4       R# )r+   r.   r/   FN)r   r   r1   rx   )r3   rz  r5   r3  r6   s   &&   r)   1test_setitem_loc_only_false_indexer_dtype_changedQTestDataFrameSetItemBooleanMask.test_setitem_loc_only_false_indexer_dtype_changed\  s    
 cUC!cA378ug,!#bgw~cC5#sC!=>
b+!"w~
b+r(   c                    \        R . ROR. RO/4      pVP                  4       pVR ,          ^8  pV! V4      V,          P                  R 4      V! V4      V&   \        P                  ! W#4       R# )r+   r.   N)r_   ri   r<   r`   r  )r   r	  r  r1   rx   )r3   
indexer_slr5   r6   r  s   &&   r)   "test_setitem_boolean_mask_aligningBTestDataFrameSetItemBooleanMask.test_setitem_boolean_mask_aligningj  s]    \3=>779#w!|)"~d3??D
2t
b+r(   c                   \        . ROR R.R7      p\        . R	O4      p. R
Op\        RVRV/VR7      p\        . RO. ROR7      p\        . R	O4      p. ROp\        RVRV/VR7      pVP                  4       p	V	R,          P                  P                  . RO4      V	R&   VR,          R8H  p
R^.W&   \        P                  ! W4       R# )r+   r.   )rY  catsrN   r,   r/   N)r+   r+   r.   r.   r+   r+   r+   )rf   rg   rh   klr_  r   )r_   r_   r<   r<   r_   r_   r_   )r+   r+   r/   r/   r+   r+   r+   rd   )r_   r_   r`   r`   r_   r_   r_   )r
   r   r   r	  r   set_categoriesr1   rx   )r3   cats2idx2values2exp_multi_rowcatsfidxfvaluesfr5   	exp_fancyr  s   &          r)   test_setitem_mask_categorical=TestDataFrameSetItemBooleanMask.test_setitem_mask_categoricalr  s     ?SRUJW89'!65(G"DDQ/O
 89'x9F!&&(	%f-11@@Q	&&zS 8
b,r(   r9   r   r:   kwargsr-   r\   rJ   c                ~    WR &   \        R/ VB pVP                  4       p^/W3V8  &   \        P                  ! W44       R# )r9   Nr!   r  )r3   r9   r  r5   r  s   &&&  r)   %test_setitem_empty_frame_with_booleanETestDataFrameSetItemBooleanMask.test_setitem_empty_frame_with_boolean  s<      w  ggi8
b&r(   c           
     V   \        \        ^4      4      p. ROp\        VV\        P                  ! . RO. R	O. R
O.\
        R7      R7      p\        W\        P                  ! \        V4      \        V4      34      R7      p\        VV\        P                  ! . RO. RO. RO.\
        R7      R7      pRW3RV,          8  &   \        P                  ! W54       \        P                  ! \        RR7      ;_uu_ 4        RW3P                  RR ^8  &   RRR4       R#   + '       g   i     R# ; i)r`   r   r=   )r-   r\   r4   zItem wrong lengthrX   Nr   )        r   r   )      ?r         @)r         @      @rs  )r  r   rs  )rs  rs  rs  )r`  ru   r   r>   rC   r   r  rB   r1   rx   rj   rk   rl   r-   )r3   r   r/  df1r  r6   s   &     r)   test_setitem_boolean_indexing=TestDataFrameSetItemBooleanMask.test_setitem_boolean_indexing  s    58n /?C5
 cbggs3xT>S6TU?NLIQVW
  "#)O
c,]]:-@AA&(C		#2"# BAAAs   8DD(	c                    \        R ^^.R^^.RR /4      pVP                  4       p\        RR.RR7      p\        R^^./^ ^.R7      VP                  VR.3&   \        P
                  ! W4       R# )r+   r.   r/   FrU   r,   N)r   r	  r   r   r1   rx   )r3   r5   r6   r3  s   &   r)   -test_loc_setitem_all_false_boolean_two_blocksMTestDataFrameSetItemBooleanMask.test_loc_setitem_all_false_boolean_two_blocks  sq    aVS1a&#s;<779%c2!*C!Q=A!Gw~
b+r(   c           
         \        R^.^R..4      p\        ^ ^.^^ ..4      p\        ^ \        RR.RR7      ^\        \        P                  R.RR7      /4      p^ W&   \        P
                  ! W4       R# )r_   TFbooleanr=   Nrs  r;  )r   r   r  NAr1   rx   )r3   r5   r6   boolean_indexers   &   r)   test_setitem_ea_boolean_mask<TestDataFrameSetItemBooleanMask.test_setitem_ea_boolean_mask  s{    Q!R)*q!fq!f-.#64-y96255$-y9
  
b+r(   r!   N)r"   r#   r$   r%   rj   r  r  re  xfailr`  r>   rC   r   ri  rl  rp  r  r  r  r  r  r'   r  r  s   @r)   rZ  rZ  ;  s1    [[	')PQ'"  
0
0 [[QR[[UT288V$<=	) > S	) [[UT288V$<=, >,,-. [[Ww&89[[XWqcNY<N'OP' Q :').,, ,r(   rZ  c                     a  ] tR tRt o R t]P                  P                  RRR.4      R 4       tR t	]P                  P                  R^]
P                  ! ^.^..R	R
7      ^.^...4      R 4       t]P                  P                  RR]
P                  ! R.R..4      R.R...4      R 4       t]P                  P                  RRR.]P                  ! RR.]P                  P                  RRR7      R7      .4      ]P                  P                  RR#R$]! R4      ]! R4      3R%.4      R 4       4       tR tR t]P                  P                  RR	R.4      R 4       tR tR tR t]P                  P                  R^ ^.]! R 4      3. R&O. R&O.3]! R 4      ]! R 4      3^3^ ^^.3^^.3^ .^3^.3^ .]
P2                  ! ^4      3^.3]! R 4      ]
P2                  ! ^4      3. R'O3]! R ^4      ]
P2                  ! ^4      3^^.3]
P                  ! . R(O4      ]
P                  ! . R)O4      3^^.3.4      R! 4       tR"tV tR # )*%TestDataFrameSetitemCopyViewSemanticsi  c                "   R VP                   9  g   Q hVR,          P                  4       pW!R &   \        P                  VP                  RVP                   P                  R 4      3&   \        VR,          4      P                  4       '       g   Q hR# )rR  rJ   :r   rt   NN)r\   r	  r>   r5  r
  get_locr   r   )r3   rE   r   s   && r)   test_setitem_always_copy>TestDataFrameSetitemCopyViewSemantics.test_setitem_always_copy  sx    +-----!!#CCE66{22::3??@QtW~!!####r(   consolidateTFc                
   \        R . ROR. RO/. ROR7      pVP                  ^R\        P                  4       V'       d8   VP	                  4        \        VP                  P                  4      ^8X  g   Q hM&\        VP                  P                  4      ^8X  g   Q h^*VP                  R	&   \        \        P                  \        P                  ^*^*.VP                  RR7      p\        P                  ! VR,          V4       R# )
r   r   r,   r   rW   N)g?g @g@gffffff@)gffffff@gffffff@gffffff@g333333 @)r   r_   r<   r`   ):r<   NNr   )r   r   r>   r5  _consolidate_inplacerB   r  r  r   r   r-   r1   r   )r3   r  r5   r6   s   &&  r)   #test_setitem_partial_column_inplaceITestDataFrameSetitemCopyViewSemantics.test_setitem_partial_column_inplace  s     &-AB,
 			!S"&&!##%rww~~&!+++rww~~&!+++w2662662r2"((M
r#w1r(   c                    R R.^,          p\        R^ .VR7      pVP                  4       pVR,          pRVR&   \        . RO.VR7      p\        P                  ! WC4       \        P                  ! W%4       R# )	rJ   rK   r  r   r(  r[   Nr:  )r  r<   r  r   r  )r3   r/  r5   df_copydf_viewr6   s   &     r)   *test_setitem_duplicate_columns_not_inplacePTestDataFrameSetitemCopyViewSemantics.test_setitem_duplicate_columns_not_inplace  sh    SzA~s1#t4'')Q%3./>
g/
b+r(   r  r:   r=   c                    R R.p\        ^ ^ ^.VR7      pVP                  4       pVR,          pWR.&   \        ^ ^.^ ^..VR7      p\        P                  ! W64       \        P                  ! WT4       R# )rJ   rK   r   r(  r[   Nr  r3   r  r/  r5   r  r  r6   s   &&     r)   #test_setitem_same_dtype_not_inplaceITestDataFrameSetitemCopyViewSemantics.test_setitem_same_dtype_not_inplace  sq    
 SzqA5'')Q%C5	q!fq!f-t<
b+
g/r(   r   c                    R R.p\        ^ ^ ^.VR7      pVP                  4       pVR,          pWR.&   \        ^ R.^ R..VR7      p\        P                  ! WT4       \        P                  ! W64       R# )rJ   rK   r   r(  r   r[   Nr  r  s   &&     r)   2test_setitem_listlike_key_scalar_value_not_inplaceXTestDataFrameSetitemCopyViewSemantics.test_setitem_listlike_key_scalar_value_not_inplace  sq     SzqA5'')Q%C5	q#hC14@
g/
b+r(   r3  r+   z+Boolean indexer incorrectly setting inplace)rg  strict)markszvalue, set_valuez
2020-12-31z
2021-12-31c                    \        R V/^ ^.R7      pVP                  4       pVR,          pW$V&   \        P                  ! We4       R# )r+   r,   r(  Nr  )r3   r  	set_valuer3  r5   r6   r   s   &&&&   r)   "test_setitem_not_operating_inplaceHTestDataFrameSetitemCopyViewSemantics.test_setitem_not_operating_inplace  sA    2 U|Aq62779!u7
d-r(   c                h   \        ^
4       Uu. uF  pRV 2NK
  	  pp\        V Uu/ uF"  q3\        P                  ! \	        V4      4      bK$  	  upVR7      pVP
                  P                  ^ ,          P                  p\        P                  ! 4       ;_uu_ 4        VP                   F  p^WF,          V&   K  	  RRR4       \        P                  ! V\        P                  ! ^
4      \        P                  ! ^
4      3,          ^ 8H  4      '       g   Q hR# u upi u upi   + '       g   i     Ln; i)rt   r/   r,   N)ru   r   r>   r<  rB   r  r  rN   r1   raises_chained_assignment_errorr\   r   r   )r3   rg   r  colr5   rN   labels   &      r)   "test_setitem_column_update_inplaceHTestDataFrameSetitemCopyViewSemantics.test_setitem_column_update_inplace'  s     $)9-9aAaS'9-fEfsRXXc&k22fEVT"))//11#$	%  $ 2 vvfRYYr]BIIbM9:a?@@@@ .E 21s   D(DD!!D1	c                    \        . RO4      p\        . RORR7      VR&   \        . RORR7      VR&   \        . RO. R	O\        R7      p\        P                  ! VP
                  V4       R# )
r_   r^  r=   col1col2)r-   r9   Nr^   )r:   r^  r^  )r   r  r  )r   r   r.  r1   r   r   )r3   r5   expected_typess   &  r)   %test_setitem_column_frame_as_categoryKTestDataFrameSetitemCopyViewSemantics.test_setitem_column_frame_as_category4  sU    y!y
;6
IZ86
-5HPV
 	ryy.9r(   r9   r  c                <   \        R \        P                  ! ^4      /VR7      p\        P                  ! ^..4      VP                  \        P                  ! ^ .4      \        P                  ! ^ .4      3&   \        R . RO/VR7      p\
        P                  ! W#4       R# )r+   r=   N)r<   r_   r_   )r   r>   r  rC   r
  r1   rx   )r3   r9   r5   r6   s   &&  r)   "test_setitem_iloc_with_numpy_arrayHTestDataFrameSetitemCopyViewSemantics.test_setitem_iloc_with_numpy_array?  so     RWWQZ(6021#!rxx},-c9-U;
b+r(   c                j   \        . RO. RO.. ROR7      p\        ^ R.^R..RR.R7      pW!R&   \        . R	O. R
O.. ROR7      p\        P                  ! W4       \        . RO. RO.. ROR7      p\        ^ R.^R..RR.R7      pW!R&   \        . RO. RO.. ROR7      p\        P                  ! W4       R# )r_   r+   r[   r  r  Nr  )ri   r   r  r  )r+   r.   r+   r/   )r   r<   r  ri   )r<   r   r  r  r^   r  )r+   r+   r.   )r   r  r`   )r<   r  r  r   rK  s   &   r)   !test_setitem_frame_dup_cols_dtypeGTestDataFrameSetitemCopyViewSemantics.test_setitem_frame_dup_cols_dtypeH  s    l3=QR!SAs8,sCjA3^,6J
 	b+	9-G!SAs8,sCjA3k;7Q
b+r(   c                    \        R .RRR7      p\        RV/4      P                  R4      pVR,          P                  4       pRVR&   RVR&   \        . RR.VR,          R	7      p\        P
                  ! W#4       R# )
z
2000-01-01r   date)r9   rD   :r   r   NN30102010:Nr   Nre  )r   r   r  r	  r1   rx   )r3   dtir5   r6   s   &   r)   "test_frame_setitem_empty_dataframeHTestDataFrameSetitemCopyViewSemantics.test_frame_setitem_empty_dataframeX  sx    \N(H}%//7W\\^6
6
V$b'

 	b+r(   c                    \        R . ROR. R	OR. R
OR. RO/4      pVP                  4       pVRR.,          p^^.^^..VP                  ^^.R3&   \        P                  ! W4       \        R. ROR. RO/VP
                  R7      p\        P                  ! W44       V. RO,          p^^.^^..VP                  ^^.R3&   \        P                  ! W4       \        R. ROR. R
OR. RO/VP
                  R7      p\        P                  ! W44       R# )rJ   rK   r   r   r(  r,   :r   r`   r<   N)r_   ri   r_   r   )r<   r   r<   r  )r`   r  r_   r  )r  r  rt   rI  r<   r<   r<   r<   )r`   r<   r_   r<   r  )r  r<   rt   r<   )r   r	  r
  r1   rx   r-   )r3   	df_parentdf_origr5   r6   s   &    r)   test_iloc_setitem_view_2dblockDTestDataFrameSetitemCopyViewSemantics.test_iloc_setitem_view_2dblockh  s   \\\^	
	 .."Sz"  !!fq!f-A	 	i1 c<lC288T
b+ '#$a&1a&!1A
i1,\3Fbhh
 	b+r(   zindexer, valueNc                   \         P                  ! ^4      P                  ^^4      p\        VP	                  4       \        R4      R7      pVP                  P                  '       g   Q hVP	                  4       pVR,          pW&P                  V&   \        P                  ! WE4       W#V&   \        V\        R4      R7      p\        P                  ! Wg4       R# )rJ  ABCDr[   r(  N)r>   r   rw   r   r	  r`  r  is_single_blockr
  r1   rx   )r3   r3  r  rF   r  r  r5   r6   s   &&&     r)   test_setitem_2dblock_with_refCTestDataFrameSetitemCopyViewSemantics.test_setitem_2dblock_with_ref  s    $ iim##Aq)chhj$v,?	 ~~-----.."q\  	i1 GS$v,7
b+r(   r!   )r_   r   )r   g      @r  r  )r<   r<   r<   )FTF)FTFT)r"   r#   r$   r%   r  rj   r  r  r  r  r>   rC   r  r  paramr  r   r  r  r  r  r  r  r  slicer:   r  r'   r  r  s   @r)   r  r    s    $ [[]T5M:2 ;2*
, [[!RXXsQCj8A3*E
0
0 [[WsBHHsecU^,Dusen&UV
, W
, [[ELLukk''H  ( 
	
 [[|$i&=>		
...A	: [[Ww&89, :,, , ,@ [[!feDk"\<$@ADk5;'+!Q[1a&!c1XsOc288A;!%Dk288A;'3D!nbhhqk*QF3./:T1UVA	
 ,! ,r(   r  c                     \        R ^^./4      p \        P                  ! \        RR7      ;_uu_ 4        RV P                  R&   RRR4       \        P                  ! \        RR7      ;_uu_ 4        ^ R^R/V P                  R&   RRR4       ^ ^^^/V P                  R&   \        R ^^./4      p\
        P                  ! W4       R#   + '       g   i     L; i  + '       g   i     L]; i)r+   zInvalid valuerX   TNr  g      @r  )r   rj   rk   	TypeErrorr   r1   rx   r5   r6   s     r)   'test_full_setter_loc_incompatible_dtyper    s    	C!Q=	!B	y	8	8v 
9 
y	8	8S!S)v 
9 Aq\BFF6N#1v'H"' 
9	8 
9	8s   C0CC	C)	c                      \        R . ROR. R	O/4      p R
V P                  V P                  ^8*  RR.3&   \        R . ROR. R	ORR^\        R4      .RRR\        R4      ./4      p\        P
                  ! W4       R# )rJ   rK   r.  rS  Gr   r5  Nr^   )r  r   r  )r_   r.  )r   r   r-   r   r1   rx   r  s     r)   )test_setitem_partial_row_multiple_columnsr    s}    	CC5	6B(2BFF288q=3*$%#q%,'%e-		
H "'r(   )0r   numpyr>   rj   pandas.core.dtypes.baser   r  pandas.core.dtypes.commonr   pandas.core.dtypes.dtypesr   r   r   r	   pandasr  r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   pandas._testing_testingr1   pandas.core.arraysr   pandas.tseries.offsetsr   r   r  r  r9  rM  rZ  r  r  r  r!   r(   r)   <module>r     s       < 5      $  * 'I, I,X(1 (1VW, W,t), ),X, ,,C, C,Lj, j,Z((r(   