
    g	                     &    d dl Zd dlZd Zd Zd Zy)    Nc                  `   t        j                         } | j                  sJ | j                  }|j                  sJ |j                  dk(  sJ |j
                  t         j                  j                  k(  sJ |j                  dk(  sJ ||k(  sJ t        |      dk(  sJ t        |       dk(  sJ y )Nzarrow::CPUDevicez<pyarrow.Device: CPUDevice()>z+<pyarrow.MemoryManager device: CPUDevice()>)
padefault_cpu_memory_manageris_cpudevice	device_iddevice_typeDeviceAllocationTypeCPU	type_namerepr)mmr   s     N/var/www/openai/venv/lib/python3.12/site-packages/pyarrow/tests/test_device.pytest_device_memory_managerr      s    		&	&	(B999YYF===r!!!!8!8!<!<<<<1111V<::::8DDDD    c                     t        j                  g d      } | j                         d   }|j                  t         j                  j
                  k(  sJ t        |j                  t         j                        sJ t        |j                  t         j                        sJ |j                  sJ |j                  j                  sJ |j                  t        j                         j                  k(  sJ |j                  j                  sJ y )Nr         r   )r   arraybuffersr
   r   r   
isinstancer   Devicememory_managerMemoryManagerr   r   )arrbufs     r   test_buffer_devicer   $   s    
((9
C
++-
C??b5599999cjj")),,,c(("*:*:;;;:::::::668?????$$$$r   c                  F   t        j                         } t        j                  g d      }t        j                  d|i      }| | j                  fD ]#  }|j                  |      }|j                  |      sJ |j                         d   j                  | j                  k(  sJ |j                         d   j                  |j                         d   j                  k7  sJ |j                  |      }|j                  |      sJ |d   j                         d   j                  | j                  k(  sJ |d   j                         d   j                  |j                         d   j                  k7  r$J  t        j                  t        d      5  |j                  | j                  j                         d d d        t        j                  t        d      5  |j                  | j                  j                         d d d        y # 1 sw Y   SxY w# 1 sw Y   y xY w)Nr   colr   z)Argument 'destination' has incorrect type)match)r   r   r   record_batchr   copy_toequalsr   addresspytestraises	TypeErrorr
   )r   r   batchdest
arr_copiedbatch_copieds         r   test_copy_tor.   0   s   		&	&	(B
((9
COOUCL)ERYY[[&
  %%%!!#A&--:::!!#A&..#++-2B2J2JJJJ}}T*""5)))E"**,Q/66"))CCCE"**,Q/773;;=;K;S;SSSS   
y(S	TBII))* 
U 
y(S	Tbii++, 
U	T 
U	T 
U	Ts   &H&HHH )pyarrowr   r'   r   r   r.    r   r   <module>r1      s   $  
E	%-r   