File: //usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/__pycache__/test_callback.cpython-310.pyc
o
���g^ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
ddl m Z G dd� de j
�ZG dd� de�ZG d d
� d
e j
�ZG d d � d e j
�ZG d
d� de j
�ZdS )� N)�IS_PYPY� )�utilc @ s� e Zd Ze�dddd�gZej�dd� d��dd � �Z
ejj e d
d �d d
� �Z
dd� Zejjejdkdd �dd� �Zejjejdkdd �dd� �Zdd� Zdd� ZdS )�TestF77Callback�tests�src�callbackzfoo.f�namezt,t2�,c C s | � |� d S �N)�check_function)�selfr � r �I/usr/local/lib/python3.10/dist-packages/numpy/f2py/tests/test_callback.py�test_all s zTestF77Callback.test_allz,PyPy cannot modify tp_doc after PyType_Ready)�reasonc C s t �d�}| jjj|ksJ �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 sF t | j|�}|dd� �}|dksJ �|dd� dd�}|dks J �|dd� dd�}|d ks.J �|d
d� d d�}|d ks<J �|d
d� tjfd�}|dksLJ �|tjtjfd�}|dks[J �|| jjdd�}|dksiJ �|| jj�}|dksuJ �|| jjj�}|dks�J �G dd� d�}|� }||�}|dks�J �||j�}|dks�J �d S )Nc S � dS )N� r r r r r �<lambda>6 � z0TestF77Callback.check_function.<locals>.<lambda>r c S r �N� r ��ar r r r 8 r )� )�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 < � )� � c S s
t �| �S r )�math�degreesr r r r r > s
� � � c @ s e Zd Zdd� Zdd� ZdS )z)TestF77Callback.check_function.<locals>.Ac S r )Nr$ r �r
r r r �__call__K � z2TestF77Callback.check_function.<locals>.A.__call__c S r )N� r r+ r r r �mthN r- z-TestF77Callback.check_function.<locals>.A.mthN)�__name__�
__module__� __qualname__r, r/ r r r r �AJ s r3 r$ r. ) �getattrr r& �pir' �func�func0� _cpointerr/ )r
r r �rr3 r r r r r 4 s2
z TestF77Callback.check_function�win32z)Fails with MinGW64 Gfortran (Issue #9673)c C s, dd� }t | jd�}||�}|dksJ �d S )Nc S s | dkrdS dS )Nr9 r r r )�coder r r r Z s z6TestF77Callback.test_string_callback.<locals>.callback�string_callbackr )r4 r )r
r �fr9 r r r �test_string_callbackW s z$TestF77Callback.test_string_callbackc C sh t �dd�}t �dd�}t �dgd�}dd� }t| jd�}|||fD ]}||||j�}|d ks1J �q"d S )
N)r �S8)r � �c� c S s4 | j |fkrdS | jdkrdS t�| dk�sdS dS )Nr r? � � � r )�shape�dtype�np�all)�cu�lencur r r r l s
z<TestF77Callback.test_string_callback_array.<locals>.callback�string_callback_arrayr )rH �zeros�arrayr4 r �size)r
�cu1�cu2�cu3r r= rJ �resr r r �test_string_callback_arrayd s �z*TestF77Callback.test_string_callback_arrayc sp g ��fdd�� � ��fdd���fdd�dD �}|D ]}|� � q|D ]}|�� q$d�����r6t���d S ) Nc s* t �d� � j�dd� �} | dksJ �dS )Ng����MbP?c S r �N�{ r r r r r r � r z?TestF77Callback.test_threadsafety.<locals>.cb.<locals>.<lambda>rV �* )�time�sleepr r )r9 r+ r r �cb s
z-TestF77Callback.test_threadsafety.<locals>.cbc sZ zt d�D ]}�j�� �}|dksJ ���| � qW d S ty, ��t�� � Y d S w )N�2 rW )�ranger r r � Exception�append� traceback�
format_exc)r �jr9 )rZ �errorsr
r r �runner� s � �z1TestF77Callback.test_threadsafety.<locals>.runnerc s* g | ]}t d �D ]
}tj� |fd��qqS )� )�target�args)r\ � threading�Thread)�.0�arg�n)rc r r �
<listcomp>� s ���z5TestF77Callback.test_threadsafety.<locals>.<listcomp>)r �t2z
)�start�join�AssertionError)r
�threadsr r )rZ rb rc r
r �test_threadsafetyz s
�
�z!TestF77Callback.test_threadsafetyc
C sh z| j �d� W n ty# } zt|��d�sJ �W Y d }~nd }~ww z| j �d� W n tyG } zt|��d�s=J �W Y d }~nd }~ww dd� | j _| j �d�}|dksZJ �dd� | j _| j �d�}|dkslJ �| j `z| j �d� W n ty� } zt|��d�s�J �W Y d }~nd }~ww d d
d�| j _| j �d�}|d ks�J �| j �d�}|dks�J �d S )
NrC z Callback global_f not definedz!cb: Callback global_f not definedc S � | d S )Nr r ��xr r r r � r# z6TestF77Callback.test_hidden_callback.<locals>.<lambda>rE c S rs )NrC r rt r r r r � r# r r c S rs )NrE r rt r r r r � r# r )r )r �hidden_callbackr] �str�
startswith�hidden_callback2�global_f)r
�msgr9 r r r �test_hidden_callback� s< �� �� �� z$TestF77Callback.test_hidden_callbackN)r0 r1 r2 r �getpath�sources�pytest�mark� parametrize�splitr �xfailr r r �skipif�sys�platformr> rT rr r| r r r r r s&
�
#�
�
(r c @ s e Zd ZdZdgZdS )�TestF77CallbackPythonTLSz[
Callback tests using Python thread-local storage instead of
compiler-provided
z-DF2PY_USE_PYTHON_TLSN)r0 r1 r2 r �optionsr r r r r� � s
r� c @ s& e Zd Ze�dddd�gZdd� ZdS )�TestF90Callbackr r r z gh17797.f90c C s: dd� }t jg d�t jd�}| j�||�}|dksJ �d S )Nc S rs rU r rt r r r �incr� � z*TestF90Callback.test_gh17797.<locals>.incr)r rC rE )rG � )rH rN �int64r �gh17797)r
r� �yr9 r r r � test_gh17797� s z TestF90Callback.test_gh17797N)r0 r1 r2 r r} r~ r� r r r r r� � s r� c @ s* e Zd ZdZe�dddd�gZdd� ZdS ) � TestGH18335z�The 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 �$ dd� }| j �|�}|dksJ �d S )Nc S s | d d7 < d S )Nr r r rt r r r �foo� s z%TestGH18335.test_gh18335.<locals>.foo�| )r �gh18335)r
r� r9 r r r � test_gh18335� � zTestGH18335.test_gh18335N)r0 r1 r2 r r r} r~ r� r r r r r� � s r� c @ s8 e Zd Ze�dddd�e�dddd�gZdZdd� Zd S )
� TestGH25211r r r z gh25211.fz gh25211.pyf� callback2c C r� )Nc S s | | S r r rt r r r �bar� r� z%TestGH25211.test_gh18335.<locals>.bar�n )r r� )r
r� rS r r r r� � r� zTestGH25211.test_gh18335N)r0 r1 r2 r r} r~ � module_namer� r r r r r� � s � r� )r& r r� r rg r_ rX �numpyrH �
numpy.testingr rB r �F2PyTestr r� r� r� r� r r r r �<module> s 7