o
    h                  
   @   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   Z/var/www/html/scripts/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castc_void_pvaluecffiFFI)base	signaturevoidpaddressffifuncr   r   r   _get_cffi_func   s   
r!   c                  C   s   t d} t t | t jS Nr   )r   c_doubler   pointer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(   )simple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 )Nplus1_cythonr   from_cython_test_ccallback_cythonr   r   r   r   r.   D       c                   C      t jS r-   )r6   plus1_ctypesr   r   r   r   r.   E       c                   C      t tjdS )Nz!double (*)(double, int *, void *))r!   r6   r9   r   r   r   r   r.   F       c                   C   r/   r-   )r   test_get_plus1b_capsuler   r   r   r   r.   H   r1   c                   C   r2   )Nplus1b_cythonr4   r   r   r   r   r.   I   r7   c                   C   r8   r-   )r6   plus1b_ctypesr   r   r   r   r.   J   r:   c                   C   r;   )Nz)double (*)(double, double, int *, void *))r!   r6   r?   r   r   r   r   r.   K   r<   )	pythoncapsuler,   r   r   	capsule_bcython_bctypes_bcffi_bc                   C   r/   r-   )r   test_get_plus1bc_capsuler   r   r   r   r.   Q   r1   c                   C   r2   )Nplus1bc_cythonr4   r   r   r   r   r.   R   r7   c                   C   r8   r-   )r6   plus1bc_ctypesr   r   r   r   r.   S   r:   c                   C   r;   )Nz1double (*)(double, double, double, int *, void *))r!   r6   rH   r   r   r   r   r.   T   r<   )
capsule_bc	cython_bc	ctypes_bccffi_bc)r   r   rA   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   func2f      
z,test_callbacks.<locals>.check.<locals>.func2      ?r   g      @)	CALLERSFUNCS
USER_DATASr   r   r   assert_raisesr   r
   )callerr    r   rR   r   rQ   r   check`   s   


ztest_callbacks.<locals>.check)sortedrU   keysrV   rW   rZ   rY   r    r   r   r   r   test_callbacks_   s   r^   c                  C   rM   )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                    rN   r"   r   rO   rQ   r   r   rR      rS   z0test_bad_callbacks.<locals>.check.<locals>.func2rT   z&double (double, double, int *, void *))
rU   rW   	BAD_FUNCSr   r   rX   r   strr   r   )rY   r    r   rR   llfuncerrmsgr   rQ   r   rZ   |   s$   


z!test_bad_callbacks.<locals>.check)r[   rU   r\   r_   rW   r]   r   r   r   test_bad_callbacks{   s   rd   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_simpler0   r   r   r   rX   r   )rY   r    
llcallabler   r   r   test_signature_override   s   ri   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   rO   callbackrY   r   r   r.         
 z5test_threadsafety.<locals>.callback.<locals>.<lambda>   r   )r   rY   resrl   )rY   r   rl      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                    rj   r-   r   rO   rk   r   r   r.      rm   z?test_threadsafety.<locals>.check.<locals>.run.<locals>.<lambda>)timesleepappend)r)rl   rY   countresultsr   r   run   s   
z-test_threadsafety.<locals>.check.<locals>.runc                    s   g | ]}t j d qS ))target)	threadingThread).0j)rx   r   r   
<listcomp>   s    z4test_threadsafety.<locals>.check.<locals>.<listcomp>   r   )rU   rangestartjoinr   len)rY   threadsthreadrp   )rY   rv   rw   rx   r   rZ      s   

z test_threadsafety.<locals>.check)rU   r\   )rZ   rY   r   rp   r   test_threadsafety   s
   
r   r-   )$numpy.testingr   r   r   r   rX   rr   r   rz   
scipy._libr   r6   r   scipy._lib._ccallbackr   r   r   ImportErrorr
   r   r!   r&   r(   rg   test_call_nodatatest_call_nonlocaltest_call_cythonrU   rV   r_   test_get_data_capsulerW   r^   rd   ri   r   r   r   r   r   <module>   s`    
			"