Ë
    ìmxió  ã                   óÄ  — d dl Z d dlZd dlZd dlZd dlZ	 d dlZd dl
Zd ej                  «       fd ej                  «       fd ej                  «       fd ej                  «       fd ej                   «       fd ej"                  «       fd ej$                  «       fd	 ej&                  «       fd
 ej(                  «       fd ej*                  «       fd ej,                  «       fgZd„ Zd„ Zej                  j5                  de«      d„ «       Zd„ Zej                  j:                  d„ «       Zd„ Zd„ Z d„ Z!d„ Z"d„ Z#d„ Z$y# e$ r ej                  j
                  Z	Y �ŒDw xY w)é    NÚi1Úi2Úi4Úi8Úu1Úu2Úu4Úu8Úf2Úf4Úf8c                  óZ  — t         j                  j                  dd«      } t        j                  j                  | «      }|j                  dk(  sJ ‚|j                  g k(  sJ ‚|j                  dk(  sJ ‚|j                  | j                  k(  sJ ‚|j                  | j                  k(  sJ ‚|j                  sJ ‚|j                  sJ ‚| j                  «       }d|j                  _        t        j                  j                  |«      }|j                  rJ ‚t        j                  j                  | d¬«      }|j                  dk(  sJ ‚|j                  dd	gk(  sJ ‚|j!                  d
«      dk(  sJ ‚|j!                  d«      d	k(  sJ ‚t#        j$                  |«      } |«       €J ‚~ |«       �J ‚y )Né
   é   é   é(   F)ÚxÚy)Ú	dim_namesr   r   r   é   )ÚnpÚrandomÚrandnÚpaÚTensorÚ
from_numpyÚndimr   ÚsizeÚshapeÚstridesÚis_contiguousÚ
is_mutableÚcopyÚflagsÚ	writeableÚdim_nameÚweakrefÚref)ÚdataÚtensorÚdata2Úwrs       úP/home/htdocs/ttos/venv/lib/python3.12/site-packages/pyarrow/tests/test_tensor.pyÚtest_tensor_attrsr.   .   s†  € Ü�9‰9�?‰?˜2˜qÓ!€Dä�Y‰Y×!Ñ! $Ó'€Fà�;‰;˜!ÒÐÐØ×Ñ˜rÒ!Ð!Ð!Ø�;‰;˜"ÒÐÐØ�<‰<˜4Ÿ:™:Ò%Ð%Ð%Ø�>‰>˜TŸ\™\Ò)Ð)Ð)à×ÒÐÐØ×ÒÐÐð �I‰I‹K€EØ!€E‡K�KÔÜ�Y‰Y×!Ñ! %Ó(€FØ× Ò Ð Ð ô �Y‰Y×!Ñ! $°*Ð!Ó=€FØ�;‰;˜!ÒÐÐØ×Ñ  S˜zÒ)Ð)Ð)Ø�?‰?˜1Ó Ò$Ð$Ð$Ø�?‰?˜1Ó Ò$Ð$Ð$ä	�‰�VÓ	€BÙ‹4ÐÐÐØÙ‹4ˆ<Ð‰<ó    c                  ó  — t         j                  j                  t        j                  j                  dd«      «      } t        j                  | «      }| j                  «       }t        j                  | «      |dz   k(  sJ ‚y )Nr   r   r   )	r   r   r   r   r   r   ÚsysÚgetrefcountÚto_numpy)r*   ÚnÚarrays      r-   Útest_tensor_base_objectr6   O   sZ   € Ü�Y‰Y×!Ñ!¤"§)¡)§/¡/°"°aÓ"8Ó9€FÜ�‰˜Ó€AØ�O‰OÓ€EÜ�?‰?˜6Ó" a¨!¡eÒ+Ð+Ñ+r/   zdtype_str,arrow_typec                 óÎ  — t        j                  | «      }t        j                  «       5  t        j                  d«       dt         j
                  j                  dd«      z  j                  |«      }d d d «       t        j                  j                  «      }|j                  |k(  sJ ‚t        |«       |j                  «       }||k(  j                  «       sJ ‚y # 1 sw Y   ŒjxY w)NÚignoreéd   r   r   )r   ÚdtypeÚwarningsÚcatch_warningsÚsimplefilterr   r   Úastyper   r   r   ÚtypeÚreprr3   Úall)Ú	dtype_strÚ
arrow_typer:   r)   r*   Úresults         r-   Útest_tensor_numpy_roundtriprE   V   s´   € ä�H‰H�YÓ€Eä	×	 Ñ	 Ó	"ñ <Ü×Ñ˜hÔ'Ø”b—i‘i—o‘o b¨!Ó,Ñ,×4Ñ4°UÓ;ˆ÷<ô �Y‰Y×!Ñ! $Ó'€FØ�;‰;˜*Ò$Ð$Ð$äˆ„Là�_‰_Ó€FØ�F‰N×ÑÔ!Ð!Ñ!÷<ð <ús   ªACÃC$c                 óÆ  — t         j                  j                  dd«      }t        j                  j                  |«      }t        j                  j                  t        | «      d«      }t        j                  |d«      }t        j                  j                  ||«       |j                  d«       t        j                  j                  |«      }|j                  |«      sJ ‚y )Nr   r   zpyarrow-tensor-ipc-roundtripi   r   )r   r   r   r   r   r   ÚosÚpathÚjoinÚstrÚcreate_memory_mapÚipcÚwrite_tensorÚseekÚread_tensorÚequals)Útmpdirr)   r*   rH   ÚmmaprD   s         r-   Útest_tensor_ipc_roundtriprS   f   sš   € Ü�9‰9�?‰?˜2˜qÓ!€DÜ�Y‰Y×!Ñ! $Ó'€Fä�7‰7�<‰<œ˜F›Ð%CÓD€DÜ×Ñ  dÓ+€Dä‡F�F×Ñ˜ Ô%à‡I�Iˆa„LÜ�V‰V×Ñ Ó%€Fà�=‰=˜Ô Ð Ñ r/   c                 ó¨  — t         j                  j                  dd«      }t        j                  j                  |«      }| dz  }t        j                  |d¬«      }t        j                  j                  ||«       |j                  «        t        j                  j                  t        j                  |d¬«      «      }|j                  |«      sJ ‚y )Nr   r   ztensor-compressed-fileÚgzip)Úcompression)r   r   r   r   r   r   Úoutput_streamrL   rM   ÚcloserO   Úinput_streamrP   )Útempdirr)   r*   rH   Ú
out_streamrD   s         r-   Ú$test_tensor_ipc_read_from_compressedr\   u   sš   € ô �9‰9�?‰?˜2˜qÓ!€DÜ�Y‰Y×!Ñ! $Ó'€FàÐ-Ñ-€Dä×!Ñ! $°FÔ;€JÜ‡F�F×Ñ˜ 
Ô+Ø×ÑÔä�V‰V×Ñ¤§¡°À&Ô IÓJ€FØ�=‰=˜Ô Ð Ñ r/   c                 ó   — t         j                  j                  dd«      }t        j                  j                  |d d d…   «      }t         j                  j                  ddd«      }t        j                  j                  |d d …d d d…d d …f   «      }t        j                  j                  t        | «      d«      }t        j                  |d«      }||fD ]v  }|j                  d«       t        j                  j                  ||«       |j                  d«       t        j                  j                  |«      }|j                  |«      rŒvJ ‚ y )Nr   r   r   é   zpyarrow-tensor-ipc-stridedi   r   )r   r   r   r   r   r   rG   rH   rI   rJ   rK   rN   rL   rM   rO   rP   )	rQ   Údata1Útensor1r+   Útensor2rH   rR   r*   rD   s	            r-   Útest_tensor_ipc_stridedrb   …   sú   € Ü�I‰I�O‰O˜B Ó"€EÜ�i‰i×"Ñ" 5©¨1¨¡:Ó.€Gä�I‰I�O‰O˜B  1Ó%€EÜ�i‰i×"Ñ" 5ª©S¨q¨S²"¨Ñ#5Ó6€Gä�7‰7�<‰<œ˜F›Ð%AÓB€DÜ×Ñ  dÓ+€Dà˜GÐ$ò %ˆØ�	‰	�!ŒÜ
�‰×Ñ˜F DÔ)à�	‰	�!ŒÜ—‘×#Ñ# DÓ)ˆà�}‰}˜VÕ$Ð$Ð$ñ%r/   c                  óÀ  — d„ } d„ }t         j                  j                  ddd«      d d …d d d…d d …f   }t        j                  j                  |«      }t        j                  j                  t        j                  |«      «      } | ||«       |j                  «       }d|d<   t        j                  j                  t        j                  |«      «      } |||«       y )	Nc                 óF   — | j                  |«      sJ ‚| |k(  sJ ‚| |k7  rJ ‚y ©N©rP   ©ÚaÚbs     r-   Úeqztest_tensor_equals.<locals>.eqš   s+   € Ø�x‰x˜Œ{Ðˆ{Ø�AŠvˆˆvØ˜’FÐˆ|�Fr/   c                 óF   — | j                  |«      rJ ‚| |k(  rJ ‚| |k7  sJ ‚y re   rf   rg   s     r-   Úneztest_tensor_equals.<locals>.neŸ   s)   € Ø—8‘8˜A”;ÐˆØ˜’FÐˆ|Ø�AŠvˆ‰vr/   r   r^   r   r   g      ð?)é	   r   r   )r   r   r   r   r   r   Úascontiguousarrayr#   )rj   rl   r)   r`   ra   s        r-   Útest_tensor_equalsro   ™   s­   € òò
ô
 �9‰9�?‰?˜2˜q !Ó$¢R©¨1¨ªb [Ñ1€DÜ�i‰i×"Ñ" 4Ó(€GÜ�i‰i×"Ñ"¤2×#7Ñ#7¸Ó#=Ó>€GÙ€w�ÔØ�9‰9‹;€DØ€Dˆ�MÜ�i‰i×"Ñ"¤2×#7Ñ#7¸Ó#=Ó>€GÙ€w�Õr/   c                  óÚ   — t        j                  t        d¬«      5  t        t        j
                  j                  t        j                  d«      «      «       d d d «       y # 1 sw Y   y xY w)NÚ
unhashable)Úmatchr   )	ÚpytestÚraisesÚ	TypeErrorÚhashr   r   r   r   Úarange© r/   r-   Útest_tensor_hashingry   ®   sD   € ä	�‰”y¨Ô	5ñ 2ÜŒR�Y‰Y×!Ñ!¤"§)¡)¨B£-Ó0Ô1÷2÷ 2ñ 2ús   œ<A!Á!A*c                  óä   — t         j                  j                  dd«      } t        j                  j                  | «      }t        j                  j                  |«      | j                  dz  kD  sJ ‚y )Nr   r   é   )	r   r   r   r   r   r   rL   Úget_tensor_sizer   )r)   r*   s     r-   Útest_tensor_sizer}   ´   sO   € Ü�9‰9�?‰?˜2˜qÓ!€DÜ�Y‰Y×!Ñ! $Ó'€FÜ�6‰6×!Ñ! &Ó)¨T¯Y©Y¸©]Ò;Ð;Ñ;r/   c                 óŒ  — t         j                  j                  dd«      }t        j                  j                  |«      }t        j                  j                  |«      }t        j                  j                  t        | «      d«      }t        j                  ||«      }t        j                  j                  ||«       t        j                  dk(  r|j!                  «        t        j"                  |d¬«      }t        j                  j%                  |«      j'                  «       }t         j(                  j+                  ||«       y )Nr   r   zpyarrow-tensor-ipc-read-tensorÚ
emscriptenÚr)Úmode)r   r   r   r   r   r   rL   r|   rG   rH   rI   rJ   rK   rM   r1   ÚplatformrX   Ú
memory_maprO   r3   ÚtestingÚassert_equal)rQ   r)   r*   Ú	data_sizerH   Ú
write_mmapÚ	read_mmapr5   s           r-   Útest_read_tensorr‰   º   sØ   € ä�9‰9�?‰?˜2˜qÓ!€DÜ�Y‰Y×!Ñ! $Ó'€FÜ—‘×&Ñ& vÓ.€IÜ�7‰7�<‰<œ˜F›Ð%EÓF€DÜ×%Ñ% d¨IÓ6€JÜ‡F�F×Ñ˜ 
Ô+Ü
‡|�|�|Ò#ð 	×ÑÔä—‘˜d¨Ô-€IÜ�F‰F×Ñ˜yÓ)×2Ñ2Ó4€EÜ‡J�J×Ñ˜D %Õ(r/   c                  ó”  — t         j                  dft         j                  dft         j                  dft         j                  dft         j
                  dffD �]m  \  } }t        j                  d| ¬«      }|j                  } |j                  «       }t        j                  j                  |«      }t        |«      }|j                  |k(  sJ ‚|j                  |j                  k(  sJ ‚|j                  |j                  k(  sJ ‚|j                   dk(  sJ ‚|j"                  |j"                  k(  sJ ‚|j$                  |j$                  k(  sJ ‚|j$                  d	z  |j&                  j(                  k(  sJ ‚t        j*                  || «      j                  «       |k(  sJ ‚~~t        j*                  || «      j                  «       |k(  r�ŒnJ ‚ y )
Nz=bz=qz=QÚeÚdr   )r:   r   r{   )r   Úint8Úint64Úuint64Úfloat16Úfloat64rw   r:   Útolistr   r   r   Ú
memoryviewÚformatr   r    r   ÚnbytesÚitemsizer?   Ú	bit_widthÚ
frombuffer)r:   Úexpected_formatr)   Úlstr*   Úms         r-   Útest_tensor_memoryviewrœ   Ì   sz  € ä$&§G¡G¨T ?Ü$&§H¡H¨dÐ#3Ü$&§I¡I¨tÐ#4Ü$&§J¡J°Ð#4Ü$&§J¡J°Ð#4ð	#%ó 7Ñˆˆô �y‰y˜ 5Ô)ˆØ—
‘
ˆØ�k‰k‹mˆÜ—‘×%Ñ% dÓ+ˆÜ�vÓˆØ�x‰x˜?Ò*Ð*Ð*Ø�w‰w˜$Ÿ*™*Ò$Ð$Ð$Ø�y‰y˜DŸL™LÒ(Ð(Ð(Ø�v‰v˜Š{Ðˆ{Ø�x‰x˜4Ÿ;™;Ò&Ð&Ð&Ø�z‰z˜TŸ]™]Ò*Ð*Ð*Ø�z‰z˜A‰~ §¡×!6Ñ!6Ò6Ð6Ð6Ü�}‰}˜Q Ó&×-Ñ-Ó/°3Ò6Ð6Ð6Ø�DÜ�}‰}˜Q Ó&×-Ñ-Ó/°3Ô6Ð6Ð6ñ)7r/   )%rG   r1   rs   r;   r'   Únumpyr   ÚImportErrorÚmarkÚ
pytestmarkÚpyarrowr   r�   Úint16Úint32rŽ   Úuint8Úuint16Úuint32r�   r�   Úfloat32r‘   Útensor_type_pairsr.   r6   ÚparametrizerE   rS   rU   r\   rb   ro   ry   r}   r‰   rœ   rx   r/   r-   ú<module>rª      sr  ðó$ 
Û 
Û Û Û ð#Ûó ð 
ˆ7ˆ2�7‰7‹9ÐØ	ˆ8ˆ2�8‰8‹:ÐØ	ˆ8ˆ2�8‰8‹:ÐØ	ˆ8ˆ2�8‰8‹:ÐØ	ˆ8ˆ2�8‰8‹:ÐØ	ˆ9ˆ2�9‰9‹;ÐØ	ˆ9ˆ2�9‰9‹;ÐØ	ˆ9ˆ2�9‰9‹;ÐØ	ˆ:ˆ2�:‰:‹<ÐØ	ˆ:ˆ2�:‰:‹<ÐØ	ˆ:ˆ2�:‰:‹<ÐðÐ òòB,ð ‡�×ÑÐ/Ð1BÓCñ"ó Dð"ò!ð ‡�×Ññ!ó ð!ò%ò(ò*2ò<ò)ó$7øðe ò #Ø—‘×"Ñ"ƒJð#ús   –E  Å EÅE