
    Mh                     >    S SK rS SKrS SKJr  S rS rS r	S r
S rg)    Nc                     [         R                  " / SQ/ SQS.5      R                  S5      n Sn[        R                  " [
        US9   U R                  S/SS	9R                  S
 5      nS S S 5        Sn[        R                  " [
        US9   U R                  S/SS	9R                  S 5      nS S S 5        [        R                  " WW5        g ! , (       d  f       Nk= f! , (       d  f       N7= f)N)AliceBobCarl)      r   )nameager	   7DataFrameGroupBy.apply operated on the grouping columnsmatchr
   F
group_keysc                     U $ N groups    X/var/www/html/env/lib/python3.13/site-packages/pandas/tests/groupby/test_apply_mutate.py<lambda>$test_group_by_copy.<locals>.<lambda>   s    %    c                 "    U R                  5       $ r   )copyr   s    r   r   r      s
    %**,r   )	pd	DataFrame	set_indextmassert_produces_warningDeprecationWarninggroupbyapplyassert_frame_equal)dfmsggrp_by_same_valuegrp_by_copys       r   test_group_by_copyr(      s    	,	


 i  DC		#	#$6c	BJJw5JAGG
 
C DC		#	#$6c	Bjj%Uj;AA&
 
C +[9 
C	B
 
C	Bs   "C"C 
C 
C.c            
         [         R                  " S/S-  S/S-  -   S/S-  S/S-  -   S/S-  -   S	/S-  -   S/S-  -   S/S-  -   S/S-  -   [        S
S5       V s/ s H  n SU  3PM
     sn [        R                  R                  S5      R                  SSS9S.5      nS nS nSn[        R                  " [        US9   UR                  S5      R                  U5      nS S S 5        [        R                  " [        US9   UR                  S5      R                  U5      nS S S 5        [        R                  " WW5        g s  sn f ! , (       d  f       Nm= f! , (       d  f       N<= f)Na   b   c   def      gd      )size)cat1cat2cat3valc                     U R                  5       n U R                  R                  SS9U S'   U R                  S5      S   R	                  5       $ Nmin)methodrankr:   )r   r<   rA   r!   r?   xs    r   f_copy"test_mutate_groups.<locals>.f_copy/   sB    FFHEEJJeJ,&	yy (,,..r   c                 ~    U R                   R                  SS9U S'   U R                  S5      S   R                  5       $ r>   )r<   rA   r!   r?   rB   s    r   	f_no_copy%test_mutate_groups.<locals>.f_no_copy4   s7    EEJJeJ,&	yy (,,..r   r   r   r9   )r   r   rangenprandomdefault_rngintegersr   r   r    r!   r"   assert_series_equal)rC   r$   rD   rG   r%   
grpby_copygrpby_no_copys          r   test_mutate_groupsrQ      sj    
EAI	)EAIeaieai eai eai	
 eai eai ',Arl3lqWl399((+44Sr4B	

B/
/ DC		#	#$6c	BZZ'--f5
 
C		#	#$6c	B

6*00; 
C:}5' 4 
C	B	B	Bs   E?!E!E
E
E&c                     [         R                  " / SQ[        S5      S.5      n Sn[        R                  " [
        US9   U R                  SSS9R                  S	 5      nS S S 5        [        R                  " [
        US9   U R                  SSS9R                  S
 5      nS S S 5        [        R                  " WW5        g ! , (       d  f       Nh= f! , (       d  f       N7= f)N)	r3   r3   r3   r/   r/   r/      rS   rS   	   )keyvaluer   r   rU   Tr   c                      U S S  R                   $ r   rU   rB   s    r   r   /test_no_mutate_but_looks_like.<locals>.<lambda>H   s    QqTXXr   c                     U R                   $ r   rX   rB   s    r   r   rY   J   s    QUUr   )	r   r   rI   r   r   r    r!   r"   rN   )r$   r%   result1result2s       r   test_no_mutate_but_looks_liker]   @   s     
9E!HM	NB
CC		#	#$6c	B**Ut*4::;MN 
C		#	#$6c	B**Ut*4::?K 
C7G,	 
C	B	B	Bs    !C!C
C
C"c                    [         R                  " / SQ/ SQS.5      nS nSn[        R                  " [        X0(       + S9   UR                  S/SS	9R                  U5      nS S S 5        [         R                  " / S
Q[         R                  R                  / SQ5      SS9n[        R                  " WU5        g ! , (       d  f       N]= f)N)Ar_   r_   Br`   r`   )r3   r/   rS         r-   )col1col2c                 V    SU R                   U R                  S   S4'   U R                  $ )Nr   rd   )locindexrd   rB   s    r   fn-test_apply_function_with_indexing.<locals>.fnT   s&    %&aggbk6!"vvr   r   )r   raise_on_extra_warningsrc   F)as_index)r3   r/   r   ra   rb   r   ))r   r   )r   r3   )r   r/   )r3   rS   )r3   ra   )r3   rb   rd   )rh   r	   )r   r   r   r   r    r!   r"   Series
MultiIndexfrom_tuplesrN   )warn_copy_on_writer$   ri   r%   resultexpecteds         r   !test_apply_function_with_indexingrs   N   s    	/9KL
B DC		#	##?U
 VHu5;;B?
 yymm''<
 H 68,
 
s   !B99
Cc            	         [         R                  " / SQ/ SQ/ SQ/ SQ/ SQ/ SQS.[         R                  R                  / SQ5      S9n S nSn[        R
                  " [        US9   U R                  SSS	9nS S S 5        WR                  U5      n[         R                  " / S
Q/ SQ/ SQ/[         R                  R                  / SQ5      S9n[        R                  " XE5        g ! , (       d  f       Ns= f)N)r3   r/   rS   )Cjulianr`   geoffreyr_   rw   r`   rw   r_   ry   rv   ry   )rz   r|   r{   rx   ru   r}   )columnsc                 R    U R                   S   S   nU R                  SS9U SU4'   U $ )Nr   r3   )axissum)r~   r   )groupedr	   s     r   
add_column8test_apply_mutate_columns_multiindex.<locals>.add_column~   s3    q!!$&{{{2tr   z+DataFrame.groupby with axis=1 is deprecatedr   r3   )levelr   )r3   r3   r3   rS   r3   r3   r3   rS   )r/   r/   r/   r-   r/   r/   r/   r-   )rS   rS   rS   rT   rS   rS   rS   rT   ))ry   r_   ry   )ry   r`   ry   )ry   rv   ry   )ry   r   ry   )rw   r_   rw   )rw   r`   rw   )rw   rv   rw   )rw   r   rw   )
r   r   rn   ro   r   r   FutureWarningr!   r"   r#   )r$   r   r%   gbrq   rr   s         r   $test_apply_mutate_columns_multiindexr   g   s    	&(&&((	
 ))	

B*
 8C		#	#M	=ZZaaZ( 
>XXj!F||$$		
 ))	
H8 &+? 
>	=s   &C$$
C2)numpyrJ   pandasr   pandas._testing_testingr   r(   rQ   r]   rs   r   r   r   r   <module>r      s(      :, 6F--2<,r   