o g%@sddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl Z Gddde j ZGd d d ZGd d d e j ZGd dde j ZGdddZGddde j ZGdddZdS)N)assert_array_equal assert_equal)markinnerspaces)util) crackfortranc@seZdZdZddZdS) TestNoSpacea| subroutine subb(k) real(8), intent(inout) :: k(:) k=k+1 endsubroutine subroutine subc(w,k) real(8), intent(in) :: w(:) real(8), intent(out) :: k(size(w)) k=w+1 endsubroutine function t0(value) character value character t0 t0 = value endfunction cCsxtjgdtjd}tjgdtjd}|j|t||d|j||gt||d|jddks:JdS)N)r)dtyper2)nparrayfloat64modulesubbrsubct0)selfkwr`/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/numpy/f2py/tests/test_crackfortran.py test_module s zTestNoSpace.test_moduleN)__name__ __module__ __qualname__coderrrrrr s rc@seZdZddZddZdS)TestPublicPrivatecCs|d}|d}|tdWdn1swYtt|g}t|dks1J|d}d|ddd vsAJd |ddd vsMJd|dd d vsYJd |dd d vseJd|dd d vsqJd |dd d vs}JdS) Nmod.f90ra> module foo private integer :: a public :: setA integer :: b contains subroutine setA(v) integer, intent(in) :: v a = v end subroutine setA end module foo rrprivatevarsaattrspecpublicbsetaopenwritetextwrapdedentrstrlenrtmp_pathf_pathffmodrrrtest_defaultPrivate,s z%TestPublicPrivate.test_defaultPrivatecCs|d}|d}|tdWdn1swYtt|g}t|dks1J|d}d|ddd vsAJd |ddd vsMJd|dd d vsYJd |dd d vseJdS) Nr ra+ module foo public integer, private :: a public :: setA contains subroutine setA(v) integer, intent(in) :: v a = v end subroutine setA end module foo rrr!r"r#r$r%r'r(r/rrrtest_defaultPublicFs  z$TestPublicPrivate.test_defaultPublicN)rrrr4r5rrrrr*s rc@s eZdZdZddZddZdS) TestExternalaY integer(8) function external_as_statement(fcn) implicit none external fcn integer(8) :: fcn external_as_statement = fcn(0) end integer(8) function external_as_attribute(fcn) implicit none integer(8), external :: fcn external_as_attribute = fcn(0) end cC$dd}|j|}|dksJdS)NcS|dSN{rxrrrincrpz5TestExternal.test_external_as_statement..incrr:)rexternal_as_statementrr=rrrrtest_external_as_statemento z'TestExternal.test_external_as_statementcCr7)NcSr8r9rr;rrrr=vr>z5TestExternal.test_external_as_attribute..incrr:)rexternal_as_attributer@rrrtest_external_as_attributeurCz'TestExternal.test_external_as_attributeN)rrrrrBrErrrrr6^s r6c@s"eZdZdZedZddZdS)TestCrackFortran.f90a subroutine gh2848( & ! first 2 parameters par1, par2,& ! last 2 parameters par3, par4) integer, intent(in) :: par1, par2 integer, intent(out) :: par3, par4 par3 = par1 par4 = par2 end subroutine gh2848 cCs|jdd}|dks JdS)Nrr )rr )rgh2848)rrArrr test_gh2848szTestCrackFortran.test_gh2848N)rrrsuffixr+r,rrIrrrrrF|s  rFc@s,eZdZddZddZddZddZd S) TestMarkinnerspacescCs$gd}|D] }tt||qdS)N)za z aza b cz 'abcdefghij'rr)r test_listirrrtest_do_not_touch_normal_spacessz3TestMarkinnerspaces.test_do_not_touch_normal_spacescC ttddttdddS)Nz a 'b c' \' \'za 'b@_@c' \' \'z a "b c" \" \"za "b@_@c" \" \"rLrrrrtest_one_relevant_spacez+TestMarkinnerspaces.test_one_relevant_spacecCrP)Nza 'b c" " d' eza 'b@_@c"@_@"@_@d' eza "b c' ' d" eza "b@_@c'@_@'@_@d" erLrQrrrtest_ignore_inner_quotess z,TestMarkinnerspaces.test_ignore_inner_quotescCrP)Nz a 'b c' 'd e'za 'b@_@c' 'd@_@e'z a "b c" "d e"za "b@_@c" "d@_@e"rLrQrrrtest_multiple_relevant_spacesrSz1TestMarkinnerspaces.test_multiple_relevant_spacesN)rrrrOrRrTrUrrrrrKs  rKc@seZdZdZdZedZgdZdgZ ee Z dZ e e D]\Z Ze eje edevr2eddnd d 7Z qejd e d d Zejd e ddZdS) TestDimSpecaThis 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. rGa 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)z 2*n:3*n*n+2*n:r1)countdimspecfirstr\cCsH|j|}t|jd|}dD]}||\}}t||ks!JqdS)N get_arr_size_rr r ) all_dimspecsindexgetattrrr.)rr\r[ get_arr_sizerWszr#rrrtest_array_sizes  zTestDimSpec.test_array_sizec Cs|j|}t|jd|}t|jd|}dD])}||\}}||jvr-|||}n||}||\} } || ksCJ|||| fqdS)Nr^get_inv_arr_size_r_)rbrcrdrnonlinear_dimspecs) rr\r[reget_inv_arr_sizerWrfr#n1sz1_rrrtest_inv_array_sizes     zTestDimSpec.test_inv_array_sizeN)rrr__doc__rJr+r, code_templatelinear_dimspecsrirbr enumerater[r\formatsplitpytestmark parametrizergrnrrrrrVs"    rVc@seZdZddZdS)TestModuleDeclarationcCs|d}|d}|tdWdn1swYtt|g}t|dks1J|ddddd ks?JdS) Nr rz module foo type bar character(len = 4) :: text end type bar type(bar), parameter :: abar = bar('abar') end module foo rrr"abar=z bar('abar')r(r/rrrtest_dependenciess   z'TestModuleDeclaration.test_dependenciesN)rrrr{rrrrrx s rx)runumpyr numpy.testingrrnumpy.f2py.crackfortranrrXr numpy.f2pyrr+F2PyTestrrr6rFrKrVrxrrrrs    4a