U
|
P ±d°* ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ ddlm Z d dl
|
m Z d dl Z G dd de j ZG dd d ZG d
|
d d ZG d d d e j ZG dd de j ZG dd dZG dd de j ZG dd dZG dd de j ZG dd de j ZG dd de j ZdS ) é N)Úmarkinnerspacesé )Úutil)Ú crackfortranc @ s& e Zd Ze dddd¡gZdd ZdS )Ú TestNoSpaceÚtestsÚsrcr z gh15035.fc C s t jdddgt jd}t jdddgt jd}| j |¡ t ||d ¡sLt| j ||g¡ t ||d ¡spt| j d¡dkstd S )Nr é é )ZdtypeZ23ó 2) ÚnpÚarrayÚfloat64ÚmoduleZsubbZallcloseÚAssertionErrorZsubcÚt0)ÚselfÚkÚw© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/f2py/tests/test_crackfortran.pyÚ test_module s zTestNoSpace.test_moduleN)Ú__name__Ú
|
__module__Ú __qualname__r ÚgetpathÚsourcesr r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestPublicPrivatec C sÌ t dddd¡}t t|g¡}t|dks0t|d }d|d d d
|
ksPtd |d d d
|
kshtd|d d d
|
kstd |d d d
|
kstd|d d d
|
ks°td |d d d
|
ksÈtd S )Nr r r zprivatemod.f90r r ÚprivateÚvarsÚaÚattrspecÚpublicÚbÚseta©r r r ÚstrÚlenr )r ÚfpathÚmodr r r Útest_defaultPrivate s z%TestPublicPrivate.test_defaultPrivatec C s t dddd¡}t t|g¡}t|dks0t|d }d|d d d
|
ksPtd |d d d
|
kshtd|d d d
|
kstd |d d d
|
kstd S ) Nr r r z publicmod.f90r r r r r r! r" r$ r% ©r Útmp_pathr( r) r r r Útest_defaultPublic( s z$TestPublicPrivate.test_defaultPublicc C s t dddd¡}t t|g¡}t|dks0t|d d }t|d d d
|
d hksXtt|d d d d hksttt|d d d hkstd S )Nr r r zaccesstype.f90r r r r r! r zbind(c)Zb_r" Úc)r r r r& r' r Úset)r r, r( r) Úttr r r Útest_access_type2 s z"TestPublicPrivate.test_access_typeN)r r r r* r- r1 r r r r r s
|
r c @ s e Zd Zdd Zdd ZdS )ÚTestModuleProcedurec C s2 t dddd¡}t t|g¡}t|dks0t|d }d|krPt|d dksTt|d d d d
|
ksltd |d d kst|d d d d d gkst|d d d dks´td |d d ksÈt|d d d dgksât|d d d dksútd |d d kst|d d d ddgks.td S )Nr r r z operators.f90r r Úbodyé Únamezoperator(.item.)Z implementedbyZitem_intZ item_realr z operator(==)Zitems_are_equalr
|
z assignment(=)Zget_intZget_realr% r+ r r r Útest_moduleOperators= s$ ÿÿz(TestModuleProcedure.test_moduleOperatorsc C s t dddd¡}t t|g¡}t|dks0t|d }|d d d d
|
gksRt|d d d d gkslt|d d d d gkstd S )Nr r r zpubprivmod.f90r r r r r! r r# r" r$ r% r+ r r r Útest_notPublicPrivateO s z)TestModuleProcedure.test_notPublicPrivateN)r r r r6 r7 r r r r r2 < s r2 c @ s. e Zd Ze dddd¡gZdd Zdd Zd S )
|
Ú TestExternalr r r z gh17859.fc C s$ dd }| j |¡}|dks td S )Nc S s | d S ©Né{ r ©Úxr r r Úincr^ s z5TestExternal.test_external_as_statement.<locals>.incrr: )r Zexternal_as_statementr ©r r= Úrr r r Útest_external_as_statement] s z'TestExternal.test_external_as_statementc C s$ dd }| j |¡}|dks td S )Nc S s | d S r9 r r; r r r r= e s z5TestExternal.test_external_as_attribute.<locals>.incrr: )r Zexternal_as_attributer r> r r r Útest_external_as_attributed s z'TestExternal.test_external_as_attributeN)r r r r r r r@ rA r r r r r8 Y s r8 c @ s& e Zd Ze dddd¡gZdd ZdS )ÚTestCrackFortranr r r z
|
gh2848.f90c C s | j dd¡}|dkstd S )Nr r )r r )r Zgh2848r )r r? r r r Ú test_gh2848p s z TestCrackFortran.test_gh2848N)r r r r r r rC r r r r rB l s rB c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestMarkinnerspacesc C s* ddddg}|D ]}t ||kstqd S )Nza z aza b cz 'abcdefghij'©r r )r Z test_listÚir r r Ú test_do_not_touch_normal_spacesx s z3TestMarkinnerspaces.test_do_not_touch_normal_spacesc C s$ t ddkstt ddks td S )Nz a 'b c' \' \'za 'b@_@c' \' \'z a "b c" \" \"za "b@_@c" \" \"rE ©r r r r Útest_one_relevant_space} s z+TestMarkinnerspaces.test_one_relevant_spacec C s$ t ddkstt ddks td S )Nza 'b c" " d' eza 'b@_@c"@_@"@_@d' eza "b c' ' d" eza "b@_@c'@_@'@_@d" erE rH r r r Útest_ignore_inner_quotes s z,TestMarkinnerspaces.test_ignore_inner_quotesc C s$ t ddkstt ddks td S )Nz a 'b c' 'd e'za 'b@_@c' 'd@_@e'z a "b c" "d e"za "b@_@c" "d@_@e"rE rH r r r Ú test_multiple_relevant_spaces
s z1TestMarkinnerspaces.test_multiple_relevant_spacesN)r r r rG rI rJ rK r r r r rD u s rD c @ s° e Zd ZdZdZe d¡Zdddddd d
|
d gZd gZ ee Z
|
d Z e e
|
D ]6\Z Zdd e d¡D Ze eje ed e¡d7 Z qHej de
|
¡dd Zej de
|
¡dd ZdS )Ú TestDimSpecaÎ This test suite tests various expressions that are used as dimension
|
specifications.
|
|
There exists two usage cases where analyzing dimensions
|
specifications are important.
|
|
In the first case, the size of output arrays must be defined based
|
on the inputs to a Fortran function. Because Fortran supports
|
arbitrary bases for indexing, for instance, `arr(lower:upper)`,
|
f2py has to evaluate an expression `upper - lower + 1` where
|
`lower` and `upper` are arbitrary expressions of input parameters.
|
The evaluation is performed in C, so f2py has to translate Fortran
|
expressions to valid C expressions (an alternative approach is
|
that a developer specifies the corresponding C expressions in a
|
.pyf file).
|
|
In the second case, when user provides an input array with a given
|
size but some hidden parameters used in dimensions specifications
|
need to be determined based on the input array size. This is a
|
harder problem because f2py has to solve the inverse problem: find
|
a parameter `p` such that `upper(p) - lower(p) + 1` equals to the
|
size of input array. In the case when this equation cannot be
|
solved (e.g. because the input array size is wrong), raise an
|
error before calling the Fortran function (that otherwise would
|
likely crash Python process when the size of input arrays is
|
wrong). f2py currently supports this case only when the equation
|
is linear with respect to unknown parameter.
|
|
z.f90a
|
function get_arr_size_{count}(a, n) result (length)
|
integer, intent(in) :: n
|
integer, dimension({dimspec}), intent(out) :: a
|
integer length
|
length = size(a)
|
end function
|
|
subroutine get_inv_arr_size_{count}(a, n)
|
integer :: n
|
! the value of n is computed in f2py wrapper
|
!f2py intent(out) n
|
integer, dimension({dimspec}), intent(in) :: a
|
if (a({first}).gt.0) then
|
print*, "a=", a
|
endif
|
end subroutine
|
Únz2*nz2:nzn/2z5 - n/2z3*n:20zn*(n+1):n*(n+5)z2*n, nz 2*n:3*n*n+2*nÚ c C s& g | ] }d |kr | d ¡d ndqS )ú:r Ú1)Úsplit)Ú.0Údr r r Ú
|
<listcomp>Æ s zTestDimSpec.<listcomp>ú,z, )ÚcountÚdimspecÚfirstrW c C sF | j |¡}t| jd| }dD ] }||\}}|j|ks"tq"d S )NÚ get_arr_size_©r r r
|
é é )Ú all_dimspecsÚindexÚgetattrr Úsizer )r rW rV Ú get_arr_sizerM Úszr r r r Útest_array_sizeÍ s
|
zTestDimSpec.test_array_sizec C s | j |¡}t| jd| }t| jd| }dD ]R}||\}}|| jkrZ|||}n||}||\} }
|
|| ks4t|||| fq4d S )NrY Zget_inv_arr_size_rZ )r] r^ r_ r Únonlinear_dimspecsr ) r rW rV ra Zget_inv_arr_sizerM rb r Zn1Zsz1Ú_r r r Útest_inv_array_size× s
|
z TestDimSpec.test_inv_array_sizeN)r r r Ú__doc__ÚsuffixÚtextwrapÚdedentZ code_templateZlinear_dimspecsrd r] ÚcodeÚ enumeraterV rW rQ ÚlstÚformatÚjoinÚpytestÚmarkÚ parametrizerc rf r r r r rL s4
|
þý
|
|
rL c @ s e Zd Zdd ZdS )ÚTestModuleDeclarationc C sP t dddd¡}t t|g¡}t|dks0t|d d d d d
|
ksLtd S ) Nr r r z foo_deps.f90r r r Zabarú=z bar('abar')r% r+ r r r Útest_dependenciesð s z'TestModuleDeclaration.test_dependenciesN)r r r ru r r r r rs ï s rs c @ s e Zd Zdd ZdS )ÚTestEvalc C sZ t j}|di dkst|di dks*t|dtddddksDt|d i d
|
ksVtd S ) NZ123z12 + 3Z15za + br r )r r# Ú3z"123"z'123')r Z _eval_scalarr Údict)r Z eval_scalarr r r Útest_eval_scalar÷ s
|
zTestEval.test_eval_scalarN)r r r ry r r r r rv ö s rv c @ s* e Zd Zej dddddg¡dd ZdS ) ÚTestFortranReaderÚencodingÚasciizutf-8zutf-16zutf-32c C s\ |d| d }|j d|d}| d¡ W 5 Q R X t t|g¡}|d d dksXtd S ) NZ input_with_z _encoding.f90r )r{ zd
|
subroutine foo()
|
end subroutine foo
|
r r5 Úfoo)ÚopenÚwriter r& r )r r, r{ Zf_pathÚffr) r r r Útest_input_encoding s
|
z%TestFortranReader.test_input_encodingN)r r r rp rq rr r r r r r rz s
|
ÿrz c @ sB e Zd Ze dddd¡gZejje j
|
d¡dkddd d
|
Z dS ) ÚTestUnicodeCommentr r r zunicode_comment.f90Zcharset_normalizerNz7test requires charset_normalizer which is not installed)Úreasonc C s | j d¡ d S )Nr
|
)r r} rH r r r Útest_encoding_comment s z(TestUnicodeComment.test_encoding_comment) r r r r r r rp rq ZskipifÚ importlibÚ find_specr r r r r r s þr )r
ÚcodecsÚ unicodedatarp Únumpyr Znumpy.f2py.crackfortranr rN r Z
|
numpy.f2pyr ri ZF2PyTestr r r2 r8 rB rD rL rs rv rz r r r r r Ú<module> s& ! f
|
|
