a kh@sHddlmZddlmZddlmZmZmZddlm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZddlmZddlmZdd lmZmZdd lmZdd lmZm Z dd l!m"Z"m#Z#dd l$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCddlmDZDddlmEZEddlFmGZGmHZHmIZImJZJmKZKddlLmMZMmNZNddlOmPZPmQZQddlRmSZSmTZTddlUmVZVmWZWddlXmYZYddlZm[Z[ddZ\d d!Z]d"d#Z^d$d%Z_d&d'Z`d(d)Zad*d+Zbd,d-Zcd.d/Zdd0d1Zed2d3Zfd4d5Zgd6d7Zhd8d9Zid:d;Zjdd?Zld@dAZmdBdCZndDdEZodFdGZpdHdIZqdJdKZrdLdMZsdNdOZtdPdQZudRdSZvdTdUZwdVdWZxdXdYZydZd[Zzd\d]Z{d^d_Z|d`daZ}dbdcZ~dddeZdfdgZdhdiZdjdkZdldmZdndoZdpdqZdrdsZdtS)u)Add)Expr)FunctionLambdadiff)Mod)Catalan EulerGamma GoldenRatio)EFloatIIntegerRationalpi)Eq)S)Dummysymbols) factorial) conjugatesign)explog)sqrt) Piecewise)atan2cossin)gamma)Integral)Range)For Assignment aug_assign) DeclarationVariablefloat32float64 value_constrealbool_WhileFunctionPrototypeFunctionDefinitionintegerReturnElement)UnevaluatedExpr) Relational)AndOrNot EquivalentXor)Matrix MatrixSymbol)fcode FCodePrinter) IndexedBaseIdx) ArraySymbol ArrayElement)implemented_function)raisesc Cstddd\}}}t||}tt||}t|dddksBJtd\}}}t||}tt||} t| dd d d ksJt| d d iddksJdS)Nzp q rT)r*free source_formatzexp(p + (q + r))x y zFhumanzexp(re(x) + re(y + z))reZrealpartZuser_functionsz"exp(realpart(x) + realpart(y + z)))rr2absrr;lstrip) pqrZq_rexprxyzZy_zZexpr2rUO/var/www/auris/lib/python3.9/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExpr#s  rWcCs0td}Gdddt}t||dks,JdS)NrRc@seZdZddZdS)ztest_printmethod..nintcSsd||jdS)Nznint(%s)r)_printargs)selfprinterrUrUrV_fcode3sz%test_printmethod..nint._fcodeN)__name__ __module__ __qualname__r\rUrUrUrVnint2sr`z nint(x))rrr;)rRr`rUrUrVtest_printmethod/sracstdtddd}tddd}ttddd d ks:Jtt|ddd d ksTJtt|ddd d ksnJttfd ddS)NrRrSTr/rT)complex_rCstandardrEz#merge(0d0, dsign(1d0, x), x == 0d0)zmerge(0, isign(1, y), y == 0)z/merge(cmplx(0d0, 0d0), z/abs(z), abs(z) == 0d0)cs ttSN)r;rrUrRrUrV?z!test_fcode_sign..)rr;rrBNotImplementedError)rSrTrUrhrVtest_fcode_sign8s  rlcCs,td\}}tddd}t|ddks,Jt||ddksDJtdt|d ||||d |d ksxJtt|d ksJtt|d ksJt|dd ksJtt|d ksJttddksJt|ddksJt|dddddks Jt|tdddks(JdS)Nx,ynTrbz x**3z x**(y**3)g @rIz, (3.5d0*sin(x))**(-x + y**x)/(x**2 + y)z sqrt(x)z sqrt(dble(n))g? z sqrt(10.0d0)gz 1d0/xgrSrCrDzy = x**(-2.0d0)z x**(3.0d0/7.0d0))rr;rrr)rRrSrnrUrUrVtest_fcode_PowBs  *rscCstd}ttdddksJttdddks4Jttddd ksJJttd ddks`Jt|tddd kszJttdd|d ksJdS) NrRrorrz 3.0d0/7.0d0 z 2iz -3.0d0/7.0d0z x + 3.0d0/7.0d0z (3.0d0/7.0d0)*x)rr;rrhrUrUrVtest_fcode_RationalTsrwcCs,ttddksJttddks(JdS)NCz 67z -1)r;rrUrUrUrVtest_fcode_Integer^srzcCs,ttddksJttddks(JdS)NgE@z 42.0000000000000d0g@xz -1.00000000000000d+20)r;r rUrUrUrVtest_fcode_Floatcsr{cstd\tttdks(JttfddttfddttfddttfdddD]6}tt|d d ksJt|d d ks|Jq|dS) Nrmz sin(x)**cos(y)csttddSNBrfr;rrUrRrSrUrVrikrjz&test_fcode_functions..cstddSr|r;rUrrUrVrilrjcsttddSNMr~rrUrrUrVrimrjcstddSrrrUrrUrVrinrj)Zrdiir~z modulo(x, y))rr;rrrBrkrr~rUrrVtest_fcode_functionshs rc Cst}td\}}}}}}}tt|t||t||dksHJtt|d|||tdddksvJtt|t||t||dddksJt|t|ddd ksJ|j|t||d d d ksJ|j|t|d d dksJ|j|dd dksJtddd\}} td} td}td}t d| } t d|} t| | | f|| || dddksJdS)Nzx,x_,x__,y,X,X_,Yz& exp(x_) + sin(x*y) + cos(X__*Y_)rIrrz( 2*X_**(7.0d0/2.0d0)*Y*x + exp(x__)F)Z name_manglingz# exp(x_) + sin(x*y) + cos(X*Y)z x - cos(X)me assign_toz me = X*sin(x_) + x__muz mu = X*sin(x_)adz ad = x__n,mTrbArRrSir rCrrEzndo i = 1, m y(i) = 0 end do do i = 1, m do I_ = 1, n y(i) = A(i, I_)*x(I_) + y(i) end do end do) r<rr;rrrrdoprintr=r>) obrRZx_Zx__rSXZX_Yrnmrrr rUrUrV test_casets0"$&"    rcCstd}tdd}d}t|tdd|ks6Jt|tdd|ksRJt|ttdd|ksrJtttdd|ksJttddtddksJt|ttdd|ksJt|tttdd|ksJdS)NrRrqz0.8340324452479558d0z x*%sd0z %sd0z x*%s)rrevalfr;rr)rRZlog10_17Z loglog10_17rUrUrV"test_fcode_functions_with_integerss " rcCsd}t}ttdt|ks$Jttdt|ks>Jttdt|ksXJttdt|ksrJttdt|ksJttdddtdksJttd d t|t|fht d fksJttd d t|t|fht d fksJttd d t|t|fht d fks8Jttd d t|t|fht dfkshJttd d t|t|fht dfksJttdd dt|tdfht dfksJdS)Nrz. parameter (Catalan = %sd0) Catalanz4 parameter (EulerGamma = %sd0) EulerGammaz" parameter (E = %sd0) Ez6 parameter (GoldenRatio = %sd0) GoldenRatioz$ parameter (pi = %sd0) pi) precisionFrGz Catalanz EulerGammaz Ez GoldenRatioz pi)rrH) r<r;rrr r r rrXset)precrNrUrUrVtest_fcode_NumberSymbolsH          rcCsttdksJtd}tdtdks,JtddtdksDJtddt|dks`Jtt|dkstJtddt|d ksJtdd d }td |d ksJtt|dksJtd|dksJdS)Nz cmplx(0,1)rRz cmplx(0,4)roz cmplx(3,4)z cmplx(3,4) + xz cmplx(0,1)*xz cmplx(3,4) - xT)Z imaginaryrz 5*xz x + 3)r;r rrhrUrUrVtest_fcode_complexs rcCsNtd\}}tt|dks Jtt||dks6Jtt|dksJJdS)Nrmz sin(x)z atan2(x, y)z conjg(x))rr;rrrrrUrUrV test_implicits rcCstd}td}tttt|Wdn1s:0Yttt|dddks`Jttt||Wdn1s0YdS)NrRgF)strictzHC Not supported in Fortran: C Integral Integral(sin(x), x))rrrBrkr;rr r)rRrrUrUrVtest_not_fortrans * rcCstd}tt|ddiddks$Jtd}tt|ddiddksHJtd }t||d d idd kslJtd d d}tt|ddiddksJdS)NrRrZzsinrKz zsin(x)rZmygammaz mygamma(x)rZgreatz great(x)rnTrbrZfctz fct(n))rr;rrrr)rRrrnrUrUrVtest_user_functionss    rcCstd}tdt|d|}t||dks0Jtdt|dt|}t||dtdksfJtd}tdtd d d }tdt||d |d|}t|||||d dksJdS)NrRrrIz 2*xz( parameter (pi = %sd0) 2*pi/xrrrrnTrbrprzI do i = 1, n A(i) = (A(i) + 1)*(A(i) + 2)*A(i) end do)rrArr;rrr=r>)rRrrrrUrUrVtest_inline_functions   rcCs$td}tt|dddks JdS)NrRsrz s = sin(x))rr;rrhrUrUrVtest_assign_tosrcsZtd\}t|ddddks,JfddtdD}tt|d ksVJdS) Nrmrqvarrz var = x**10 + 10*x**9*y + 45*x**8*y**2 + 120*x**7*y**3 + 210*x**6* @ y**4 + 252*x**5*y**5 + 210*x**4*y**6 + 120*x**3*y**7 + 45*x**2*y @ **8 + 10*x*y**9 + y**10csg|] }|qSrUrU).0rrhrUrV rjz&test_line_wrapping.. zR x**10 + x**9 + x**8 + x**7 + x**6 + x**5 + x**4 + x**3 + x**2 + x @ + 1)rr;expandranger)rSerUrhrVtest_line_wrappings  rcCstd\}}tt||k||dkdddks2Jtt||k||dkdddksXJtt||k||dkddddd ksJtt||k||dkddd ksJdS) Nx yrprCrDzx < y .and. y < x + 1zx < y .or. y < x + 1Fevaluatezx < y .neqv. y < x + 1zx < y .eqv. y < x + 1)rr;r4r5r8r7rrUrUrVtest_fcode_precedence s rcCstd\}}}tt|dddks&Jtt||dddks@Jtt|t|dddks^Jttt||dddks|Jttt|t|dddksJttt||d d ddd ksJtt||ddd ksJtt|t|ddd ksJttt||dddksJttt|t|dddks=zx >= yz<=zx <= y>zx > y.rdr~z merge(x, x**2, x < 1)rrzR if (x < 1) then var = x else var = x**2 end ifrqa/ if (x < 0) then weird_name = -cos(x)/x + 10*sin(x)/x**2 + 90*cos(x)/x**3 - 720* @ sin(x)/x**4 - 5040*cos(x)/x**5 + 30240*sin(x)/x**6 + 151200*cos(x @ )/x**7 - 604800*sin(x)/x**8 - 1814400*cos(x)/x**9 + 3628800*sin(x @ )/x**10 + 3628800*cos(x)/x**11 else weird_name = -sin(x)/x - 10*cos(x)/x**2 + 90*sin(x)/x**3 + 720* @ cos(x)/x**4 - 5040*sin(x)/x**5 - 30240*cos(x)/x**6 + 151200*sin(x @ )/x**7 + 604800*cos(x)/x**8 - 1814400*sin(x)/x**9 - 3628800*cos(x @ )/x**10 + 3628800*sin(x)/x**11 end ifrZ weird_namez1 merge(x, merge(x**2, sin(x), x > 1), x < 1)cstSrgrrUrrUrVrirj) rrrBrkr;rrrr ValueError)rRcodeexpectedabrrUrrVtest_fcode_Piecewises.          0,rcCsrt}gd}||}gd}|D]}t|dks$Jq$t||D]\}}||ksDJqDt|t|ksnJdS)N)zbC This is a long comment on a single line that must be wrapped properly to produce nice outputzf this = is + a + long + and + nasty + fortran + statement + that * must + be + wrapped + properlyzg this = is + a + long + and + nasty + fortran + statement + that * must + be + wrapped + properlyzh this = is + a + long + and + nasty + fortran + statement + that * must + be + wrapped + properlyzd this = is + a + long + and + nasty + fortran + statement + that*must + be + wrapped + properlyzf this = is + a + long + and + nasty + fortran + statement + that*must + be + wrapped + properlyzg this = is + a + long + and + nasty + fortran + statement + that*must + be + wrapped + properlyzh this = is + a + long + and + nasty + fortran + statement + that*must + be + wrapped + properlyze this = is + a + long + and + nasty + fortran + statement + that**must + be + wrapped + properlyzf this = is + a + long + and + nasty + fortran + statement + that**must + be + wrapped + properlyzg this = is + a + long + and + nasty + fortran + statement + that**must + be + wrapped + properlyzh this = is + a + long + and + nasty + fortran + statement + that**must + be + wrapped + properlyzi this = is + a + long + and + nasty + fortran + statement + that**must + be + wrapped + properlyzc this = is + a + long + and + nasty + fortran + statement(that)/must + be + wrapped + properlyzg this = is + a + long + and + nasty + fortran + statement(that)/must + be + wrapped + properly)zBC This is a long comment on a single line that must be wrappedz%C properly to produce nice outputzG this = is + a + long + and + nasty + fortran + statement + that *% @ must + be + wrapped + properlyzH this = is + a + long + and + nasty + fortran + statement + that *rG this = is + a + long + and + nasty + fortran + statement + thatz' @ * must + be + wrapped + properlyzF this = is + a + long + and + nasty + fortran + statement + that*rzH this = is + a + long + and + nasty + fortran + statement + that*rH this = is + a + long + and + nasty + fortran + statement + thatz& @ *must + be + wrapped + properly@ this = is + a + long + and + nasty + fortran + statement +z* @ that*must + be + wrapped + properlyzG this = is + a + long + and + nasty + fortran + statement + that**rzH this = is + a + long + and + nasty + fortran + statement + that**rr' @ **must + be + wrapped + properlyrrrz+ @ that**must + be + wrapped + properlyzE this = is + a + long + and + nasty + fortran + statement(that)/rzH this = is + a + long + and + nasty + fortran + statement(that)z& @ /must + be + wrapped + properlyH)r< _wrap_fortranlenzip)r[linesZ wrapped_linesZexpected_lineslinewrrUrUrVtest_wrap_fortrans  rcCs,t}gd}gd}|||ks(JdS)N)H this_variable_is_very_long_because_we_try_to_test_line_break=1.0d0zI this_variable_is_very_long_because_we_try_to_test_line_break =1.0d0zK this_variable_is_very_long_because_we_try_to_test_line_break = 1.0d0zL this_variable_is_very_long_because_we_try_to_test_line_break = 1.0d0zM this_variable_is_very_long_because_we_try_to_test_line_break = 1.0d0zK this_variable_is_very_long_because_we_try_to_test_line_break = 10.0d0) rD this_variable_is_very_long_because_we_try_to_test_line_break = @ 1.0d0zE this_variable_is_very_long_because_we_try_to_test_line_break =rzF this_variable_is_very_long_because_we_try_to_test_line_break =rzG this_variable_is_very_long_because_we_try_to_test_line_break =rrz @ 10.0d0r<rr[rrrUrUrVtest_wrap_fortran_keep_d0s rcCsttdddS)NcSsttdddS)Nrgarbage)method)r;rrUrUrUrVri6rjztest_settings..)rB TypeErrorrUrUrUrV test_settings5srcCs0td\}}tt|t|dddks,JdS)NrmrCrDzsin(y) + cos(x))rr;rrrrUrUrVtest_free_form_code_line9s rcCs@td\}}tt|t|ddd}d}||kss   rcCs0tddi}dg}ddg}|||ks,JdS)NrErCz^! This is a long comment on a single line that must be wrapped properly to produce nice outputzG! This is a long comment on a single line that must be wrapped properlyz! to produce nice outputrrrUrUrVtest_free_form_comment_lineIs  rc Cstddd\}}td}td}td}td|}td|}d }t|||f||||d d }||d d iks||d diks||d diks||d diksJdS)NrTrbrrRrSrjz_do i = 1, m y(i) = 0 end do do i = 1, m do j = 1, n y(i) = %(rhs)s end do end dorCrrhszy(i) + A(i, j)*x(j)zy(i) + x(j)*A(i, j)zx(j)*A(i, j) + y(i)zA(i, j)*x(j) + y(i))rr=r>r;) rnrrrRrSrrrrrUrUrV test_loopsRs    "rcCsftddtd\}}td}td}t||}d|jj|jd}t||||dd }||ksbJdS) Nzi mT)r/clsrRrSzMdo i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do)ZicountZmcountrCr)rrr=r>labelZ dummy_indexr;)rrrRrSrrrUrUrVtest_dummy_loopsls rcCsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}t|j|jdd }|d sJdS) NrrSshaperRDyrprF)rZcontract+Dy(i) = (y(i + 1) - y(i))/(x(i + 1) - x(i)))r=r>rr;rlhsendswithZlen_yrSrRrrrZcode0rUrUrV2test_fcode_Indexed_without_looking_for_contraction{s2rcsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}tt|j|j}|dsJGd d d tt | fd d | t D}t||||d||||d||}tt|j|j}|dsJdS) NrrSrrRrrprrc@s eZdZdS)z.test_element_like_objects..ElementExprN)r]r^r_rUrUrUrV ElementExprsrc3s |]}||j|jfVqdSrg)nameindices)rrrrUrV rjz,test_element_like_objects..) r?r>rr;r#rrrr1rsubsZatomsr@rrUrrVtest_element_like_objectss22rcCs:Gdddt}|}td}|t|ddks6JdS)Nc@seZdZejZdS)z1test_derived_classes..MyFancyFCodePrinterN)r]r^r_r<Z_default_settingscopyrUrUrUrVMyFancyFCodePrintersrrRZborkz bork = sin(x))r<rrr)rr[rRrUrUrVtest_derived_classessrcCs.d}d}tddi}||}||ks*JdS)Nausubroutine test(a) integer :: a, i, j do do do j = 1, 5 if (a>b) then if(b>0) then a = 3 donot_indent_me = 2 do_not_indent_me_either = 2 ifIam_indented_something_went_wrong = 2 if_I_am_indented_something_went_wrong = 2 end should not be unindented here end if endif end do end do enddo end subroutine subroutine test2(a) integer :: a do a = a + 1 end do end subroutine asubroutine test(a) integer :: a, i, j do do do j = 1, 5 if (a>b) then if(b>0) then a = 3 donot_indent_me = 2 do_not_indent_me_either = 2 ifIam_indented_something_went_wrong = 2 if_I_am_indented_something_went_wrong = 2 end should not be unindented here end if endif end do end do enddo end subroutine subroutine test2(a) integer :: a do a = a + 1 end do end subroutine rErC)r<Z indent_code)Z codelinesrrNrrUrUrV test_indents  rc Cs.td\}}}t||td||dkf|dft|g}tddd}t||dksXJtd|d |dkf|d dft|d |d }t|d d dksJtddd}tddd}tt|d dt|d g|d |d |ddgd|d|d t|d ddgg}t||dks*JdS)Nzx,y,zrIrTrrorpz A(1, 1) = x*y if (y > 0) then A(2, 1) = x + 2 else A(2, 1) = y end if A(3, 1) = sin(z))rIr)rprrrrdr~z? merge(2*A(3, 1), A(3, 1), x > 0) + sin(A(2, 1)) + A(1, 1)rOrM)ror)rrrz M(1, 1) = sin(q(2, 1)) M(2, 1) = q(2, 1) + q(3, 1) M(3, 1) = 2*q(5, 1)/q(2, 1) M(1, 2) = 0 M(2, 2) = q(4, 1) M(3, 2) = sqrt(q(1, 1)) + 4 M(1, 3) = cos(q(3, 1)) M(2, 3) = 5 M(3, 3) = 0)rr9rrr:r;rr) rRrSrTmatrrQrOrrrUrUrVtest_Matrix_printings&,  6   $rcCsDtd\}}t|tdddt|||g}t|}|dks@JdS)NrrrqrIz2 do x = 0, 9, 2 y = x*y end do)rr"r!r#r;)rRrSfZsolrUrUrVtest_fcode_Fors  rc Csdd}tddd}t|}t|}||dtd\}}t|tdthd }t|}||d t|td }t|} || d |td |td|td ttid|tdttiddS)Nc[s"t|fddd||ksJdS)NrdrCrer)rQrefkwargsrUrUrVchecksz%test_fcode_Declaration..checkrTrbzinteger*4 :: ir*)valueattrszreal*4, parameter :: x = 42typez logical :: yzreal*4zreal*8)Z type_aliases) rr&Zdeducedr%r'r)r+r(r*) rrZvar1Zdcl1rRrSZvar2Zdcl2bZvar3Zdcl3rUrUrVtest_fcode_Declarations          rcCsxtddd}tddd}tddd}t|ddks8Jtd|ddksPJ|d|||}t|d kstJdS) NrrproBCrz A(1, 1)z 3*A(1, 1)z (A - B)(1, 1))r:r;r)rrrFrUrUrVtest_MatrixElement_printing's   rcCs(td}tt|dddddks$JdS)NrR+rprCrDz x = x + 1)rr;r$rhrUrUrVtest_aug_assign4srcCs8td}ttt|dkt|ddgdddks4JdS)NrRrp-rCrDz)do while (abs(x) > 1) x = x - 1 end do)rr;r,rLr$rhrUrUrV test_While9s"r csRtd}tddd}t|td}t|td}ttd||gttfdddS) NrRrnTrbrpowercstSrgrrUfp1rUrVriJrjz.test_FunctionPrototype_print..)rr&r*r/r-rBrk)rRrnvxvnrUr rVtest_FunctionPrototype_printBs    rcsjtd}tddd}t|td}t|td}t|||t|g}ttd||g|ttfdddS) NrRrnTrbrr cstSrgrrUZfd1rUrVriVrjz/test_FunctionDefinition_print..) rr&r*r/r#r0r.rBrk)rRrnrrbodyrUrrVtest_FunctionDefinition_printMs   rN)Zsympy.core.addrZsympy.core.exprrZsympy.core.functionrrrZsympy.core.modrZ sympy.corerr r Zsympy.core.numbersr r r rrrZsympy.core.relationalrZsympy.core.singletonrZsympy.core.symbolrrZ(sympy.functions.combinatorial.factorialsrZ$sympy.functions.elementary.complexesrrZ&sympy.functions.elementary.exponentialrrZ(sympy.functions.elementary.miscellaneousrZ$sympy.functions.elementary.piecewiserZ(sympy.functions.elementary.trigonometricrrrZ'sympy.functions.special.gamma_functionsrZsympy.integrals.integralsr Zsympy.sets.fancysetsr!Z sympy.codegenr"r#r$Zsympy.codegen.astr%r&r'r(r)r*r+r,r-r.r/r0r1r2r3Zsympy.logic.boolalgr4r5r6r7r8Zsympy.matricesr9r:Zsympy.printing.fortranr;r<Z sympy.tensorr=r>Zsympy.tensor.array.expressionsr?r@Zsympy.utilities.lambdifyrAZsympy.testing.pytestrBrWrarlrsrwrzr{rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrUrUrUrVs            <             1c +<    ?#