a
    p=ic                     @   s   d dl Z d dlZd dlZd dl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 G dd dejZG dd deZG d	d
 d
ejZG dd dejZdS )    N)IS_PYPY   )utilc                   @   s   e Zd ZeddddgZejdd	ddd	 Z
ejjed
ddd Zdd Zejjejdkdddd Zejjejdkdddd Zdd Zdd ZdS )TestF77Callbacktestssrccallbackzfoo.fnamezt,t2,c                 C   s   |  | d S N)check_function)selfr	    r   o/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/numpy/f2py/tests/test_callback.pytest_all   s    zTestF77Callback.test_allz,PyPy cannot modify tp_doc after PyType_Ready)reasonc                 C   s    t d}| jjj|ksJ d S )Na          a = t(fun,[fun_extra_args])

        Wrapper for ``t``.

        Parameters
        ----------
        fun : call-back function

        Other Parameters
        ----------------
        fun_extra_args : input tuple, optional
            Default: ()

        Returns
        -------
        a : int

        Notes
        -----
        Call-back functions::

            def fun(): return a
            Return objects:
                a : int
        )textwrapdedentmodulet__doc__)r   expectedr   r   r   test_docstring   s    
zTestF77Callback.test_docstringc                 C   sL  t | j|}|dd }|dks$J |dd dd}|dks@J |dd dd}|d	ks\J |d
d dd}|dksxJ |dd tjfd}|dksJ |tjtjfd}|dksJ || jjdd}|dksJ || jj}|dksJ || jjj}|dksJ G dd d}| }||}|dks0J ||j}|dksHJ d S )Nc                   S   s   dS )N   r   r   r   r   r   <lambda>6       z0TestF77Callback.check_function.<locals>.<lambda>r   c                 S   s   dS N   r   ar   r   r   r   8   r   )   )Zfun_extra_argsr   c                 S   s   | S r   r   r   r   r   r   r   :   r   r    c                 S   s   d|  S r   r   r   r   r   r   r   <   r   )      c                 S   s
   t | S r   )mathdegreesr   r   r   r   r   >   r            c                   @   s   e Zd Zdd Zdd ZdS )z)TestF77Callback.check_function.<locals>.Ac                 S   s   dS )Nr!   r   r   r   r   r   __call__K   s    z2TestF77Callback.check_function.<locals>.A.__call__c                 S   s   dS )N	   r   r(   r   r   r   mthN   s    z-TestF77Callback.check_function.<locals>.A.mthN)__name__
__module____qualname__r)   r+   r   r   r   r   AJ   s   r/   r!   r*   )	getattrr   r#   pir$   funcZfunc0Z	_cpointerr+   )r   r	   r   rr/   r   r   r   r   r   4   s2    
zTestF77Callback.check_functionwin32z)Fails with MinGW64 Gfortran (Issue #9673)c                 C   s,   dd }t | jd}||}|dks(J d S )Nc                 S   s   | dkrdS dS d S )Nr3   r   r   r   )coder   r   r   r   Z   s    z6TestF77Callback.test_string_callback.<locals>.callbackZstring_callbackr   )r0   r   )r   r   fr3   r   r   r   test_string_callbackW   s    z$TestF77Callback.test_string_callbackc                 C   s@   t dd}dd }t| jd}|||t|}|dks<J d S )N)r      S1c                 S   s6   | j |dfkrdS | jdkr dS t| dks2dS dS )Nr8   r   r9      r      r   )shapedtypenpall)cuZlencur   r   r   r   j   s    
z<TestF77Callback.test_string_callback_array.<locals>.callbackZstring_callback_arrayr   )r>   Zzerosr0   r   len)r   r@   r   r6   resr   r   r   test_string_callback_arrayd   s
    	z*TestF77Callback.test_string_callback_arrayc                    sp   g fdd  fddfdddD }|D ]}|   q6|D ]}|  qHdrltd S )	Nc                     s*   t d  jdd } | dks&J dS )NgMbP?c                   S   s   dS N{   r   r   r   r   r   r      r   z?TestF77Callback.test_threadsafety.<locals>.cb.<locals>.<lambda>rE   *   )timesleepr   r   )r3   r(   r   r   cb|   s    
z-TestF77Callback.test_threadsafety.<locals>.cbc                    sZ   z4t dD ]&}j }|dks&J |  q
W n  tyT   t  Y n0 d S )N2   rF   )ranger   r   r   	Exceptionappend	traceback
format_exc)r	   jr3   )rI   errorsr   r   r   runner   s    z1TestF77Callback.test_threadsafety.<locals>.runnerc                    s*   g | ]"}t d D ]}tj |fdqqS )   )targetargs)rK   	threadingThread).0argn)rR   r   r   
<listcomp>   s   z5TestF77Callback.test_threadsafety.<locals>.<listcomp>)r   t2z

)startjoinAssertionError)r   threadsr   r   )rI   rQ   rR   r   r   test_threadsafetyw   s    	



z!TestF77Callback.test_threadsafetyc              
   C   sp  z| j d W n6 tyF } zt|ds2J W Y d }~n
d }~0 0 z| j d W n6 ty } zt|dszJ W Y d }~n
d }~0 0 dd | j _| j d}|dksJ dd | j _| j d}|dksJ | j `z| j d W n: ty( } z t|dsJ W Y d }~n
d }~0 0 dd
d| j _| j d}|dksRJ | j d}|dkslJ d S )Nr:   zCallback global_f not definedz!cb: Callback global_f not definedc                 S   s   | d S )Nr   r   xr   r   r   r      r   z6TestF77Callback.test_hidden_callback.<locals>.<lambda>r;   c                 S   s   | d S )Nr:   r   rb   r   r   r   r      r   r   r   c                 S   s   | d S )Nr;   r   rb   r   r   r   r      r   r   )r   )r   Zhidden_callbackrL   str
startswithZhidden_callback2Zglobal_f)r   msgr3   r   r   r   test_hidden_callback   s0    ((*z$TestF77Callback.test_hidden_callbackN)r,   r-   r.   r   getpathsourcespytestmarkZparametrizesplitr   Zxfailr   r   r   Zskipifsysplatformr7   rC   ra   rg   r   r   r   r   r      s$   

#

(r   c                   @   s   e Zd ZdZdgZdS )TestF77CallbackPythonTLSz[
    Callback tests using Python thread-local storage instead of
    compiler-provided
    z-DF2PY_USE_PYTHON_TLSN)r,   r-   r.   r   optionsr   r   r   r   ro      s   ro   c                   @   s&   e Zd ZeddddgZdd ZdS )TestF90Callbackr   r   r   zgh17797.f90c                 C   s:   dd }t jg dt jd}| j||}|dks6J d S )Nc                 S   s   | d S rD   r   rb   r   r   r   incr   s    z*TestF90Callback.test_gh17797.<locals>.incrr   r:   r;   r=      )r>   arrayint64r   Zgh17797)r   rr   yr3   r   r   r   test_gh17797   s    zTestF90Callback.test_gh17797N)r,   r-   r.   r   rh   ri   ry   r   r   r   r   rq      s   rq   c                   @   s*   e Zd ZdZeddddgZdd ZdS )	TestGH18335zThe reproduction of the reported issue requires specific input that
    extensions may break the issue conditions, so the reproducer is
    implemented as a separate test class. Do not extend this test with
    other tests!
    r   r   r   zgh18335.f90c                 C   s8   dd }t jg dt jd}| j|}|dks4J d S )Nc                 S   s   | d  d7  < d S )Nr   r   r   rb   r   r   r   foo   s    z%TestGH18335.test_gh18335.<locals>.foors   rt   |   )r>   rv   Zint8r   Zgh18335)r   r{   rx   r3   r   r   r   test_gh18335   s    zTestGH18335.test_gh18335N)r,   r-   r.   r   r   rh   ri   r}   r   r   r   r   rz      s   rz   )r#   r   rm   rj   rV   rN   rG   numpyr>   Znumpy.testingr    r   ZF2PyTestr   ro   rq   rz   r   r   r   r   <module>   s    4	