
    g?                     R    d dl Z d dlZe j                  j                  Zd Zd Zd Zy)    Nc                      t        j                  t        d            } d}t        j                  t
        |      5  | j                          d d d        y # 1 sw Y   y xY w)N
   5Cannot return a numpy.ndarray if NumPy is not presentmatch)paarrayrangepytestraisesImportErrorto_numpy)arrmsgs     U/var/www/openai/venv/lib/python3.12/site-packages/pyarrow/tests/test_without_numpy.pytest_array_to_npr      s<    
((59
C
AC	{#	. 
/	.	.s   AAc                      t        j                  g dg dg g      } d}t        j                  t        |      5  | j                          d d d        y # 1 sw Y   y xY w)N)         )         r   r   )r   chunked_arrayr   r   r   r   )datar   s     r   test_chunked_array_to_npr   $   sK    
 D
 BC	{#	. 
/	.	.s   AAc                     t        j                  t        j                         ddg      } g dg dg dg}t        j                  |t        j                  t        j                         d            }t         j
                  j                  | |      }|j                         }d}t        j                  t        |      5  |j                          d d d        y # 1 sw Y   y xY w)Nr   )r   r   r   r   )r         (   )d      i,  i  r   r   r   )r   fixed_shape_tensorint32r	   list_ExtensionArrayfrom_storage	to_tensorr   r   r   r   )tensor_typer   storagetensor_arraytensorr   s         r   test_tensor_to_npr-   0   s    ''
QF;K)+?
@ChhsBHHRXXZ34G$$11+wGL##%F
AC	{#	. 
/	.	.s   ?CC")	r   pyarrowr   marknonumpy
pytestmarkr   r   r-        r   <module>r4      s-   $   [[  
	
r3   