ó
    wñ:i®/  ã                   ól   • S SK r S SKrS SKrS SKrS SKJrJrJr  S SK	J
r   " S S5      r " S S5      rg)é    N)Ú	DataFrameÚ
MultiIndexÚSeriesc                   ó¼   • \ rS rSrS rS rS rS rS rS r	S r
S	 rS
 rS rS rS r\R"                  R%                  SSS/5      S 5       rS rS rS rS rSrg)ÚTestMultiLevelé   c                 óö  • UnUR                  S5      R                  5       nUR                  UR                  SS9nUR                  SS9R	                  S5      n[
        R                  " XE5        US   R                  UR                  SS9nUS   R                  SS9R	                  S5      n[
        R                  " XESS9  Sn[
        R                  " [        US	9   UR                  R                  SSS
9nS S S 5        WR                  5       nUR                  UR                  SS9nUR                  SS9R	                  S5      R                  n[
        R                  " XE5        g ! , (       d  f       Nw= f)NÚmonthé   ©ÚlevelÚsumÚAF©Úcheck_namesú+DataFrame.groupby with axis=1 is deprecated©Úmatch©Úaxis)Úcolumnsr   )Úgroupbyr   ÚreindexÚindexÚ	transformÚtmÚassert_frame_equalÚassert_series_equalÚassert_produces_warningÚFutureWarningÚT)ÚselfÚ/multiindex_year_month_day_dataframe_random_dataÚymdÚ
month_sumsÚresultÚexpectedÚmsgÚgbs           Ú_/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/pandas/tests/test_multilevel.pyÚtest_reindex_levelÚ!TestMultiLevel.test_reindex_level   sC  € à=ˆà—[‘[ Ó)×-Ñ-Ó/ˆ
Ø×#Ñ# C§I¡I°QÐ#Ð7ˆØ—;‘; W�;Ð-×7Ñ7¸Ó>ˆä
×Ò˜fÔ/ð ˜C‘×(Ñ(¨¯©¸!Ð(Ð<ˆØ�s‘8×#Ñ#¨'Ð#Ð2×<Ñ<¸UÓCˆÜ
×Ò˜v¸UÒCð <ˆÜ×'Ò'¬¸SÓAØ—‘—‘˜w¨Q�Ð/ˆB÷ Bð —V‘V“Xˆ
Ø×#Ñ#¨C¯I©I¸QÐ#Ð?ˆØ—;‘; W�;Ð-×7Ñ7¸Ó>×@Ñ@ˆÜ
×Ò˜fÕ/÷ BÕAús   ÃE*Å*
E8c                 óx   • UnUR                   SS/   nUR                  SS/   n[        R                  " XC5        g )Nr   é   ©ÚfooÚone©Úbarr1   )ÚilocÚlocr   r   )r"   Ú multiindex_dataframe_random_dataÚframer'   Ú	reindexeds        r*   Útest_reindexÚTestMultiLevel.test_reindex)   s:   € Ø0ˆà—:‘:˜q !˜fÑ%ˆØ—I‘I˜~¨~Ð>Ñ?ˆ	Ü
×Ò˜iÕ2ó    c                 ó<  • UnUR                   S S S2   nUR                  U5      nU(       a#  UR                   R                  U5      (       d   eOUR                   UL d   eUR                  U   nUR                   R	                  U5      (       d   eUR
                  nUR                  US9nU(       a#  UR                  R                  U5      (       d   eOUR                  UL d   eUR                  S S 2U4   nUR                  R	                  U5      (       d   eg )Né
   ©r   )r   r   Úis_r5   Úequalsr!   r   )r"   r#   Úusing_copy_on_writer$   Ú	new_indexÚchunkÚymdTs          r*   Útest_reindex_preserve_levelsÚ+TestMultiLevel.test_reindex_preserve_levels0   sõ   € ð >ˆà—I‘I™d ˜d‘Oˆ	Ø—‘˜IÓ&ˆÞØ—;‘;—?‘? 9×-Ñ-Ð-Ñ-à—;‘; )Ò+Ð+Ð+à—‘˜	Ñ"ˆØ�{‰{×!Ñ! )×,Ñ,Ð,Ð,à�u‰uˆØ—‘ Y�Ð/ˆÞØ—=‘=×$Ñ$ Y×/Ñ/Ð/Ñ/à—=‘= IÒ-Ð-Ð-à—‘š˜I˜Ñ&ˆØ�}‰}×#Ñ# I×.Ñ.Ð.Ñ.r;   c                 ó  • UnUS   nUR                   R                  S5      nUR                  USS9nUR                  S 5      nUR	                  S 5      nUR                  UR                   5      n[        R                  " X‡SS9  g )Nr   r   F)Ú
group_keysc                 ó   • U S-  $ ©Né   © ©Úxs    r*   Ú<lambda>Ú7TestMultiLevel.test_groupby_transform.<locals>.<lambda>Q   s   € ¨!¨aª%r;   c                 ó   • U S-  $ rJ   rL   rM   s    r*   rO   rP   R   s   € ¨q°1ªur;   r   )r   Úget_level_valuesr   Úapplyr   r   r   r   )	r"   r6   r7   ÚsÚgrouperÚgroupedÚappliedr'   r&   s	            r*   Útest_groupby_transformÚ%TestMultiLevel.test_groupby_transformI   sv   € Ø0ˆà�#‰JˆØ—'‘'×*Ñ*¨1Ó-ˆà—)‘)˜G°�)Ð6ˆà—-‘-¡Ó0ˆØ×$Ñ$¡_Ó5ˆØ—‘ §¡Ó0ˆÜ
×Ò˜v¸UÓCr;   c                 óÈ   • [        S/S/S//S/S/S/// SQS9n[        [        R                  R	                  S5      R                  S5      // S	QUS
9nUR                  SS9  g )Nr0   r3   Úbazr   )r1   ÚtwoÚthree©ÚlevelsÚcodesÚnamesrK   é   ©ÚaÚbÚcÚd)r   r   r]   r   )r   r   ÚnpÚrandomÚdefault_rngr   )r"   ÚmidxÚdfs      r*   Útest_groupby_cornerÚ"TestMultiLevel.test_groupby_cornerV   sp   € ÜØ�G˜e˜W u gÐ.Ø�3˜˜˜a˜S�/Ú)ñ
ˆô
 Ü�Y‰Y×"Ñ" 1Ó%×,Ñ,¨QÓ/Ð0Ú(Øñ
ˆð 	�
‰
˜ˆ
Ò!r;   c                 ó¢  • [         R                  " / SQ5      n[        / SQ/ SQ/US9nUR                  SS9UR                  R                  S 5         nSn[        R                  " [        US	9   UR                  SS
S9nS S S 5        WR                  5       nUR                  SS/:H  R                  5       (       d   eg ! , (       d  f       NE= f)N))Úf1Ús1)rp   Ús2)Úf2rq   )rs   rr   )Úf3rq   )rt   rr   )r   rK   r.   rb   é   é   )é   é   é	   r=   é   é   r>   r   r   c                 ó   • U S   S;   $ )Nr   )rs   rt   rL   )Úus    r*   rO   Ú:TestMultiLevel.test_groupby_level_no_obs.<locals>.<lambda>q   s   € °a¸±d¸lÒ6Jr;   r   r   r   ©r   r   rs   rt   )r   Úfrom_tuplesr   r5   r   Úmapr   r   r    r   r   Úall)r"   rk   rl   Údf1r(   rV   r&   s          r*   Útest_groupby_level_no_obsÚ(TestMultiLevel.test_groupby_level_no_obsd   s²   € ä×%Ò%òó	
ˆô Ò*Ò,AÐBÈDÑQˆØ�f‰f˜!ˆfˆn˜RŸZ™ZŸ^™^Ñ,JÓKÑLˆà;ˆÜ×'Ò'¬¸SÓAØ—k‘k q°�kÐ2ˆG÷ Bà—‘“ˆØ—‘ 4¨ ,Ñ.×3Ñ3×5Ñ5Ð5Ñ5÷ BÕAús   Á0C Ã 
Cc                 ó¸   • UnUS S R                   nUS   US'   [        UR                  [        5      (       d   eUS   US   :H  R	                  5       (       d   eg )Nru   )éÐ  r   rw   )r‡   r   r=   )r!   Ú
isinstancer   r   r‚   )r"   r#   r$   rl   s       r*   Ú.test_setitem_with_expansion_multiindex_columnsÚ=TestMultiLevel.test_setitem_with_expansion_multiindex_columnsy   s^   € ð >ˆà��!ˆW�Y‰YˆØ˜Z™.ˆˆ;‰Ü˜"Ÿ*™*¤j×1Ñ1Ð1Ð1Ø�;‘ 2 j¡>Ñ1×6Ñ6×8Ñ8Ð8Ñ8r;   c                 ó.  • [        / SQ[        R                  " / SQ5      S9n[        / SQ[        R                  " / SQ5      S9nX-
  nUR                  R	                  UR                  5      nUR                  U5      UR                  U5      -
  n[        R                  " X55        US S S2   US S S2   -
  nUR                  R	                  UR                  5      nUR                  U5      UR                  U5      -
  n[        R                  " X55        g )N)r   rK   r.   ))r   r   )r   rK   ©ÚBr.   ©Údatar   )rb   ru   rv   ))ÚZr   )r�   rK   rŒ   éÿÿÿÿ)r   r   r€   r   Úunionr   r   r   )r"   rN   ÚyÚresÚ	exp_indexÚexps         r*   Útest_alignmentÚTestMultiLevel.test_alignmentƒ   sß   € ÜÚ¤*×"8Ò"8Ò9WÓ"Xñ
ˆô Ú¤*×"8Ò"8Ò9WÓ"Xñ
ˆð ‰eˆØ—G‘G—M‘M !§'¡'Ó*ˆ	Ø�i‰i˜	Ó" Q§Y¡Y¨yÓ%9Ñ9ˆÜ
×Ò˜sÔ(ð ‘�"�‰g˜™$˜B˜$™ÑˆØ—G‘G—M‘M !§'¡'Ó*ˆ	Ø�i‰i˜	Ó" Q§Y¡Y¨yÓ%9Ñ9ˆÜ
×Ò˜sÕ(r;   c                 ó  • UnUR                  SS/S9R                  5       nUR                  R                  S5      nUR                  R                  S5      nUR                  XE/5      R                  5       n[        R
                  " X6SS9  UR                  R                  UR                  R                  S S :X  d   eUR                  UR                  R                  S S S9R                  5       n[        R
                  " X75        g )Nr   r   r   Fr   rK   )r   Úmeanr   rR   r   r   ra   )r"   r#   r$   r&   Úk1Úk2r'   Úresult2s           r*   Útest_groupby_multilevelÚ&TestMultiLevel.test_groupby_multilevel—   sÔ   € Ø=ˆà—‘ A q 6�Ð*×/Ñ/Ó1ˆà�Y‰Y×'Ñ'¨Ó*ˆØ�Y‰Y×'Ñ'¨Ó*ˆà—;‘; ˜xÓ(×-Ñ-Ó/ˆô 	×Ò˜f¸EÒBØ�|‰|×!Ñ! S§Y¡Y§_¡_°R°aÐ%8Ó8Ð8Ð8à—+‘+ C§I¡I§O¡O°B°QÐ$7�+Ð8×=Ñ=Ó?ˆÜ
×Ò˜fÕ.r;   c                 óè   • [         R                  " / SQ5      n[        [        R                  R                  S5      R                  S5      XS9nUR                  S5      US'   UR                  5       ng )N)r/   )r0   r\   r2   )r3   r\   rK   )rb   rb   )r   r   r   )ÚTotalsÚ )	r   r€   r   rh   ri   rj   Ústandard_normalr   Ú_consolidate)r"   r   rl   s      r*   Útest_multilevel_consolidateÚ*TestMultiLevel.test_multilevel_consolidate¨   s`   € Ü×&Ò&ÚLó
ˆô Ü�I‰I×!Ñ! !Ó$×4Ñ4°VÓ<ÀEñ
ˆð Ÿ6™6 !›9ˆˆ<ÑØ�_‰_Ó‰r;   c                 óÖ  • [        / SQSS/// SQ/ SQ/S9n[        [        R                  R	                  S5      R                  S5      US	9n[        [        R                  R	                  S5      R                  S
5      US	9nUS   nUR                  S   nUS S nUR                  R                  S5      Ul        [        R                  " XF5        [        R                  " XV5        [        R                  " [        SS9   US     S S S 5        UR                  S   nUR                  S5      nUS S nUR                  R                  S5      Ul        [        R                   " XF5        [        R                   " XV5        [        / SQSS/// SQ/ SQ/S9n[        [        R                  R	                  S5      R                  S5      US	9n[        [        R                  R	                  S5      R                  S
5      US	9nUS   nUR                  S   nUS S nUR                  R                  S5      Ul        [        R                  " XF5        [        R                  " XV5        UR                  S   nUR                  S5      nUS S nUR                  R                  S5      Ul        [        R                   " XF5        [        R                   " XV5        g ! , (       d  f       GNÙ= f)N)©r0   r3   r   )r0   r[   r   )r0   Úquxr   r   r   )r   r   r   r   rK   rK   )r   r   r   r   r   r   )r_   r`   rK   rv   ©r   )rv   rb   r¨   z^\(\('foo', 'bar', 0\), 2\)$r   )r¨   rK   )©r0   r3   )r0   r[   )r0   r©   r«   )r   r   rh   ri   rj   r£   r   r5   r   Ú	droplevelr   r   ÚpytestÚraisesÚKeyErrorÚxsr   )r"   r   Úseriesr7   r&   r�   r'   s          r*   Útest_level_with_tuplesÚ%TestMultiLevel.test_level_with_tuples²   sj  € ÜÚMÐPQÐSTÈvÐVÚ%Ò'9Ð:ñ
ˆô
 œŸ	™	×-Ñ-¨aÓ0×@Ñ@ÀÓCÈ5ÑQˆÜœ"Ÿ)™)×/Ñ/°Ó2×BÑBÀ6ÓJÐRWÑXˆàÐ)Ñ*ˆØ—*‘*Ð.Ñ/ˆØ˜"˜1�:ˆØ!Ÿ™×1Ñ1°!Ó4ˆŒÜ
×Ò˜vÔ0Ü
×Ò˜wÔ1ä�]Š]œ8Ð+JÓKØÐ'Ò(÷ Lð —‘Ð,Ñ-ˆØ—(‘(Ð,Ó-ˆØ˜˜!�9ˆØ!Ÿ™×1Ñ1°!Ó4ˆŒÜ
×Ò˜fÔ/Ü
×Ò˜gÔ0äÚDÀqÈ!ÀfÐMÚ%Ò'9Ð:ñ
ˆô
 œŸ	™	×-Ñ-¨aÓ0×@Ñ@ÀÓCÈ5ÑQˆÜœ"Ÿ)™)×/Ñ/°Ó2×BÑBÀ6ÓJÐRWÑXˆà˜Ñ'ˆØ—*‘*˜^Ñ,ˆØ˜"˜1�:ˆØ!Ÿ™×1Ñ1°!Ó4ˆŒÜ
×Ò˜vÔ0Ü
×Ò˜wÔ1à—‘˜>Ñ*ˆØ—(‘(˜>Ó*ˆØ˜˜!�9ˆØ!Ÿ™×1Ñ1°!Ó4ˆŒÜ
×Ò˜fÔ/Ü
×Ò˜gÕ0÷= LÖKús   ÄKË
K(c                 ó  • UnUR                  SS/SS9nUR                  / SQ   n[        R                  " X45        UR                  R                  SS/SSS9n[        R                  " X4R                  5        UR
                  SS/   n[        R                  " X45        US   R
                  SS/   n[        R                  " X4S   5        UR                  R
                  S S 2SS/4   n[        R                  " X4R                  5        g )	Nr0   r©   r   r   )r   r   rK   rw   rx   ry   r   r   r   )r   r4   r   r   r!   r5   r   )r"   r6   r7   r&   r'   s        r*   Ú$test_reindex_level_partial_selectionÚ3TestMultiLevel.test_reindex_level_partial_selectionâ   sÝ   € Ø0ˆà—‘  u˜~°Q�Ð7ˆØ—:‘:Ò0Ñ1ˆÜ
×Ò˜fÔ/à—‘—‘ %¨ °a¸q�ÐAˆÜ
×Ò˜f§j¡jÔ1à—‘˜E 5˜>Ñ*ˆÜ
×Ò˜fÔ/à�s‘—‘ ¨ Ñ/ˆÜ
×Ò˜v°¡}Ô5à—‘—‘šQ ¨ Ð.Ñ/ˆÜ
×Ò˜f§j¡jÕ1r;   rg   rb   c                 ó   • / SQn[        S/S/S/U/S.5      nX3R                  S:H     R                  U5      nUR                  5       R                  n[        S/S/S/// / / // SQS9n[        R                  " XV5        g )N)rd   re   rf   r   rK   r.   rc   r^   )r   rd   r   Úfirstr   r   r   Úassert_index_equal)r"   rg   rH   rl   Úgr&   r'   s          r*   Ú+test_empty_frame_groupby_dtypes_consistencyÚ:TestMultiLevel.test_empty_frame_groupby_dtypes_consistencyõ   s€   € ò %ˆ
Ü˜a˜S¨ s°!°¸A¸3Ñ?Ó@ˆà�t‰t�q‰y‰M×!Ñ! *Ó-ˆØ—‘“—‘ˆÜØ�C˜!˜˜q˜c�?¨2¨r°2¨,ºoñ
ˆô 	×Ò˜fÕ/r;   c                 óÈ   • / SQn/ SQn[         R                  " USS/S9n[        X#S9nUR                  UR                  5      R                  5       n[        U5      S:X  d   eg )N))Ú600809Ú20061231©r¾   Ú20070331)r¾   Ú20070630rÀ   )ÚdemorÃ   rÃ   rÃ   ÚSTK_IDÚRPT_Date)ra   rª   r.   )r   r€   r   r   r   r¸   Úlen)r"   Úidx_tpÚdtÚidxrT   r&   s         r*   Útest_duplicate_groupby_issuesÚ,TestMultiLevel.test_duplicate_groupby_issues  s^   € ò
ˆò .ˆä×$Ò$ V°H¸jÐ3IÑJˆÜ�2Ñ!ˆà—‘˜1Ÿ7™7Ó#×)Ñ)Ó+ˆÜ�6‹{˜aÓÐÑr;   c                 ó   • SS//nSS[         R                  4S/n[        U[        R                  " U5      S9nUR
                  R                  R                  nUR                  R                  R
                  n[        R                  " XT5        g )NrN   r   rd   re   )rd   rf   ç        r>   )
rh   Únanr   r   r€   Údtypesrd   re   r   r   )r"   r�   r   rl   r'   r&   s         r*   Útest_subsets_multiindex_dtypeÚ,TestMultiLevel.test_subsets_multiindex_dtype  si   € à�a�ˆzˆØ˜œbŸf™fÐ% Ð7ˆÜ�t¤Z×%;Ò%;¸GÓ%DÑEˆØ—9‘9—;‘;—=‘=ˆØ—‘—‘—‘ˆÜ
×Ò˜vÕ0r;   c                 ó.  • S[         R                  " SSS5      4SSS.S[         R                  " SSS5      4SS	S.S[         R                  " SSS5      4SS
S.S[         R                  " SSS5      4SSS.0n[        R                  " USS9n/ SQ/ SQS.n/ SQ[         R                  " SSS5      [         R                  " SSS5      [         R                  " SSS5      [         R                  " SSS5      //n[        X4S9n[        R
                  " X%5        g )Nr   iâ  r.   r   r=   )r   r�   rb   rK   rz   r{   é   r   )Úorient)r   rK   r.   rb   )r=   rz   r{   rÓ   )r   r   r   r   rŽ   )ÚdatetimeÚdater   Ú	from_dictr   r   )r"   Údata_dicr&   r�   r   r'   s         r*   Útest_datetime_object_multiindexÚ.TestMultiLevel.test_datetime_object_multiindex  s  € à”—’˜d A qÓ)Ð*°!¸"Ñ,=Ø”—’˜d A qÓ)Ð*°!¸"Ñ,=Ø”—’˜d A qÓ)Ð*°!¸"Ñ,=Ø”—’˜d A qÓ)Ð*°!¸"Ñ,=ð	
ˆô ×$Ò$ X°gÑ>ˆÚ!Ò(8Ñ9ˆâä—’˜d A qÓ)Ü—’˜d A qÓ)Ü—’˜d A qÓ)Ü—’˜d A qÓ)ð	ð
ˆô  $Ñ4ˆä
×Ò˜fÕ/r;   c           	      óL  • [        S[        R                  SS// SQ/ SQ// SQS9R                  SS	/5      nS
UR                  S'   [        S[        R                  SS// SQ/ SQSS[        R                  S
/// SQS9R                  SS	/5      n[
        R                  " X5        g )Nr   ç®Gáz®ó?ç=
×£p=@)r   ÚGrÜ   rÝ   )r   ÚDg=
×£p½#@g®Gáz%@)Úpivot_0Úpivot_1Úcol_1Úcol_2r>   rà   rá   rÍ   ))r   ÚFrã   rä   )r   rh   rÎ   Ú	set_indexÚatr   r   )r"   rl   r'   s      r*   Útest_multiindex_with_naÚ&TestMultiLevel.test_multiindex_with_na1  s¯   € Üà”b—f‘f˜d DÐ)Ú&Ú'ðò
 =ñ
÷ ‰)�Y 	Ð*Ó
+ð 	ð &)ˆ�‰Ð!Ñ"äà”b—f‘f˜d DÐ)Ú&Ú'Ø�cœ2Ÿ6™6 3Ð'ð	ò =ñ
÷ ‰)�Y 	Ð*Ó
+ð 	ô 	×Ò˜bÕ+r;   rL   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r+   r9   rE   rX   rm   r„   r‰   r—   rž   r¥   r²   rµ   r­   ÚmarkÚparametrizer»   rÊ   rÐ   rÙ   rç   Ú__static_attributes__rL   r;   r*   r   r      s}   † ò0ò23ò/ò2Dò"ò6ò*9ò)ò(/ò"ò.1ò`2ð& ‡[�[×Ñ˜S 1 c (Ó+ñ0ó ,ð0ò ò1ò0õ,,r;   r   c                   ó   • \ rS rSrSrS rSrg)Ú
TestSortediJ  z+everything you wanted to test about sortingc                 óÈ  • [        / SQ/ SQ// SQ/ SQ/5      n[        S[        [        U5      5      0USS9nUR                  R
                  SL d   eUR                  5       nUR                  R
                  S	L d   e[        S/ S
Q0[         R                  " / SQ5      SS9nUR                  [        R                  SS2SS24   S S 24   n[        R                  " XT5        g )N)r   r�   ÚC)rf   re   rd   )r   r   rK   r   r   rK   )r   rK   r   r   r   rK   ÚcolÚint64)r   ÚdtypeFT)r   rb   ru   rK   ))r�   rd   )r�   rf   )ró   rd   )ró   re   r�   ró   rd   rf   )r   r   ÚrangerÆ   r   Úis_monotonic_increasingÚ
sort_indexr€   r5   ÚpdÚ
IndexSlicer   r   )r"   rÉ   rl   Úsortedr'   r&   s         r*   Útest_sort_non_lexsortedÚ"TestSorted.test_sort_non_lexsortedM  sØ   € ô ÚšoÐ.Ò1CÒEWÐ0Xó
ˆô ˜œu¤S¨£X›Ð/°sÀ'ÑJˆØ�x‰x×/Ñ/°5Ò8Ð8Ð8à—‘“ˆØ�|‰|×3Ñ3°tÒ;Ð;Ð;äØ’LÐ!Ü×(Ò(Ú@óð ñ
ˆð —‘œBŸM™M¨#¨c¨'°3°s°7Ð*:Ñ;ºQÐ>Ñ?ˆÜ
×Ò˜fÕ/r;   rL   N)ré   rê   rë   rì   Ú__doc__rý   rï   rL   r;   r*   rñ   rñ   J  s
   † Ù5õ0r;   rñ   )rÕ   Únumpyrh   r­   Úpandasrú   r   r   r   Úpandas._testingÚ_testingr   r   rñ   rL   r;   r*   Ú<module>r     s7   ðÛ ã Û ã ÷ñ õ
 ÷x,ñ x,÷v	0ò 0r;   