o
    i                  
   @   sP  d dl mZmZ d dlmZ d dlZd dlZd dlZd dlZd dl	m
Z d dl	mZ d dlmZ zd dlZdZW n eyC   dZY nw d	Zd,d
dZdd Zdd Zdd ZejejejejdZdd dd dd dd dd dd dd dd dd d	Zdd dd d d d!d d"Zeeejd#Zd$d% Z d&d' Z!d(d) Z"d*d+ Z#dS )-    )assert_equalassert_)raisesN)_ccallback_c)_test_ccallback)LowLevelCallableTF       @c                 C   s(   | t krtd|d u r| d S | | S )Nz	bad value   )ERROR_VALUE
ValueError)a	user_data r   Y/home/kim/smarthome/.venv/lib/python3.10/site-packages/scipy/_lib/tests/test_ccallback.pycallback_python   s
   r   c                 C   s:   t std t| tj}|j}t }|||}|S )Ncffi not installed)		HAVE_CFFIpytestskipctypescastZc_void_pvaluecffiFFI)base	signatureZvoidpaddressffifuncr   r   r   _get_cffi_func   s   
r   c                  C   s   t d} t t | t jS Nr   )r   Zc_doubler   ZpointerZc_voidp)r   r   r   r   _get_ctypes_data-   s   
r!   c                  C   s"   t std t } | ddS )Nr   zdouble *r   )r   r   r   r   r   new)r   r   r   r   _get_cffi_data2   s   
r#   )simpleZnodatanonlocalcythonc                   C   s   t S N)r   r   r   r   r   <lambda>B   s    r(   c                   C      t  S r'   )r   test_get_plus1_capsuler   r   r   r   r(   C       c                   C      t tdS )NZplus1_cythonr   Zfrom_cython_test_ccallback_cythonr   r   r   r   r(   D       c                   C      t jS r'   )r.   plus1_ctypesr   r   r   r   r(   F       c                   C      t tjdS )Nz!double (*)(double, int *, void *))r   r.   r1   r   r   r   r   r(   G   r/   c                   C   r)   r'   )r   Ztest_get_plus1b_capsuler   r   r   r   r(   I   r+   c                   C   r,   )NZplus1b_cythonr-   r   r   r   r   r(   J   r/   c                   C   r0   r'   )r.   plus1b_ctypesr   r   r   r   r(   L   r2   c                   C   r3   )Nz)double (*)(double, double, int *, void *))r   r.   r4   r   r   r   r   r(   M   r/   )	pythoncapsuler&   r   r   Z	capsule_bZcython_bZctypes_bZcffi_bc                   C   r)   r'   )r   Ztest_get_plus1bc_capsuler   r   r   r   r(   S   r+   c                   C   r,   )NZplus1bc_cythonr-   r   r   r   r   r(   T   r/   c                   C   r0   r'   )r.   plus1bc_ctypesr   r   r   r   r(   V   r2   c                   C   r3   )Nz1double (*)(double, double, double, int *, void *))r   r.   r7   r   r   r   r   r(   W   s    )Z
capsule_bcZ	cython_bcZ	ctypes_bcZcffi_bc)r   r   r6   c                  C   N   dd } t t D ]}t t D ]}t t D ]}| ||| qqq
d S )Nc                    sv   t |  } t    t|  } tu r fdd}n	t |}t  t|  dd tt|  t t| |dd d S )Nc                    
    | dS r    r   xr   r   r   func2k      
z,test_callbacks.<locals>.check.<locals>.func2      ?r   g      @)	CALLERSFUNCS
USER_DATASr   r   r   assert_raisesr   r
   )callerr   r   r=   r   r<   r   checke   s   


ztest_callbacks.<locals>.check)sortedr@   keysrA   rB   rE   rD   r   r   r   r   r   test_callbacksd   s   rI   c                  C   r8   )Nc              
      s   t |  } t|  }t     tu r fdd}n	t |}t  tt| t d tt| |d t }z| |d W d S  tyd } zt|}t|j	|v | td|v | W Y d }~d S d }~ww )Nc                    r9   r    r   r:   r<   r   r   r=      r>   z0test_bad_callbacks.<locals>.check.<locals>.func2r?   z&double (double, double, int *, void *))
r@   rB   	BAD_FUNCSr   r   rC   r   strr   r   )rD   r   r   r=   Zllfuncerrmsgr   r<   r   rE      s$   


z!test_bad_callbacks.<locals>.check)rF   r@   rG   rJ   rB   rH   r   r   r   test_bad_callbacks   s   rN   c                  C   s`   t j} t  }t|dd}t|jd tt| |d t|dd}t|jd t| |dd d S )Nzbad signature)r      zdouble (double, int *, void *)   )r   test_call_simpler*   r   r   r   rC   r   )rD   r   Z
llcallabler   r   r   test_signature_override   s   rR   c                     s2    fdd  fdd} t  D ]}| | qd S )Nc                    s,   | dkrdS   fdd| d }d| S )Nr   r	   c                    
    | S r'   r   r:   callbackrD   r   r   r(         
 z5test_threadsafety.<locals>.callback.<locals>.<lambda>   r   )r   rD   resrU   )rD   r   rU      s   z#test_threadsafety.<locals>.callbackc                    sx   t    g d fddfddtdD }|D ]}|  q|D ]}|  q'td gt|  d S )N
   c                     s,   t d  fdd} |  d S )Ng{Gz?c                    rS   r'   r   r:   rT   r   r   r(      rV   z?test_threadsafety.<locals>.check.<locals>.run.<locals>.<lambda>)timesleepappend)r)rU   rD   countresultsr   r   run   s   
z-test_threadsafety.<locals>.check.<locals>.runc                    s   g | ]}t j d qS ))target)	threadingThread).0j)ra   r   r   
<listcomp>   s    z4test_threadsafety.<locals>.check.<locals>.<listcomp>   r   )r@   rangestartjoinr   len)rD   threadsthreadrY   )rD   r_   r`   ra   r   rE      s   

z test_threadsafety.<locals>.check)r@   rG   )rE   rD   r   rY   r   test_threadsafety   s
   
ro   r'   )$Znumpy.testingr   r   r   r   rC   r[   r   rc   Z
scipy._libr   r.   r   Zscipy._lib._ccallbackr   r   r   ImportErrorr
   r   r   r!   r#   rQ   Ztest_call_nodataZtest_call_nonlocalZtest_call_cythonr@   rA   rJ   Ztest_get_data_capsulerB   rI   rN   rR   ro   r   r   r   r   <module>   s`    
		"