+
    yȇiH                         ^ RI t^ RIt^ RIt^ RIHtHtHtHtH	t	H
t
 ^ RIHt R t ! R R4      t ! R R4      tR tR	 tR
 tR tR# )    N)	DataFrame
MultiIndexSeries
date_rangeisnanotnac                     W8X  g   Q hR # N )abs   &&}/Users/max/.openclaw/workspace/postharvest/venv/lib/python3.14/site-packages/pandas/tests/indexing/multiindex/test_setitem.pyassert_equalr      s    6M6    c                      a  ] tR t^t o ]R3R l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R t]P*                  P-                  R4      R 4       tR tR tR tR tR tRtV tR# )TestMultiIndexSetItemNc                d    W1P                   V&   VP                   V,          pVf   TpV! We4       R # r
   )loc)selftargetindexersvalue
compare_fnexpectedresults   &&&&&& r   checkTestMultiIndexSetItem.check   s/    $

8H%H6$r   c                   . R	Op\         P                  ! \        P                  ! ^ ^d4      \        P                  ! ^ ^P4      .RR.R7      p^ ^rC\	        \        P
                  VVR7      pV P                  WSV3R3^ R7       \	        R
WR7      pV P                  WSV3R3^R7       \	        WR7      pV P                  WSV3R3^R7       \	        R
WR7      pV P                  VW43R3\        P                  ! ^4      ^R7       R# )AXtimefirmnamescolumnsindex)r   r   r   )r   r   r   r   N)r   wlr   xr    dprofiti)r   from_productnparanger   nanr   array)r   colsr'   tndfs   &     r   test_setitem_multiindex-TestMultiIndexSetItem.test_setitem_multiindex   s    <''YYq#		!R 01&&9I
 !1FF

 	

"A}A
>tT7

"A}A
>t1

"A}A
> tT7

fc]((1+	 	 	
r   c                   \        \        P                  ! ^4      P                  ^^4      RP	                  R4      \
        R7      p^cVR&   VR,          ^,          ^ 8H  pRR.pVR,          VP                  W#3&   \        R. RORR.R	7      p\        P                  ! VP                  W#3,          V4       V P                  VW#3VR,          \        P                  VR
7       R# )   z	A,B,C,D,E,r&   dtypeFr   BCg     X@r'   r&   r   r   r   r   r   N)r         )
r   r.   r/   reshapesplitfloatr   tmassert_frame_equalr   )r   r5   row_selectioncol_selectionoutputs   &    r   test_setitem_multiindex2.TestMultiIndexSetItem.test_setitem_multiindex2:   s    IIbM!!!Q'1B1B31Gu
 33!q(c
/1#w}+,4y3*E
bff]%ABFK

#3S',, 	 	
r   c           	        \         P                  ! . RO\        RRRR7      .4      p\         P                  ! RR.\        RRRR7      .4      p\        \        P
                  P                  ^4      P                  R4      WR	7      p\         P                  ! R R .\        RR
RR7      .4      p\         P                  ! RR.\        RRRR7      .4      p\        \        P
                  P                  ^4      P                  R4      WER	7      pV P                  VWE3V\        P                  R7       \        \        P
                  P                  ^4      P                  R4      WBR	7      pV P                  VV\        RRR4      3V\        P                  R7       VP                  4       pV P                  VVP                  VP                  3V\        P                  VR7       R# )r   z
2015-01-01z
2015-04-01MS)freqfoobarz
2016-01-01z
2016-02-01r@   z
2015-02-01)r   r   r   r   NrA   r   r>   r?   )   rC   )rB   rB   )rB   rC   )r   r-   r   r   r.   randomdefault_rngfrom_arraysr   rG   rH   slicecopyr'   r&   )r   idxr2   r5   subidxsubcolsvalsrY   s   &       r   test_setitem_multiindex3.TestMultiIndexSetItem.test_setitem_multiindex3M   s   %%j|$OP
 &&U^ZlNO
 II!!!$++G4C
 ''3ZL,TJK
 ((U^ZlNO
 II!!!$++F36
 	

&,,	 	 	
 II!!!$++F36
 	

eD$56,,	 	 	
 wwy

hh

+,, 	 	
r   c                   \         P                  ! . RO4      \         P                  ! . RO4      \         P                  ! ^ ^^4      .p\        \         P                  P                  ^4      P                  R4      V. ROR7      P                  4       pVP                  R .,          ^,          pVP                  4       pVP                  R .;;,          ^,          uu&   \        P                  ! VP                  R .,          V4       Rp\        P                  ! \        VR7      ;_uu_ 4        VP                  R ;;,          ^,          uu&   RRR4       R#   + '       g   i     R# ; i)	rR   r@   zAcannot align on a multi-index with out specifying the join levelsmatchN)rR   rR   bazquxrd   rR   )onetwore   re   rf   re   )      rS   )r.   r1   r/   r   rU   rV   standard_normal
sort_indexr   rY   rG   rH   pytestraises	TypeError)r   arraysdf_origr   r5   msgs   &     r   test_multiindex_setitem-TestMultiIndexSetItem.test_multiindex_setitem~   s     HH?@HH?@IIaA
 II!!!$44V<#
 *,	 	 ;;w'!+\\^
w1
bffeWox8 R]]9C00FF5MQM 1000s   , EE'	c                Z   \         P                  ! R R^R^R^R^R^ R	^ //4      p\        P                  ! VP                  . R
OR7      Vn        VP                  4       pVP                  . RO;;,          ^,          uu&   \        P                  pVP                  4       pVP                  VR,          R3;;,          ^,          uu&   \        P                  ! WB4       VP                  4       pVP                  VR,          R 3;;,          ^,          uu&   \        P                  ! WB4       R# )pricer#   NNNN)DECoalStock)rv   Gasrx   )rv   ElecDemand)FRry   rx   )r|   SolarSupIm)r|   Windr~   )SitComType)r      rh   )ru   ru   rx   )r   	from_dictr   from_tuplesr'   rY   ilocpd
IndexSlicer   rG   rH   )r   ro   r   rZ   r5   s   &    r   test_multiindex_setitem2.TestMultiIndexSetItem.test_multiindex_setitem2   s     %%+Q*A,a*A,a+Q	
 #..MM!7
 <<>i A% mm\\^
s=!1$%*%
b+\\^
s=!7*+q0+
b+r   c           	        \        \        P                  P                  ^4      P	                  ^^
^	R7      P                  ^^4      \        R4      . R	O. R
O.R7      p\        P                  VR&   \        P                  ! RR.4      pW!P                  R&   \        P                  ! VP                  R,          \        V^^
.RR7      4       R# )rB   sizeabcr%   r+                 ?)r'   nameNrC   rC      r   
   rT   )rC   r+   )r   r.   rU   rV   integersrD   listr0   r1   r   rG   assert_series_equalr   )r   r5   arrs   &  r   test_multiindex_assignment0TestMultiIndexSetItem.test_multiindex_assignment   s     II!!!$--a!-<DDQJKk*

 &&3hhSz"v
rvvf~vc!Rs/STr   c                   \         P                  ! R R.4      p\        \         P                  P	                  ^4      P                  ^^
^	R7      P                  ^^4      \        R4      . RO. RO.\         P                  R7      pWP                  R&   \        V^^
.RRR7      pVP                  R,          p\        P                  ! WC4       \        P                  ! \        RR	7      ;_uu_ 4        VR
,           VP                  R&   RRR4       VP!                  RR/4      pVR
,           VP                  R&   ^
VP                  R&   \        ^
^^
.RRR7      p\        P                  ! VP                  R,          V4       Rp\        P                  ! \"        VR	7      ;_uu_ 4        . ROVP                  R&   RRR4       \        P                  ! \"        VR	7      ;_uu_ 4        ^ .VP                  R&   RRR4       ^ .VP                  ^R.3&   VP                  R,          ^ 8H  P%                  4       '       g   Q hR#   + '       g   i     EL@; i  + '       g   i     L; i  + '       g   i     Lx; i)r   r   r   r   )r&   r'   r<   cint64)r'   r   r<   zInvalid valuera   g      ?Nfloat64z@Must have equal len keys and value when setting with an iterabler   r   )rC   r   )r   r   rB   rh   )r.   r1   r   rU   rV   r   rD   r   r   r   r   rG   r   rk   rl   rm   astype
ValueErrorall)r   r   r5   expr   rp   s   &     r   'test_multiindex_assignment_single_dtype=TestMultiIndexSetItem.test_multiindex_assignment_single_dtype   s    hhSz"II!!!$--a!-<DDQJKk*((	
 vSBcA
v+ ]]9O<< 3YBFF6N = YYY'(sv vR2wS	B
rvvf~s3 Q]]:S11)BFF6N 2 ]]:S11SBFF6N 2
 3q3%xv!#((****- =<< 21 21s$   0H>&I$I%>I	I"	%I5	c                j   ^dp^
p\        \        \        P                  ! V4      P	                  4       4       Uu. uF  pRV,           NK  	  ppVR,          p\        \        P                  P                  ^4      P                  ^W3R7      \        P                  VR7      pVP                  V4      P                  4       pVP                  VR,          R7      p\        P                  VR&   V F<  w  r\        P                  ! V	P                  ^ ,          4      p
WP                  VR3&   K>  	  R# u upi )	d   r   N   Nr   )r<   r&   :NrC   N)levelnew_colN)mapstrr.   r/   tolistr   rU   rV   r   r   	set_indexrj   groupbyr0   shaper   )r   NUM_ROWSNUM_COLSnum	col_names
index_colsr5   grpr   df2new_valss   &          r   test_groupby_example*TestMultiIndexSetItem.test_groupby_example   s    *-c299X3F3M3M3O*PQ*P3S3YY*P	Qr]
II!!!$--ax6J-K((

 \\*%002jjz"~j.9 IDyy1.H&.FF4?#  Rs   D0c                p   TpVR ,          p\         P                  VR&   \        VP                  R,          4      P	                  4       '       g   Q h\        VP                  R,          4      P	                  4       '       g   Q h\        VP                  R,          4      P	                  4       '       g   Q h\         P                  VR&   \        VP                  ^1,          4      '       g   Q h\        P                  ! \        RR7      ;_uu_ 4        V^1,           RRR4       R#   + '       g   i     R# ; i)	r   :*   A   N:Nr   N:r   NN49ra   N)  rh   )r   rh   r   )
r.   r0   r   valuesr   r   r   rk   rl   KeyError)r   /multiindex_year_month_day_dataframe_random_dataymdss   &&  r   test_series_setitem)TestMultiIndexSetItem.test_series_setitem  s    =HVV'
AHHUO$((****QXXc]#''))))QXXc]#''))))+AFF2J]]8400bE 1000s   
D$$D5	c                   TpVP                   P                  4       pVP                  P                  4       pW3^ 8  ,          pVP                  V^ 8  4      p\        P
                  ! WV4       ^W3^ 8  &   ^WD^ 8  &   \        P                  ! VP                  V4       ^ W3^8H  &   ^ WD^8H  &   \        P                  ! VP                  V4       ^W3RR ^ 8  &   \        P                  ! VRR VRR ^ 8  ^4       \        P                  ! VP                  V4       \        P                  ! \        RR7      ;_uu_ 4        ^W3^ ,          &   RRR4       R#   + '       g   i     R# ; i)r   Nzboolean values onlyra   )TrY   r   whererG   rH   assert_almost_equalr.   putmaskrk   rl   rm   )r    multiindex_dataframe_random_dataframer5   r   r   r   s   &&     r   "test_frame_getitem_setitem_boolean8TestMultiIndexSetItem.test_frame_getitem_setitem_boolean  s   0WW\\^!F88BF#
f/6
z
ryy&17{
ryy&1 cr7Q;


6#2;sa3
ryy&1]]9,ABBBAvJ CBBBs   :EE!	c                R   R R.. RO.p. R	O. R
O.p\        W!RR.R7      p\        R. RO/VR7      pVP                  R,          p\        P                  ! VR,          V4       VP                  VP
                  R,          R3,          p\        P                  ! VR,          R,          V4       VP                  R,          p\        P                  ! WE4       Tp^
VP                  R&   ^
VR&   \        P                  ! WE4       ^
VP                  R&   \        P                  ! WE4       R# )t1t2Nid)codeslevelsr$   r   r'   :r   rh   Nr   r   r   )r   r   r   r   r   )r   r   rB   r   r   )r   rB   rh      r   )ru   r   )ru   ru   )r   r   r   rG   r   r'   rH   )r   r   r   midxr5   r   s   &     r   %test_frame_getitem_setitem_multislice;TestMultiIndexSetItem.test_frame_getitem_setitem_multislice/  s    ,0 /2T4LI1>
#
r'{F3w./
r'{3/8
b)zw
b)t
b)r   c                   \        \        P                  P                  ^4      P	                  R4      . RO. RO.R7      pVP                  4       pVR,          VR&   \        P                  ! VR,          VR,          4       VP                  4       pVR,          P                  VR&   \        P                  ! VR,          VR,          4       R# )rB   r   r   r&   N)r   rC   )r   r   r   r   )r   r   r   r   )	r   r.   rU   rV   ri   rY   rG   rH   r   )r   r5   cps   &  r   test_frame_setitem_multi_column5TestMultiIndexSetItem.test_frame_setitem_multi_columnF  s    II!!!$44W=)<8

 WWYS'3
bgr#w/ WWYS'..3
bgr#w/r   c                   \         P                  ! . R	O4      p\        . R
OVR7      pRVR &   VR ,          P                  ^ 8H  P	                  4       '       g   Q h. ROVR&   VR,          VR &   VR,          pVR,          pVR,          p\
        P                  ! W5RR7       \
        P                  ! WERR7       VP                  R8X  g   Q hVP                  R8X  g   Q hVP                  R8X  g   Q hR# )r   r@   r   F)check_namesNr   1r   2r>   r   )r   r   r   )r   rh   r   r   rB   rh   )r   r   r   r   r   rG   r   r   )r   r&   r5   	sliced_a1	sliced_a2	sliced_b1s   &     r    test_frame_setitem_multi_column26TestMultiIndexSetItem.test_frame_setitem_multi_column2U  s     (()MNY8 33!#((**** !8X,3xL	xL	xL	
yG
yG~~+++~~+++~~+++r   c                    TpVR ,          pVP                   R. RO3,          pVP                   R,          . RO,          p\        P                  ! WE4       R# )r   N)r   r   rg   rS   )r   rG   r   )r   r   r   r5   r   r   s   &&    r   #test_loc_getitem_tuple_plus_columns9TestMultiIndexSetItem.test_loc_getitem_tuple_plus_columnsl  sF     >Wo5666*%o6
v0r   z.ignore:Setting a value on a view:FutureWarningc                   \        . RO^ ^... RO. RO.R7      p\        \        P                  P	                  ^4      P                  \        V4      ^34      V. R	OR7      p\        P                  ! W14      pVP                  R,          pVP                  VP                  R,          4      p\        P                  ! WE4       ^VP                  R&   VP                  R,          ^8H  P                  P                  4       '       g   Q hR# )
r   )r   r   r@   :r   rB   N:rB   NNNr   r   rB   )r   r   r   r   rB   rB   )r   r   r   r   r   r   )r   r   r   r+   )r   r   r.   rU   rV   ri   lenrG   get_objr   reindexr'   r   r   r   )r   frame_or_seriesr'   objresr   s   &&    r   'test_loc_getitem_setitem_slice_integers=TestMultiIndexSetItem.test_loc_getitem_setitem_slice_integersw  s    1v&/ACU.V
 II!!!$44c%j!_E(

 jj.ggclkk#))B-(
!!))--////r   c                <   TpVP                   pVR,          pWDP                  4       8  VR&   \        P                  ! VR,          WDP                  4       8  4       VP	                  R.R7      p\        P                  ! VR,          WDP                  4       8  4       R# )rQ   r   N)rQ   rf   )r   medianrG   r   r   )r   r   r   dftr   	reindexeds   &&    r   test_setitem_change_dtype/TestMultiIndexSetItem.test_setitem_change_dtype  s{    0gg
NL
s<0!hhj.A KK(8K9	
y6HHJGr   c                $   TpVP                   . RO,          p^cVP                  V&   VP                  V,          P                  ^c8H  P                  4       '       g   Q hVP	                  4       pVR,          p^aWS&   \
        P                  ! W$4       R# )r   r>   N)r   rC   r   )r'   r   r   r   rY   rG   rH   )r   r   r   subsetframe_originalcols   &&    r   test_set_column_scalar_with_loc5TestMultiIndexSetItem.test_set_column_scalar_with_loc  su    0Y'		&		&!((B.335555Cj
e4r   c                   \        . RO. R	O. R
O.\        R4      R7      pVP                  RR.4      p\        P                  ! R.4      pRVP
                  VR3&   VP                  R4      R,          R8H  P                  4       '       g   Q hR# )r   ABCDr   r   r>   _r?   N)r   r   r*   r    )r   r   yY)r   rB   zZr   r   )r   r   r   r   r   r   xsr   )r   r5   mis   &  r   test_nonunique_assignment_17504TestMultiIndexSetItem.test_nonunique_assignment_1750  s}    /1ABDQWL
 \\3*%##VH-r3wfc"c)..0000r   c                h   \         P                  ! . RO4      p\        \        P                  ! ^4      P                  R4      V\        R7      pVP                  P                  4       pVR ,          P                  \        P                  4      VR &   \        P                  ! VP                  V4       R# )r   r;   N)r   r   r   )r   rh   )r   r   r   r.   r/   rD   objectr'   rY   r   r   rG   assert_index_equal)r   r2   r5   r'   s   &   r    test_astype_assignment_with_dups6TestMultiIndexSetItem.test_astype_assignment_with_dups  ss     %%&JKryy|++F3TPS'..,3
bhh.r   c                    \         P                  ! . R
ORR.R7      p\        . ROVR.R7      p\        P                  ! ^c^eRR7      VP
                  R&   \        R. RO/VR7      p\        P                  ! W#4       R	# )r   l1l2r#   edatar'   r&   r   )r<   r   N))r   r   )r   r*   )r   r+   r   )r   r  )c   r   r   )r   r   r   r.   r/   r   rG   rH   )r   r'   r5   r   s   &   r   test_setitem_nonmonotonic/TestMultiIndexSetItem.test_setitem_nonmonotonic  sg    &&0t
 IUSEB99RG<xc<0>
b+r   r   )__name__
__module____qualname____firstlineno__r   r   r6   rL   r^   rq   r   r   r   r   r   r   r   r   r   r   rk   markfilterwarningsr   r  r  r  r  r!  __static_attributes____classdictcell____classdict__s   @r   r   r      s     8Dt %
<
&/
b2,>U)+V/, 2*.0,.	1 [[ PQ0 R0&	H5
1/, ,r   r   c                   8   a  ] tR tRt o R tR tR tR tRtV t	R# )"TestSetitemWithExpansionMultiIndexi  c                f   . RO. RO. RO.p\        \        V!  4      p\        P                  ! V4      p\	        \
        P                  P                  ^4      P                  R4      VR7      pVP                  4       pVP                  4       p. ROVR&   . ROVR	&   \        P                  ! WV4       R# )
r   r   r   N)r   topr0  routine1r1  routine2) ODr4  result1result2r5  )r3  wxwyr3  r3  r3  )rC   rg   )r   rB   rh   rC   )r   r3  r3  )sortedzipr   r   r   r.   rU   rV   ri   rY   rG   rH   )r   rn   tuplesr'   r5   r   r   s   &      r   #test_setitem_new_column_mixed_depthFTestSetitemWithExpansionMultiIndex.test_setitem_new_column_mixed_depth  s    C=(
 V%&&v.ryy,,Q/??GQVW779"s ,
f/r   c                    \         P                  ! . RO4      p\        ^^.^^.^^..VR7      p\        R^R^/4      pW2R&   VR,          P	                  4       P                  4       '       g   Q hR# )1ar   newN))r?  2a)r?  2b)r?  2cr  )r   rB   )r   r   r   r   r   r   )r   mixr5   r   s   &   r   test_setitem_new_column_all_naATestSetitemWithExpansionMultiIndex.test_setitem_new_column_all_na  sl    $$%OPAAA/s;FAvq)*5	%y~~##%%%%r   c                2   \         P                  ! R.. ROR7      p\        . RO.V. ROR7      pVP                  R,          VP                  R&   \         P                  ! RR.. ROR7      p\        . RO. RO.V. ROR7      p\        P
                  ! W$4       R# )	r   r#   r  Nr   )i1i2i3)r         rS   )r   r   r   r   r   r   r   rG   rH   )r   r  r5   mi_expectedr   s   &    r   )test_setitem_enlargement_keep_index_namesLTestSetitemWithExpansionMultiIndex.test_setitem_enlargement_keep_index_names  s    ##YK7IJ\N"oNFF9-y ,,	"*<
 -#

 	b+r   c                   \         P                  ! . RO4      p\        . ROVR
7      pRR.VP                  R&   \         P                  ! . RO4      p\        RR.RR.RR.RR	.RR..V^ ^.R7      p\        P
                  ! W$RR7       R# )r   Nr         @r         @       @      @      @      @r   g      (@g      *@r@   F)check_index_type)r   a1r   a2r>   b1r>   N))r   rR  )r   rS  )rT  rU  )rV  rW  )r   Nru   )rY  r[  r]  r_  r`  rM  )r   r'   r5   expected_indexr   s   &    r   -test_setitem_enlargement_multiindex_with_nonePTestSetitemWithExpansionMultiIndex.test_setitem_enlargement_multiindex_with_none  s     &&@
 GuU"&~#//M
 3Z#sc3Z#sdD\J F

 	bUCr   r   N)
r#  r$  r%  r&  r<  rE  rO  rb  r)  r*  r+  s   @r   r.  r.    s      0"&,D Dr   r.  c                 "   V P                   p\        P                  ! \        R R7      ;_uu_ 4        ^ VR,          P                  R&   RRR4       VR,          P                  ^ 8g  P                  4       '       g   Q hR#   + '       g   i     L>; i)z	read-onlyra   rQ   ru   N)r   rk   rl   r   r   r   r   r5   s   & r   test_frame_setitem_view_directrf     sf     
*	+	+B	z	5	55	 
6uI!&&(((( 
6	5s   A>>B	c                     V P                   p\        P                  ! 4       ;_uu_ 4        ^VR,          R&   RRR4       R#   + '       g   i     R# ; irB   rQ   re   N)r   rG   raises_chained_assignment_errorre  s   & r   test_frame_setitem_copy_raisesrj  	  s9    	)	+	+B		+	+	-	-5	% 
.	-	-	-s   AA	c                     V P                   pTpVP                  4       p\        P                  ! 4       ;_uu_ 4        ^VR,          R&   RRR4       \        P                  ! W24       R#   + '       g   i     L(; irh  )r   rY   rG   ri  rH   )r   r   r   r5   s   &   r    test_frame_setitem_copy_no_writerl    sX    ,..EH	B		+	+	-	-5	% 
. "' 
.	-s   A))A9	c            
          \        R . ROR. ROR^R^/4      P                  . RO4      p \        ^V P                  P	                  R4      R7      pV P                  4       pWR&   V P                  4       p^VR&   \        P                  ! W#4       R# )	r   r   r   r+   r   Nr   )rh   rC   r   r   )r   r   r   r'   	droplevelrY   rG   rH   )r5   serr   r   s       r   %test_frame_setitem_partial_multiindexrp    s    			

 i   "((,,S1
2CWWYF3KwwyHHSM&+r   )numpyr.   rk   pandasr   r   r   r   r   r   r   pandas._testing_testingrG   r   r   r.  rf  rj  rl  rp  r   r   r   <module>ru     sP        m, m,`9D 9Dx)(,r   