U Zʹj@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|dddksBttd\}}}t||}tt||} t| dd d d kstt| d d iddkstdS)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;AssertionErrorlstrip) pqrZq_rexprxyzZy_zZexpr2rV`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExpr#s  rXcCs0td}Gdddt}t||dks,tdS)NrSc@seZdZddZdS)ztest_printmethod..nintcSsd||jdS)Nznint(%s)r)_printargs)selfprinterrVrVrW_fcode3sz%test_printmethod..nint._fcodeN)__name__ __module__ __qualname__r]rVrVrVrWnint2sraz nint(x))rrr;rM)rSrarVrVrWtest_printmethod/srbcstdtddd}tddd}ttddd d ks:ttt|ddd d ksTttt|ddd d ksntttfd ddS)NrSrTTr/rU)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;rrVrSrVrW?z!test_fcode_sign..)rr;rrMrBNotImplementedError)rTrUrVrirWtest_fcode_sign8s  rmcCs,td\}}tddd}t|ddks,tt||ddksDttdt|d ||||d |d ksxttt|d ksttt|d kstt|dd ksttt|d kstttddkstt|ddkstt|dddddks tt|tdddks(tdS)Nx,ynTrcz 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/xgrTrCrDzy = x**(-2.0d0)z x**(3.0d0/7.0d0))rr;rMrrr)rSrTrorVrVrWtest_fcode_PowBs  *rtcCstd}ttdddkstttdddks4tttddd ksJtttd ddks`tt|tddd ksztttdd|d kstdS) NrSrprsz 3.0d0/7.0d0 z 2iz -3.0d0/7.0d0z x + 3.0d0/7.0d0z (3.0d0/7.0d0)*x)rr;rrMrirVrVrWtest_fcode_RationalTsrxcCs,ttddkstttddks(tdS)NCz 67z -1)r;rrMrVrVrVrWtest_fcode_Integer^sr{cCs,ttddkstttddks(tdS)NgE@z 42.0000000000000d0g@xz -1.00000000000000d+20)r;r rMrVrVrVrWtest_fcode_Floatcsr|cstd\tttdks(tttfddttfddttfddttfdddD]6}tt|d d kstt|d d ks|tq|dS) Nrnz sin(x)**cos(y)csttddSNBrgr;rrVrSrTrVrWrjkrkz&test_fcode_functions..cstddSr}r;rVrrVrWrjlrkcsttddSNMrrrVrrVrWrjmrkcstddSrrrVrrVrWrjnrk)Zreiirz modulo(x, y))rr;rrrMrBrlrrrVrrWtest_fcode_functionshs rc Cst}td\}}}}}}}tt|t||t||dksHttt|d|||tdddksvttt|t||t||dddkstt|t|ddd kst|j|t||d d d kst|j|t|d d dkst|j|dd dksttddd\}} t d} t d}t d}t d| } t d|} t| | | f|| || dddkstdS)Nzx,x_,x__,y,X,X_,Yz& exp(x_) + sin(x*y) + cos(X__*Y_)rIrsz( 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,mTrcArSrTir 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;rrrrMrdoprintr=r>) obrSZx_Zx__rTXZX_Yromrrr rVrVrW test_casets0"$&"    rcCstd}tdd}d}t|tdd|ks6tt|tdd|ksRtt|ttdd|ksrttttdd|kstttddtddkstt|ttdd|kstt|tttdd|kstdS)NrSrrz0.8340324452479558d0z x*%sd0z %sd0z x*%s)rrevalfr;rMrr)rSZlog10_17Z loglog10_17rVrVrW"test_fcode_functions_with_integerss " rcCsd}t}ttdt|ks$tttdt|ks>tttdt|ksXtttdt|ksrtttdt|kstttdddtdkstttd d t| t|fht d fkstttd d t| t|fht d fkstttd d t| t|fht d fks8tttd d t| t|fht dfkshtttd d t| t|fht dfkstttdd dt| tdfht dfkstdS)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;rrrMr r r rrYset)precrOrVrVrWtest_fcode_NumberSymbols\          rcCsttdksttd}tdtdks,ttddtdksDttddt|dks`ttt|dkstttddt|d ksttdd d }td |d ksttt|dksttd|dkstdS)Nz cmplx(0,1)rSz cmplx(0,4)rpz cmplx(3,4)z cmplx(3,4) + xz cmplx(0,1)*xz cmplx(3,4) - xT)Z imaginaryrz 5*xz x + 3)r;r rMrrirVrVrWtest_fcode_complexs rcCsNtd\}}tt|dks ttt||dks6ttt|dksJtdS)Nrnz sin(x)z atan2(x, y)z conjg(x))rr;rrMrrrrVrVrW test_implicits rc Csptd}td}tttt|W5QRXttt|dddksLtttt||W5QRXdS)NrSgF)strictzHC Not supported in Fortran: C Integral Integral(sin(x), x)) rrrBrlr;rr rrM)rSrrVrVrWtest_not_fortrans  rcCstd}tt|ddiddks$ttd}tt|ddiddksHttd }t||d d idd kslttd d d}tt|ddiddkstdS)NrSrZzsinrKz zsin(x)rZmygammaz mygamma(x)rZgreatz great(x)roTrcrZfctz fct(n))rr;rrMrrr)rSrrorVrVrWtest_user_functionss$ rcCstd}tdt|d|}t||dks0ttdt|dt|}t||dtdksfttd}tdtd d d }tdt||d |d|}t|||||d dkstdS)NrSrrIz 2*xz( parameter (pi = %sd0) 2*pi/xrrrroTrcrqrzI do i = 1, n A(i) = (A(i) + 1)*(A(i) + 2)*A(i) end do) rrArr;rMrrr=r>)rSrrrrVrVrWtest_inline_functions   rcCs$td}tt|dddks tdS)NrSsrz s = sin(x))rr;rrMrirVrVrWtest_assign_tosrcsZtd\}t|ddddks,tfddtdD}tt|d ksVtdS) Nrnrrvarrz 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|] }|qSrVrV).0rrirVrW sz&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;expandrMranger)rTerVrirWtest_line_wrappings  rcCstd\}}tt||k||dkdddks2ttt||k||dkdddksXttt||k||dkddddd ksttt||k||dkddd kstdS) Nx yrqrCrDzx < y .and. y < x + 1zx < y .or. y < x + 1Fevaluatezx < y .neqv. y < x + 1zx < y .eqv. y < x + 1)rr;r4rMr5r8r7rrVrVrWtest_fcode_precedence s rcCstd\}}}tt|dddks&ttt||dddks@ttt|t|dddks^tttt||dddks|tttt|t|dddkstttt||d d ddd ksttt||ddd ksttt|t|ddd kstttt||dddkstttt|t|dddks=zx >= yz<=zx <= y>zx > y.rerz merge(x, x**2, x < 1)rrzR if (x < 1) then var = x else var = x**2 end ifrra/ 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)cstSrhrrVrrVrWrjrk) rrrBrlr;rMrrrr ValueError)rScodeexpectedabrrVrrWtest_fcode_Piecewises.          0,rcCst}ddddddddd d d d d ddg}||}dddddddddddddddddddddddddd d!dd"d#g}|D]}t|d$ksvtqvt||D]\}}||kstqt|t|kstdS)%NzbC 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 + properlyzBC 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 *z% @ must + be + wrapped + properlyzH this = is + a + long + and + nasty + fortran + statement + that *zG this = is + a + long + and + nasty + fortran + statement + thatz' @ * must + be + wrapped + properlyzF this = is + a + long + and + nasty + fortran + statement + that*zH this = is + a + long + and + nasty + fortran + statement + that*zH this = is + a + long + and + nasty + fortran + statement + thatz& @ *must + be + wrapped + properlyz@ this = is + a + long + and + nasty + fortran + statement +z* @ that*must + be + wrapped + properlyzG this = is + a + long + and + nasty + fortran + statement + that**zH this = is + a + long + and + nasty + fortran + statement + that**z' @ **must + be + wrapped + properlyz+ @ that**must + be + wrapped + properlyzE this = is + a + long + and + nasty + fortran + statement(that)/zH this = is + a + long + and + nasty + fortran + statement(that)z& @ /must + be + wrapped + properlyH)r< _wrap_fortranlenrMzip)r\lines wrapped_linesZexpected_lineslinewrrVrVrWtest_wrap_fortransl  rc CsFt}ddddddg}dddd dd dd ddd g }|||ksBtdS) NzH 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.0d0zD this_variable_is_very_long_because_we_try_to_test_line_break =z @ 1.0d0zE this_variable_is_very_long_because_we_try_to_test_line_break =zF this_variable_is_very_long_because_we_try_to_test_line_break =zG this_variable_is_very_long_because_we_try_to_test_line_break =z @ 10.0d0r<rrMr\rrrVrVrWtest_wrap_fortran_keep_d0s*  rcCsttdddS)NcSsttdddS)Nrgarbage)method)r;rrVrVrVrWrj6rkztest_settings..)rB TypeErrorrVrVrVrW test_settings5srcCs0td\}}tt|t|dddks,tdS)NrnrCrDzsin(y) + cos(x))rr;rrrMrrVrVrWtest_free_form_code_line9s rcCs@td\}}tt|t|ddd}d}||kss   rcCs0tddi}dg}ddg}|||ks,tdS)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 outputrrrVrVrWtest_free_form_comment_lineIs  rc Cstddd\}}td}td}td}td|}td|}d }t|||f||||d d }||d d iks||d diks||d diks||d dikstdS)NrTrcrrSrTrjz_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;rM) rorrrSrTrrrrrVrVrW test_loopsRs    "rcCsftddtd\}}td}td}t||}d|jj|jd}t||||dd }||ksbtdS) Nzi mT)r/clsrSrTzMdo i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do)ZicountZmcountrCr)rrr=r>labelZ dummy_indexr;rM)rrrSrTrrrVrVrWtest_dummy_loopsls rcCsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}t|j|jdd }|d stdS) NrrTshaperSDyrqrF)rZcontract+Dy(i) = (y(i + 1) - y(i))/(x(i + 1) - x(i)))r=r>rr;rlhsendswithrMZlen_yrTrSrrrZcode0rVrVrW2test_fcode_Indexed_without_looking_for_contraction{s2rcsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}tt|j|j}|dstGd d d t t | fd d | t D}t||||d||||d||}tt|j|j}|dstdS) NrrTrrSrrqrrc@s eZdZdS)z.test_element_like_objects..ElementExprN)r^r_r`rVrVrVrW ElementExprsrc3s |]}||j|jfVqdSrh)nameindices)rrrrVrW sz,test_element_like_objects..)r?r>rr;r#rrrrMr1rsubsZatomsr@rrVrrWtest_element_like_objectss22rcCs:Gdddt}|}td}|t|ddks6tdS)Nc@seZdZejZdS)z1test_derived_classes..MyFancyFCodePrinterN)r^r_r`r<Z_default_settingscopyrVrVrVrWMyFancyFCodePrintersrrSZborkz bork = sin(x))r<rrrrM)rr\rSrVrVrWtest_derived_classessrcCs.d}d}tddi}||}||ks*tdS)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_coderM)Z codelinesrrOrrVrVrW test_indents  rc Cs.td\}}}t||td||dkf|dft|g}tddd}t||dksXttd|d |dkf|d dft|d |d }t|d d dksttddd}tddd}tt|d dt|d g|d |d |ddgd|d|d t|d ddgg}t||dks*tdS)Nzx,y,zrIrTrrprqz 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)rqrrrrerz? merge(2*A(3, 1), A(3, 1), x > 0) + sin(A(2, 1)) + A(1, 1)rPrM)rpr)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;rMrr) rSrTrUZmatrrRrPrrrVrVrWtest_Matrix_printings&,  6   $rcCsDtd\}}t|tdddt|||g}t|}|dks@tdS)NrrrrrIz2 do x = 0, 9, 2 y = x*y end do)rr"r!r#r;rM)rSrTfZsolrVrVrWtest_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||kstdS)NrerCrf)r;rM)rRrefkwargsrVrVrWchecksz%test_fcode_Declaration..checkrTrczinteger*4 :: ir*)valueattrszreal*4, parameter :: x = 42typez logical :: yzreal*4zreal*8)Z type_aliases) rr&Zdeducedr%r'r)r+r(r*) rrZvar1Zdcl1rSrTZvar2Zdcl2bZvar3Zdcl3rVrVrWtest_fcode_Declarations          rcCsxtddd}tddd}tddd}t|ddks8ttd|ddksPt|d|||}t|d ksttdS) NrrqrpBCrz A(1, 1)z 3*A(1, 1)z (A - B)(1, 1))r:r;rMr)rrrFrVrVrWtest_MatrixElement_printing's   rcCs(td}tt|dddddks$tdS)NrS+rqrCrDz x = x + 1)rr;r$rMrirVrVrWtest_aug_assign4srcCs8td}ttt|dkt|ddgdddks4tdS)NrSrq-rCrDz)do while (abs(x) > 1) x = x - 1 end do)rr;r,rLr$rMrirVrVrW test_While9s"rcsRtd}tddd}t|td}t|td}ttd||gttfdddS) NrSroTrcrpowercstSrhrrVfp1rVrWrjJrkz.test_FunctionPrototype_print..)rr&r*r/r-rBrl)rSrovxvnrVrrWtest_FunctionPrototype_printBs    r csjtd}tddd}t|td}t|td}t|||t|g}ttd||g|ttfdddS) NrSroTrcrrcstSrhrrVZfd1rVrWrjVrkz/test_FunctionDefinition_print..) rr&r*r/r#r0r.rBrl)rSrorrbodyrVr rWtest_FunctionDefinition_printMs   r N)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.pytestrBrXrbrmrtrxr{r|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rVrVrVrWs            <             1c +<    ?#