
    Pi	                     (    d dl Zd dlZd Zd Zd ZdS )    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 d S )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     m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pyarrow/tests/test_device.pytest_device_memory_managerr      s    		&	(	(B99YF==r!!!!!8!<<<<<11111V<<:::::88DDDDDDD    c                     t          j        g d          } |                                 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 d S )Nr         r   )r   arraybuffersr
   r   r   
isinstancer   Devicememory_managerMemoryManagerr   r   )arrbufs     r   test_buffer_devicer   $   s    
(999

C
++--
C?b599999cj"),,,,,c("*:;;;;;::::688?????$$$$$$r   c                  b   t          j                    } t          j        g d          }t          j        d|i          }| | j        fD ]?}|                    |          }|                    |          sJ |                                d         j        | j        k    sJ |                                d         j        |                                d         j        k    sJ |                    |          }|                    |          sJ |d                                         d         j        | j        k    sJ |d                                         d         j        |                                d         j        k    sJ At          j
        t          d          5  |                    | j        j                   d d d            n# 1 swxY w Y   t          j
        t          d          5  |                    | j        j                   d d d            d S # 1 swxY w Y   d S )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
(999

COUCL))ERY 	T 	T[[&&
  %%%%%!!##A&-::::!!##A&.#++--2B2JJJJJ}}T**""5)))))E"**,,Q/6")CCCCE"**,,Q/73;;==;K;SSSSSS	y(S	T	T	T + +BI)***+ + + + + + + + + + + + + + + 
y(S	T	T	T - -bi+,,,- - - - - - - - - - - - - - - - - -s$   % GGG7 H$$H(+H()pyarrowr   r'   r   r   r.    r   r   <module>r1      sV   $     
E 
E 
E	% 	% 	%- - - - -r   