3
���h� � @ s^ d dl Z d dlZd dlZd dlZd dlZd dlmZmZm Z ddl
m Z G dd� de j �Z
dS )� N)�assert_� assert_equal�IS_PYPY� )�utilc @ s� e Zd ZdZejjddjd��dd� �Zejj e
dd�d d
� �Z d d � Z ejj
ejd
kdd�dd� �Zejj
ejd
kdd�dd� �ZdS )�TestF77Callbacka�
subroutine t(fun,a)
integer a
cf2py intent(out) a
external fun
call fun(a)
end
subroutine func(a)
cf2py intent(in,out) a
integer a
a = a + 11
end
subroutine func0(a)
cf2py intent(out) a
integer a
a = 11
end
subroutine t2(a)
cf2py intent(callback) fun
integer a
cf2py intent(out) a
external fun
call fun(a)
end
subroutine string_callback(callback, a)
external callback
double precision callback
double precision a
character*1 r
cf2py intent(out) a
r = 'r'
a = callback(r)
end
subroutine string_callback_array(callback, cu, lencu, a)
external callback
integer callback
integer lencu
character*8 cu(lencu)
integer a
cf2py intent(out) a
a = callback(cu, lencu)
end
�namezt,t2�,c C s | j |� d S )N)�check_function)�selfr � r �?/tmp/pip-build-5_djhm0z/numpy/numpy/f2py/tests/test_callback.py�test_all> s zTestF77Callback.test_allz,PyPy cannot modify tp_doc after PyType_Ready)�reasonc C s t jd�}t| jjj|� 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�dedentr �module�t�__doc__)r �expectedr r r
�test_docstringB s z TestF77Callback.test_docstringc C s� t | j|�}|dd� �}t|dkt|�� |dd� dd�}t|dkt|�� |dd� dd�}t|dkt|�� |d d� dd�}t|d kt|�� |d d� tjfd�}t|d
kt|�� |tjtjfd�}t|d
kt|�� || jjdd�}t|dkt|�� || jj�}t|dkt|�� || jjj �}t|dkt|�� G dd� d�}|� }||�}t|d
kt|�� ||j
�}t|dkt|�� d S )Nc S s dS )N� r r r r r
�<lambda>c s z0TestF77Callback.check_function.<locals>.<lambda>r c S s dS )N� r )�ar r r
r e s � )Zfun_extra_argsr c S s | S )Nr )r r r r
r g s c S s d| S )Nr r )r r r r
r i s � � c S s
t j| �S )N)�math�degrees)r r r r
r k s � � � c @ s e Zd Zdd� Zdd� ZdS )z)TestF77Callback.check_function.<locals>.Ac S s dS )Nr r )r r r r
�__call__y s z2TestF77Callback.check_function.<locals>.A.__call__c S s dS )N� r )r r r r
�mth| s z-TestF77Callback.check_function.<locals>.A.mthN)�__name__�
__module__� __qualname__r# r% r r r r
�Aw s r) r$ )r )r )r )r ) �getattrr r �reprr �pir �funcZfunc0Z _cpointerr% )r r r �rr) r r r r
r
a s2
z TestF77Callback.check_function�win32z)Fails with MinGW64 Gfortran (Issue #9673)c C s2 dd� }t | jd�}||�}t|dkt|�� d S )Nc S s | dkr dS dS d S )Nr. r r r )�coder r r
�callback� s z6TestF77Callback.test_string_callback.<locals>.callbackZstring_callbackr )r* r r r+ )r r1 �fr. r r r
�test_string_callback� s z$TestF77Callback.test_string_callbackc C sF t jdd�}dd� }t| jd�}|||t|��}t|dkt|�� d S ) Nr � �S1c S s6 | j |dfkrdS | jdkr dS tj| dk�s2dS dS )Nr4 r r5 � � � r )�shapeZdtype�np�all)�cuZlencur r r
r1 � s
z<TestF77Callback.test_string_callback_array.<locals>.callbackZstring_callback_arrayr )r r4 )r: �zerosr* r �lenr r+ )r r< r1 r2 �resr r r
�test_string_callback_array� s
z*TestF77Callback.test_string_callback_arrayN)r& r' r( r0 �pytest�markZ parametrize�splitr Zxfailr r r
Zskipif�sys�platformr3 r@ r r r r
r s 1 #
r )r r rD rA Znumpyr: Z
numpy.testingr r r � r ZF2PyTestr r r r r
�<module> s |