Ë
    ž…j  ã                   ó*  — d dl Z d dlZd dlZd dlmc mZ d dlZ	d dl
mZ  ej                  d«      Z ej                  dd¬«      d„ «       Z ej                  dd¬«      d„ «       Z G d„ d	«      Z G d
„ d«      Z ej                  dd¬«      d„ «       Zy)é    NÚpyarrowz14.0)Úmin_versionc                 óÄ  — t        j                  g d¢g d¢dœ«      }|j                  «       }t        j                  j                  t        j                  |«      d«      dk(  sJ ‚t        j                  |«      }| rt        j                  «       nt        j                  «       }t        j                  g d¢t        j                  g d¢|«      dœ«      }|j                  |«      sJ ‚t        j                  dt        j                  «       fdt        j                  «       fg«      }t        j                  ||¬«      }|j                  |«      }|j                  |«      sJ ‚y )	N©é   é   é   ©ÚaÚbÚc©r   r   s   arrow_array_streamr   r   r   ©Úschema)ÚpdÚ	DataFrameÚ__arrow_c_stream__ÚctypesÚ	pythonapiÚPyCapsule_IsValidÚ	py_objectÚpaÚtableÚlarge_stringÚstringÚarrayÚequalsr   Úint8Úcast)Úusing_infer_stringÚdfÚcapsuler   Ústring_typeÚexpectedr   s          úh/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/frame/test_arrow_interface.pyÚtest_dataframe_arrow_interfacer&      s  € ä	�‰šIªOÑ<Ó	=€Bà×#Ñ#Ó%€Gä×Ñ×*Ñ*Ü×Ñ˜WÓ%Ð'<ó	
ð ò	ðð	ô �H‰H�R‹L€EÙ'9”"—/‘/Ô#¼r¿y¹y»{€KÜ�x‰xši¬b¯h©h²ÈÓ.TÑUÓV€HØ�<‰<˜Ô!Ð!Ð!ä�Y‰Y˜œbŸg™g›iÐ(¨3´·	±	³Ð*<Ð=Ó>€FÜ�H‰H�R ˆHÓ'€EØ�}‰}˜VÓ$€HØ�<‰<˜Ô!Ð!Ñ!ó    z15.0c                 ó”  — t        j                  g d¢g d¢dœ«      }t        j                  j	                  |«      j                  «       }| rt        j                  «       nt        j                  «       }t        j                  g d¢t        j                  g d¢|«      dœ«      }|j                  |«      sJ ‚t        j                  dt        j                  «       fdt        j                  «       fg«      }t        j                  j	                  ||¬«      j                  «       }|j                  |«      }|j                  |«      sJ ‚y )Nr   r
   r   r   r   r   )r   r   r   ÚRecordBatchReaderÚfrom_streamÚread_allr   r   r   r   r   r   r   r   )r    r!   r   r#   r$   r   s         r%   Útest_dataframe_to_arrowr,   %   sç   € ä	�‰šIªOÑ<Ó	=€Bä× Ñ ×,Ñ,¨RÓ0×9Ñ9Ó;€EÙ'9”"—/‘/Ô#¼r¿y¹y»{€KÜ�x‰xši¬b¯h©h²ÈÓ.TÑUÓV€HØ�<‰<˜Ô!Ð!Ð!ä�Y‰Y˜œbŸg™g›iÐ(¨3´·	±	³Ð*<Ð=Ó>€FÜ× Ñ ×,Ñ,¨R¸Ð,Ó?×HÑHÓJ€EØ�}‰}˜VÓ$€HØ�<‰<˜Ô!Ð!Ñ!r'   c                   ó   — e Zd Zd„ Zdd„Zy)ÚArrowArrayWrapperc                 ó   — || _         y ©N)r   )ÚselfÚbatchs     r%   Ú__init__zArrowArrayWrapper.__init__5   s	   € Øˆ�
r'   Nc                 ó8   — | j                   j                  |«      S r0   )r   Ú__arrow_c_array__©r1   Úrequested_schemas     r%   r5   z#ArrowArrayWrapper.__arrow_c_array__8   s   € Ø�z‰z×+Ñ+Ð,<Ó=Ð=r'   r0   )Ú__name__Ú
__module__Ú__qualname__r3   r5   © r'   r%   r.   r.   4   s   „ òô>r'   r.   c                   ó   — e Zd Zd„ Zdd„Zy)ÚArrowStreamWrapperc                 ó   — || _         y r0   )Ústream)r1   r   s     r%   r3   zArrowStreamWrapper.__init__=   s	   € Øˆ�r'   Nc                 ó8   — | j                   j                  |«      S r0   )r?   r   r6   s     r%   r   z%ArrowStreamWrapper.__arrow_c_stream__@   s   € Ø�{‰{×-Ñ-Ð.>Ó?Ð?r'   r0   )r8   r9   r:   r3   r   r;   r'   r%   r=   r=   <   s   „ òô@r'   r=   c                 ó   — t         j                  g d¢g d¢dœ«      }t        j                  j	                  |«      }t        j                  g d¢g d¢dœ«      }| s9|d   j                  t        j                  t        j                  ¬«      «      |d<   t        j                  ||«       t        j                  j	                  t        |«      «      }t        j                  ||«       t         j                  g d¢g d¢gddg¬«      }t        j                  j	                  |«      }t        j                  ||«       t        j                  j	                  t        |«      «      }t        j                  ||«       t        j                  t         d¬	«      5  t        j                  j	                  g d¢g d¢dœ«       d d d «       y # 1 sw Y   y xY w)
Nr   r
   r   r   )Úna_valuer   )Únamesz+Expected an Arrow-compatible tabular object)Úmatch)r   r   r   r   Ú
from_arrowÚastypeÚStringDtypeÚnpÚnanÚtmÚassert_frame_equalr=   Úrecord_batchr.   ÚpytestÚraisesÚ	TypeError)r    r   Úresultr$   r2   s        r%   Útest_dataframe_from_arrowrQ   D   sN  € ô �H‰Hš9ª?Ñ;Ó<€Eä�\‰\×$Ñ$ UÓ+€FÜ�|‰|¢)²/ÑBÓC€HÙØ  ™×,Ñ,¬R¯^©^ÄRÇVÁVÔ-LÓMˆ�‰Ü×Ñ˜& (Ô+ô �\‰\×$Ñ$Ô%7¸Ó%>Ó?€FÜ×Ñ˜& (Ô+ô �O‰OšYªÐ8ÀÀcÀ
ˆOÓK€Eä�\‰\×$Ñ$ UÓ+€FÜ×Ñ˜& (Ô+ä�\‰\×$Ñ$Ô%6°uÓ%=Ó>€FÜ×Ñ˜& (Ô+ô 
�‰”yÐ(UÔ	Vñ HÜ
�‰×Ñ¢i²oÑ FÔG÷H÷ Hñ Hús   Æ'GÇG)r   ÚnumpyrH   rM   Úpandas.util._test_decoratorsÚutilÚ_test_decoratorsÚtdÚpandasr   Úpandas._testingÚ_testingrJ   Úimportorskipr   Ú
skip_if_nor&   r,   r.   r=   rQ   r;   r'   r%   ú<module>r\      s§   ðÛ ã Û ç )Ð )ã Ý à€V×Ñ˜Ó#€ð €‡�ˆy fÔ-ñ"ó .ð"ð, €‡�ˆy fÔ-ñ"ó .ð"÷>ñ >÷@ñ @ð €‡�ˆy fÔ-ñHó .ñHr'   