o
    '¦iö	  ć                   @   s,   d dl Zd dlZdd Zdd Zdd ZdS )é    Nc                  C   s   t  ” } | 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s6J t	| dks>J d S )Né’’’’z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   © r   śW/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/pyarrow/tests/test_device.pyŚtest_device_memory_manager   s   

r   c                  C   s   t  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s3J |jt  ” jks=J |j	jsCJ d S )N©r   é   é   r   )r   ŚarrayŚbuffersr   r	   r
   Ś
isinstancer   ŚDeviceŚmemory_managerŚMemoryManagerr   r   )ŚarrŚbufr   r   r   Śtest_buffer_device$   s   
r   c                  C   sZ  t  ” } t  g d¢”}t  d|i”}| | jfD ]V}| |”}| |”s%J | ” d j| jks1J | ” d j| ” d jksAJ | |”}| |”sMJ |d  ” d j| jks[J |d  ” d j| ” d jksmJ qt	j
tdd | | jj” W d    n1 sw   Y  t	j
tdd | | jj” W d    d S 1 s¦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_copiedr   r   r   Śtest_copy_to0   s$   
 
&’"’r*   )Śpyarrowr   r#   r   r   r*   r   r   r   r   Ś<module>   s
   