ó
    ­œ‘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    ÚjC:\Users\psant\Downloads\Noiseback\school\venv\Lib\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ÚFutureWarningÚgroupbyÚapplyÚassert_frame_equal)ÚdfÚmsgÚgrp_by_same_valueÚgrp_by_copys       r   Útest_group_by_copyr(      sÎ   € ä	�Šâ,Úñ	
ó
÷
 �i�Óð ð D€CÜ	×	#Ò	#¤M¸Ó	=ØŸJ™J¨ w¸5˜JÐA×GÑGÙó
Ð÷ 
>ð D€CÜ	×	#Ò	#¤M¸Ó	=Ø—j‘j % °U�jÐ;×AÑAÙ&ó
ˆ÷ 
>ô ×ÒÐ+¨[Õ9÷ 
>Õ	=ú÷
 
>Õ	=ús   Á"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é   é   Úgéd   é   )Ú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   € Ø�F‰F‹HˆØ—E‘E—J‘J e�JÐ,ˆˆ&‰	Ø�y‰y˜Ó  Ñ(×,Ñ,Ó.Ð.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   € Ø—E‘E—J‘J e�JÐ,ˆˆ&‰	Ø�y‰y˜Ó  Ñ(×,Ñ,Ó.Ð.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      se  € ô 
�Šà�E˜A‘I  ¨¡	Ñ)Ø�E˜A‘IØˆe�a‰iñàˆe�a‰iñð ˆe�a‰iñð ˆe�a‰iñ	ð
 ˆe�a‰iñð ˆe�a‰iñô ',¨A¨r¤lÓ3¢l �q˜˜“W¡lÑ3Ü—9‘9×(Ñ(¨Ó+×4Ñ4°S¸rÐ4ÐBñ	
ó
€Bò/ò
/ð D€CÜ	×	#Ò	#¤M¸Ó	=Ø—Z‘Z Ó'×-Ñ-¨fÓ5ˆ
÷ 
>ä	×	#Ò	#¤M¸Ó	=ØŸ
™
 6Ó*×0Ñ0°Ó;ˆ÷ 
>ä×Ò˜: }Õ5ùò' 4÷ 
>Õ	=úç	=Õ	=ús   Á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   € ÀQÁqÀTÇXÂXr   c                 ó   • U R                   $ r   rX   rB   s    r   r   rY   J   s   € ÀQÇUÂUr   )	r   r   rI   r   r   r    r!   r"   rN   )r$   r%   Úresult1Úresult2s       r   Útest_no_mutate_but_looks_liker]   @   s¦   € ô 
�ŠÒ9ÄEÈ!ÃHÑMÓ	N€Bà
C€CÜ	×	#Ò	#¤M¸Ó	=Ø—*‘*˜U¨t�*Ð4×:Ñ:Ñ;MÓNˆ÷ 
>ä	×	#Ò	#¤M¸Ó	=Ø—*‘*˜U¨t�*Ð4×:Ñ:¹?ÓKˆ÷ 
>ä×Ò˜7 GÕ,÷	 
>Õ	=úç	=Õ	=ús   Á !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&   € Ø%&ˆ�‰ˆa�g‰g�b‰k˜6Ð!Ñ"Ø�v‰vˆ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Ü	×	#Ò	#Ü˜SÔ:Pó
ð —‘˜V˜H¨u�Ð5×;Ñ;¸BÓ?ˆ÷
ô �yŠyÚÜ�m‰m×'Ñ'Ú<ó
ð ñ€Hô ×Ò˜6 8Õ,÷
õ 
ú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_columnÚ8test_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   r    r!   r"   r#   )r$   rƒ   r%   Úgbrq   rr   s         r   Ú$test_apply_mutate_columns_multiindexr‡   g   sÑ   € ä	�Šâ&Ú(Ú&Ú&Ú(Ú(ñ	
ô —‘×)Ñ)òó	
ñ
€Bò*ð
 8€CÜ	×	#Ò	#¤M¸Ó	=Ø�Z‰Z˜a aˆZÐ(ˆ÷ 
>à�X‰X�jÓ!€FÜ�|Š|â$Ú$ò	ð	
ô —‘×)Ñ)ò	ó
ñ€Hô8 ×Ò˜&Õ+÷? 
>Õ	=ús   Á&C$Ã$
C2)ÚnumpyrJ   Úpandasr   Úpandas._testingÚ_testingr   r(   rQ   r]   rs   r‡   r   r   r   Ú<module>rŒ      s(   ðÛ ã Ý ò:ò, 6òF-ò-ó2<,r   