o
    h	                     @   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 )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    r   M/var/www/vscode/kcb/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 sw   Y  t	j
tdd || jj W d    d S 1 sw   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
   