Ë
    ž…jÂ#  ã                   óT   — d dl Zd dlZd dlZd dlmZmZmZmZ d dl	m
Z  G d„ d«      Zy)é    N)Ú	DataFrameÚIndexÚSeriesÚconcatc                   ó˜   — e Zd Zej                  j                  d¬«      d„ «       Zd„ Zd„ Zd„ Z	d„ Z
d„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Zy)ÚTestDataFrameConcatz(GH#62888 the `mi[2][1] is 1` check fails)Úreasonc                 ó’  — t        ddgdgddg¬«      }t        g d¢dgg d¢¬«      }t        ||d	œ«      }t        j                  j	                  d
gdz  dgdz  z   t        j                  g d¢t        ¬«      g«      }|d   d   du sJ ‚t        |d   d   «      t        u sJ ‚t        dg d¢i|¬«      }t        j                  ||«       y )Né{   éÈ  ÚdataTF)ÚcolumnsÚindex)é7   é×  éE   ép   r   )é   é   é   é   éc   )ÚOneÚTwor   r   r   é   )TFr   r   r   r   r   ©Údtyper   r   )r   r   r   r   r   r   r   ©r   )r   r   ÚpdÚ
MultiIndexÚfrom_arraysÚnpÚarrayÚobjectÚtypeÚintÚtmÚassert_frame_equal)ÚselfÚleftÚrightÚresultÚmiÚexpecteds         úk/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/reshape/concat/test_dataframe.pyÚ-test_concat_multiindex_level_bool_and_numericzATestDataFrameConcat.test_concat_multiindex_level_bool_and_numeric   sÓ   € ô ˜#˜s˜¨f¨X¸dÀE¸]ÔKˆÜÚ!¨F¨8Ò;Kô
ˆô  ¨UÑ3Ó4ˆô �]‰]×&Ñ&à�˜!‘˜u˜g¨™kÑ)Ü—‘Ò6¼fÔEðó
ˆð �!‰u�Q‰x˜4ÑÐÐÜ�B�q‘E˜!‘H‹~¤Ñ$Ð$Ð$Ü˜fÒ&EÐFÈbÔQˆÜ
×Ñ˜f hÕ/ó    c                 ó   — t        t        j                  d«      ddgt        j                  ¬«      }t        t        j                  d«      t        j                  ¬«      }t        ||fd¬«      j                  }t        t        j                  d«      gd	z  t        j                  d
«      gd	z  z   g d¢¬«      }t        j                  ||«       y )N)é
   r   ÚfooÚbar)r   r   r   )r   r   r   ©ÚaxisÚfloat64r   Úfloat32)r4   r5   r   r   r   )r   r"   Úonesr8   r9   r   Údtypesr   r   r'   Úassert_series_equal©r)   Údf1Údf2Úresultsr.   s        r/   Ú"test_concat_multiple_frames_dtypesz6TestDataFrameConcat.test_concat_multiple_frames_dtypes%   s™   € äœRŸW™W WÓ-¸¸u°~ÌRÏZÉZÔXˆÜœRŸW™W WÓ-´R·Z±ZÔ@ˆÜ˜#˜s˜¨!Ô,×3Ñ3ˆÜÜ�X‰X�iÓ Ð! AÑ%¬¯©°)Ó)<Ð(=ÀÑ(AÑAÚ&ô
ˆô 	×Ñ˜w¨Õ1r1   c           	      ó0  — t        t        j                  d«      t        d«      ¬«      }t        t        j                  d«      dz  t        d«      ¬«      }t	        ||fddg¬«      }t        d	d	d
d
d
dœd	d	d
d
d
dœdœ«      }t        j                  ||«       y )N)r   r   ÚAB©r   )r   r   r   )ÚbeeÚbah)rE   Úboo©Úkeysç      ð?g       @))rE   rF   r   )rE   rF   r   )rE   rG   r   )rE   rG   r   )rE   rG   r   )ÚAÚB)r   r"   r:   Úlistr   r'   r(   r=   s        r/   Útest_concat_tuple_keysz*TestDataFrameConcat.test_concat_tuple_keys0   s™   € äœŸ™ ›´°d³Ô<ˆÜœŸ™ ›¨!Ñ+´T¸$³ZÔ@ˆÜ˜#˜s˜¨>¸>Ð*JÔKˆÜð (+Ø'*Ø'*Ø'*Ø'*ñð (+Ø'*Ø'*Ø'*Ø'*ññó
ˆô$ 	×Ñ˜g xÕ0r1   c                 ó2  — t        ddgddgdœ«      }t        ddgd¬	«      }t        ||g|¬
«      }t        g d¢g d¢dœt        j                  j                  ddgddgfdd g¬«      ¬«      }t        j                  ||«       t        ddgd ¬	«      }t        ||g|dg¬«      }t        j                  ||«       t        ||g|¬
«      }t        g d¢g d¢dœt        j                  j                  ddgddgfd d g¬«      ¬«      }t        j                  ||«       y )Nr   r   çš™™™™™¹?çš™™™™™É?)r4   r5   ÚaÚbÚbaz©ÚnamerH   )r   r   r   r   )rP   rQ   rP   rQ   r   ©Únamesr   ©rI   rX   )r   r   r   r   r    Úfrom_productr'   r(   )	r)   Údfr   Úconcatted_named_from_keysÚexpected_namedÚindex_no_nameÚconcatted_named_from_namesÚconcatted_unnamedÚexpected_unnameds	            r/   Útest_concat_named_keysz*TestDataFrameConcat.test_concat_named_keysI   s#  € ä  1˜v¨s°C¨jÑ9Ó:ˆÜ�s˜C�j uÔ-ˆÜ$*¨B°¨8¸%Ô$@Ð!Ü"Ú Ò)=Ñ>Ü—-‘-×,Ñ,¨s°C¨j¸1¸a¸&Ð-AÈ%ÐQUÈÐ,ÓWô
ˆô 	×ÑÐ7¸ÔHä˜s C˜j¨tÔ4ˆÜ%+¨R°¨H¸=ÐQVÐPWÔ%XÐ"Ü
×ÑÐ8¸.ÔIä" B¨ 8°-Ô@ÐÜ$Ú Ò)=Ñ>Ü—-‘-×,Ñ,¨s°C¨j¸1¸a¸&Ð-AÈ$ÐPTÈÐ,ÓVô
Ðô 	×ÑÐ/Ð1AÕBr1   c                 óª  — t        dddgit        d«      ¬«      }t        dddgit        d«      ¬«      }t        dg d¢ig d	¢¬«      }t        ||gd
¬«      }t        j                  ||«       t        ||gd¬«      }t        j                  ||«       t        ||gd¬«      }t        j                  ||«       t        ddgddggddgddg¬«      }t        ||gd¬«      }t        j                  ||«       t        ||gd¬«      }	t        j                  |	|«       t        ddg«      }
t        ddg«      }t        g d¢g d	¢¬«      }t        |
|gd
¬«      }t        j                  ||«       t        |
|gd¬«      }t        j                  ||«       t        |
|gd¬«      }t        j                  ||«       t        ddgddggddgddg¬«      }t        |
|gd¬«      }t        j                  ||«       t        |
|gd¬«      }t        j                  ||«       t        j                  t        d¬«      5  t        |
|gd¬«       d d d «       y # 1 sw Y   y xY w)NrK   rP   rQ   r   r   ç333333Ó?çš™™™™™Ù?)rP   rQ   rd   re   )r   r   r   r   r   r6   Úrowsr   r   )r   r   r   zNo axis named)ÚmatchÚ	something)
r   Úranger   r'   r(   r   r<   ÚpytestÚraisesÚ
ValueError)r)   r>   r?   Úexpected_indexÚconcatted_indexÚconcatted_rowÚconcatted_0Úexpected_columnsÚconcatted_columnsÚconcatted_1Úseries1Úseries2Úexpected_index_seriesÚconcatted_index_seriesÚconcatted_row_seriesÚconcatted_0_seriesÚexpected_columns_seriesÚconcatted_columns_seriesÚconcatted_1_seriess                      r/   Útest_concat_axis_parameterz.TestDataFrameConcat.test_concat_axis_parameter_   sV  € ä˜˜s C˜jÐ)´°q³Ô:ˆÜ˜˜s C˜jÐ)´°q³Ô:ˆô # CÒ)=Ð#>ÂlÔSˆä  # s °'Ô:ˆÜ
×Ñ˜o¨~Ô>ä  S˜z°Ô7ˆÜ
×Ñ˜m¨^Ô<ä˜c 3˜Z¨aÔ0ˆÜ
×Ñ˜k¨>Ô:ô %Ø�3ˆZ˜#˜s˜Ð$¨Q°¨F¸SÀ#¸Jô
Ðô # C¨ :°IÔ>ÐÜ
×ÑÐ/Ð1AÔBä˜c 3˜Z¨aÔ0ˆÜ
×Ñ˜kÐ+;Ô<ä˜#˜s˜Ó$ˆÜ˜#˜s˜Ó$ˆô !'Ò';Â<Ô PÐä!'¨°'Ð(:ÀÔ!IÐÜ
×ÑÐ5Ð7LÔMä% w°Ð&8¸vÔFÐÜ
×ÑÐ3Ð5JÔKä# W¨gÐ$6¸QÔ?ÐÜ
×ÑÐ1Ð3HÔIô #,Ø�3ˆZ˜#˜s˜Ð$¨Q°¨F¸QÀ¸Fô#
Ðô $*¨7°GÐ*<À9Ô#MÐ Ü
×ÑÐ6Ð8OÔPä# W¨gÐ$6¸QÔ?ÐÜ
×ÑÐ0Ð2IÔJô �]‰]œ:¨_Ô=ñ 	9Ü�G˜WÐ%¨KÕ8÷	9÷ 	9ñ 	9ús   È0I	É	Ic           	      ó€  — t        dt        d«      idt        j                  j	                  g d¢g d¢gddg¬«      ¬	«      }t        |j                  d d…d d …f   |j                  d
d …d d …f   f«      }t        dg d¢idt        j                  j                  g d¢ddg¬«      ¬	«      }t        j                  ||«       y )NÚcolé	   Úint32)ÚA0ÚA1ÚA2)ÚB0ÚB1ÚB2r   r   rW   )r   r   éþÿÿÿ)r   r   é   é   ))r‚   r…   )r‚   r†   )r„   r†   )r„   r‡   )
r   ri   r   r    rZ   r   ÚilocÚfrom_tuplesr'   r(   ©r)   r[   r,   r.   s       r/   Útest_concat_numerical_namesz/TestDataFrameConcat.test_concat_numerical_names™   sº   € äØ”E˜!“HÐØä—‘×*Ñ*Ú'Ò);Ð<ÀQÈÀFð +ó ô	
ˆô ˜Ÿ™  ! ¢Q ™¨¯©°±²a°©Ð9Ó:ˆÜØ’LÐ!ØÜ—-‘-×+Ñ+ÚHÐQRÐTUÐPVð ,ó ô
ˆô 	×Ñ˜f hÕ/r1   c                 ó  — t        ddig«      }t        ||gd¬«      }|j                  d«      }t        t        j                  ddg«      j                  dd«      ddg¬«      j                  d«      }t        j                  ||«       y )NrR   rS   r   r6   Úcategoryr   rD   )r   r   Úastyper"   r#   Úreshaper'   r(   r�   s       r/   Útest_concat_astype_dup_colz.TestDataFrameConcat.test_concat_astype_dup_col®   s|   € ä˜˜c˜
�|Ó$ˆÜ�R˜�H 1Ô%ˆà—‘˜:Ó&ˆÜÜ�H‰H�c˜3�ZÓ ×(Ñ(¨¨AÓ.¸¸c¸
ô
ç
‰&�Ó
ð 	ô 	×Ñ˜f hÕ/r1   c           
      óþ   — t        dt        g d¢t        g d¢d¬«      ¬«      i«      }t        dt        ddgt        d	d
gd¬«      ¬«      i«      }t        ||gdddg|¬«      }t	        |j
                  «      ddgk(  sJ ‚y )NÚvalue©r   r   r   )rR   rS   ÚcÚidrU   r   r‰   rŠ   rR   rS   r   Út1Út2)r7   rI   Úsort)r™   r•   )rš   r•   )r   r   r   r   rM   r   )r)   r›   r™   rš   r,   s        r/   Útest_concat_dataframe_keys_bugz2TestDataFrameConcat.test_concat_dataframe_keys_bug¹   s   € ÜØ”fšY¬e²OÈ$Ô.OÔPÐQó
ˆô ˜¤¨¨A¨´e¸SÀ#¸JÈTÔ6RÔ!SÐTÓUˆô ˜˜R˜ q°°d¨|À$ÔGˆÜ�F—N‘NÓ#¨¸Ð'IÒIÐIÑIr1   c                 óâ   — t        t        g d¢d¬«      «      }t        t        g d¢d¬«      «      }t        ||g«      }t        |j                  d«      |g«      }t	        j
                  ||«       y )N)TFTTÚboolr   )r   r   r   Úint64)r   r   r   r‘   r'   r(   ©r)   r>   r?   r,   r.   s        r/   Útest_concat_bool_with_intz-TestDataFrameConcat.test_concat_bool_with_intÃ   s]   € ô œÒ8ÀÔGÓHˆÜœšy°Ô8Ó9ˆä˜˜c˜
Ó#ˆÜ˜3Ÿ:™: gÓ.°Ð4Ó5ˆÜ
×Ñ˜f hÕ/r1   c                 óX  — g d¢}g d¢}t        ||¬«      }t        |gdgddg¬«      }t        j                  j	                  dg|gddg¬«      }t        ||¬«      }t        j                  ||«       t        j                  |j                  j                  d	   t        d	d
gd¬«      «       y )N)r   r   r   r–   )r   r   rK   ÚIDÚdaterY   rW   r   r   rU   )r   r   r   r    rZ   r'   r(   Úassert_index_equalr   Úlevelsr   )r)   r   r   r[   r,   r-   r.   s          r/   Ú)test_concat_duplicates_in_index_with_keysz=TestDataFrameConcat.test_concat_duplicates_in_index_with_keysÍ   s—   € âˆÚˆä˜D¨Ô.ˆÜ˜˜ C 5°°v°Ô?ˆÜ�]‰]×'Ñ'¨#¨°¨¸tÀV¸nÐ'ÓMˆÜ $¨bÔ1ˆÜ
×Ñ˜f hÔ/Ü
×Ñ˜fŸl™l×1Ñ1°!Ñ4´e¸QÀ¸FÈÔ6PÕQr1   c                 óò   — t        dgdgddœ«      }t        ddgi«      }t        ||gddd¬«      }t        d	t        j                  gddgd	t        j                  gd
œ«      }t	        j
                  ||«       y )Nr   r   ©rK   rL   r   rK   éd   TÚouter©Úignore_indexÚjoinr›   rJ   )r   rK   rL   )r   r   r"   Únanr'   r(   r    s        r/   Útest_outer_sort_columnsz+TestDataFrameConcat.test_outer_sort_columnsÙ   sp   € ä˜q˜c¨¨°Ñ2Ó3ˆÜ˜˜s˜e˜Ó%ˆÜ˜˜c˜
°¸GÈ$ÔOˆÜ #¤r§v¡v °a¸°XÀSÌ"Ï&É&ÀMÑRÓSˆÜ
×Ñ˜f hÕ/r1   c                 ó¶   — t        dgdgddœ«      }t        dgddœ«      }t        ||gddd¬	«      }t        ddgddgd
œ«      }t        j                  ||«       y )Nr   r   r©   rª   r   ©rK   r   TÚinnerr¬   ©r   rK   ©r   r   r'   r(   r    s        r/   Útest_inner_sort_columnsz+TestDataFrameConcat.test_inner_sort_columnsá   sc   € ä˜q˜c¨¨°Ñ2Ó3ˆÜ˜s˜e¨Ñ*Ó+ˆÜ˜˜c˜
°¸GÈ$ÔOˆÜ ! Q ¨q°#¨hÑ7Ó8ˆÜ
×Ñ˜f hÕ/r1   c                 óŽ   — t        dgddœ«      }t        |gddd¬«      }t        dgdgdœ«      }t        j                  ||«       y )Nrª   r   r²   Tr³   r¬   r´   rµ   )r)   r>   r,   r.   s       r/   Útest_sort_columns_one_dfz,TestDataFrameConcat.test_sort_columns_one_dfé   sH   € ä˜s˜e¨Ñ*Ó+ˆÜ˜˜¨D°wÀTÔJˆÜ ! ¨C¨5Ñ1Ó2ˆÜ
×Ñ˜f hÕ/r1   N)Ú__name__Ú
__module__Ú__qualname__rj   ÚmarkÚxfailr0   rA   rN   rb   r}   rŽ   r“   rœ   r¡   r§   r°   r¶   r¸   © r1   r/   r   r      sh   „ Ø‡[�[×ÑÐHÐÓIñ0ó Jð0ò*	2ò1ò2Cò,89òt0ò*	0òJò0ò
Rò0ò0ó0r1   r   )Únumpyr"   rj   Úpandasr   r   r   r   r   Úpandas._testingÚ_testingr'   r   r¾   r1   r/   ú<module>rÃ      s)   ðÛ Û ã ÷ó õ ÷`0ò `0r1   