2i4fddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlZddlZddlZGdde j"ZGdd ZGd d ZGd d e j"ZGdde j"ZGddZGdde j"ZGddZGdde j"ZGdde j"ZGdde j"ZGddZGdde j"ZGd d!e j"ZGd"d#Z y)$N)markinnerspacesnameargspattern)util) crackfortranc>eZdZejddddgZdZy) TestNoSpacetestssrcrz gh15035.fctjgdtj}tjgdtj}|jj |tj ||dzsJ|jj ||gtj ||dzsJ|jjddk(sJy)N)r)dtyper232)nparrayfloat64modulesubballclosesubct0)selfkws Y/var/www/html/acx/venv/lib/python3.12/site-packages/numpy/f2py/tests/test_crackfortran.py test_modulezTestNoSpace.test_modules HHYbjj 1 HHYbjj 1 {{1a!e$$$ !Q {{1a!e$$${{~~d#t+++N)__name__ __module__ __qualname__rgetpathsourcesrrrr r s"t||GUNKHIG,rr c$eZdZdZdZdZdZy)TestPublicPrivatecPtjdddd}tjt|g}t |dk(sJ|d}d|dd d vsJd |dd d vsJd|dd d vsJd |dd d vsJd|dd d vsJd |dd d vsJy)Nr r rzprivatemod.f90rrprivatevarsaattrspecpublicbsetarr#rstrlen)rfpathmods rtest_defaultPrivatez%TestPublicPrivate.test_defaultPrivates We^=MN''U 53x1}}!fCK,Z8888s6{3/ ;;;;CK,Z8888s6{3/ ;;;;F F 3J ????3v;v.z::::rctjdddd}tjt|g}t |dk(sJ|d}d|dd d vsJd |dd d vsJd|dd d vsJd |dd d vsJy) Nr r rz publicmod.f90rrr)r*r+r,r-r/r0rtmp_pathr3r4s rtest_defaultPublicz$TestPublicPrivate.test_defaultPublic+s We^_M''U 53x1}}!fCK,Z8888s6{3/ ;;;;F F 3J ????3v;v.z::::rc0tjdddd}tjt|g}t |dk(sJ|dd}t |dd d d hk(sJt |d d d d hk(sJt |dd d hk(sJy)Nr r rzaccesstype.f90rrr*r+r,r)zbind(c)b_r-c)rr#rr1r2set)rr8r3r4tts rtest_access_typez"TestPublicPrivate.test_access_type5s We^=MN''U 53x1}} VF^2c7:&'Iy+AAAA2d8J'(Xy,AAAA2c7:&'H:555rctjdddd}tjt|g}t |dk(sJtj |}d|vsJy)Nr r rz gh23879.f90rbar)rr#rr1r2 crack2fortran)rr8r3r4pyfs rtest_nowrap_private_proceeduresz1TestPublicPrivate.test_nowrap_private_proceedures>sZ We^]K''U 53x1}}((-CrN)r r!r"r5r9r?rDr%rrr'r's ;;6 rr'ceZdZdZdZy)TestModuleProcedurectjdddd}tjt|g}t |dk(sJ|d}d|vrt |ddk(sJ|ddd d k(sJd |ddvsJ|ddd d d gk(sJ|ddd dk(sJd |ddvsJ|ddd dgk(sJ|ddd dk(sJd |ddvsJ|ddd ddgk(sJy)Nr r rz operators.f90rrbody namezoperator(.item.) implementedbyitem_int item_realr z operator(==)items_are_equalrz assignment(=)get_intget_realr0r7s rtest_moduleOperatorsz(TestModuleProcedure.test_moduleOperatorsFsc We^_M''U 53x1}}!f}S[!1Q!6666{1~f%);;;;#f+a.0006{1~o.  %& &&6{1~f%777#f+a.0006{1~o.3D2EEEE6{1~f%888#f+a.0006{1~o.  #$ $$rctjdddd}tjt|g}t |dk(sJ|d}|ddd d gk(sJ|dd d d gk(sJ|dd d d gk(sJy)Nr r rzpubprivmod.f90rrr*r+r,r)r.r-r/r0r7s rtest_notPublicPrivatez)TestModuleProcedure.test_notPublicPrivateXs We^=MN''U 53x1}}!f6{3 + }<<<6{3 +|;;;6{6":.8,>>>rN)r r!r"rQrSr%rrrFrFEs $$?rrFcDeZdZejddddgZdZdZy) TestExternalr r rz gh17859.fcNd}|jj|}|dk(sJy)Nc |dzSN{r%xs rincrz5TestExternal.test_external_as_statement..incrg s7NrrY)rexternal_as_statementrr\rs rtest_external_as_statementz'TestExternal.test_external_as_statementf(  KK - -d 3CxxrcNd}|jj|}|dk(sJy)Nc |dzSrXr%rZs rr\z5TestExternal.test_external_as_attribute..incrnr]rrY)rexternal_as_attributer_s rtest_external_as_attributez'TestExternal.test_external_as_attributemrbrN)r r!r"rr#r$rarfr%rrrUrUbs%t||GUNKHIGrrUc>eZdZejddddgZdZy)TestCrackFortranr r rz gh2848.f90cJ|jjdd}|dk(sJy)Nrr )rr )rgh2848)rr`s r test_gh2848zTestCrackFortran.test_gh2848ys$ KK  q! $F{{rN)r r!r"rr#r$rkr%rrrhrhus t||GUNLIJGrrhc$eZdZdZdZdZdZy)TestMarkinnerspacesc:gd}|D]}t||k(rJy)N)za z aza b cz 'abcdefghij'r)r test_listis rtest_do_not_touch_normal_spacesz3TestMarkinnerspaces.test_do_not_touch_normal_spacess)9  +A"1%* ** +rcDtddk(sJtddk(sJy)Nz a 'b c' \' \'za 'b@_@c' \' \'z a "b c" \" \"za "b@_@c" \" \"rors rtest_one_relevant_spacez+TestMarkinnerspaces.test_one_relevant_spaces,015HHHH/04FFFFrcDtddk(sJtddk(sJy)Nza 'b c" " d' eza 'b@_@c"@_@"@_@d' eza "b c' ' d" eza "b@_@c'@_@'@_@d" erorts rtest_ignore_inner_quotesz,TestMarkinnerspaces.test_ignore_inner_quotess,126NNNN126NNNNrcDtddk(sJtddk(sJy)Nz a 'b c' 'd e'za 'b@_@c' 'd@_@e'z a "b c" "d e"za "b@_@c" "d@_@e"rorts rtest_multiple_relevant_spacesz1TestMarkinnerspaces.test_multiple_relevant_spacess+/3FFFF/04HHHHrN)r r!r"rrrurwryr%rrrmrm~s+ GOIrrmc eZdZdZdZej dZgdZdgZ ee zZ dZ e e D]d\Z ZejdDcgc]}d|vr|jdd nd c}}Ze ej#e ed j%e z Z fej(j+d e dZej(j+d e dZycc}}w) 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. 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 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,:r1z, )countdimspecfirstrc|jj|}t|jd|}dD]}||\}}|j|k(rJy)N get_arr_size_rr r) all_dimspecsindexgetattrrsize)rrr get_arr_sizer|szr+s rtest_array_sizezTestDimSpec.test_array_sizes]!!''0t{{mE7,CD   A OEB66R< < rc8|jj|}t|jd|}t|jd|}dD]H}||\}}||jvr |||}n||}||\} } || k(r?J|||| fy)Nrget_inv_arr_size_r)rrrrnonlinear_dimspecs) rrrrget_inv_arr_sizer|rr+n1sz1_s rtest_inv_array_sizezTestDimSpec.test_inv_array_sizes!!''0t{{mE7,CD "4;;2CE70KL  /A OEB$111&a+&a(""%FC9 .q"b#. .9 /rN)r r!r"__doc__suffixtextwrapdedent code_templatelinear_dimspecsrrcode enumeraterrsplitlstformatjoinpytestmark parametrizerr).0ds00rr{r{s <F#HOO% M O**"%77L D#L1 w?F}}S?QR!3!8 Q4R $$))C.%    [[Y 5 6  [[Y 5/6/%Ss !Cr{ceZdZdZy)TestModuleDeclarationctjdddd}tjt|g}t |dk(sJ|dddd d k(sJy) Nr r rz foo_deps.f90rrr*abar=z bar('abar')r0r7s rtest_dependenciesz'TestModuleDeclaration.test_dependenciess\ We^^L''U 53x1}}1vf~f%c*m;;;rN)r r!r"rr%rrrrseZdZejddddgZdZy)TestFunctionReturnr r rz gh23598.f90cF|jjdddk(sJy)Nrrr)r intproductrts rtest_function_rettypez(TestFunctionReturn.test_function_rettypeJs!{{%%a+r111rN)r r!r"rr#r$rr%rrrrGs t||GUNMJKG2rrceZdZdZy)TestFortranGroupCountersctjdddd} tjt|gy#t$r}Jd|d}~wwxYw)Nr r rz gh23533.fz0'crackfortran.crackfortran' raised an exception )rr#rr1 Exception)rr3excs rtest_end_if_commentz,TestFortranGroupCounters.test_end_if_commentPsT We^[I S  % %s5zl 3 S RLSER R5 Ss: A A  AN)r r!r"rr%rrrrOsSrrceZdZdZy)TestF77CommonBlockReaderctjdddd}tjt j 5}t j t|g}ddddjvsJy#1swYxYw)Nr r rz gh22648.pyfMismatch) rr# contextlibredirect_stdoutioStringIOrr1getvalue)rr8r3 stdout_f2pyr4s r test_gh22648z%TestF77CommonBlockReader.test_gh22648Zsn We^]K  ' ' 6 :+++SZL9C :!5!5!7777 : :s A==BN)r r!r"rr%rrrrYs8rr)!rcodecsr unicodedatarnumpyrnumpy.f2py.crackfortranrrr}r numpy.f2pyrrrrF2PyTestr r'rFrUrhrmr{rrrrrrrrr%rrrs  D#  ,$-- ,% % N??:4==&t}}II*`/$--`/F<<3t}}3 ' '"E"EJ22St}}S88r