o g@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|dddks!Jtd\}}}t||}tt||} t| dd d d ksDJt| d d iddksRJdS)Nzp q rT)r*free source_formatzexp(p + (q + r))x y zFhumanzexp(re(x) + re(y + z))rerealpartuser_functionsz"exp(realpart(x) + realpart(y + z)))rr2absrr;lstrip) pqrq_rexprxyzy_zexpr2rZX/var/www/visachat/venv/lib/python3.10/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExpr#s   r\cCs0td}Gdddt}t||dksJdS)NrUc@seZdZddZdS)ztest_printmethod..nintcSsd||jdS)Nznint(%s)r)_printargs)selfprinterrZrZr[_fcode3sz%test_printmethod..nint._fcodeN)__name__ __module__ __qualname__rarZrZrZr[nint2s rez nint(x))rrr;)rUrerZrZr[test_printmethod/srfcstdtddd}tddd}ttddd d ksJtt|ddd d ks*Jtt|ddd d ks7Jttfd ddS)NrUrVTr/rW)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;rrZrUrZr[?s z!test_fcode_sign..)rr;rrBNotImplementedError)rVrWrZrmr[test_fcode_sign8s  rpcCs(td\}}tddd}t|ddksJt||ddks"Jtdt|d ||||d |d ksdS) Nrqz sin(x)**cos(y)cttddSNBrkr;rrZrUrVrZr[rnkz&test_fcode_functions..ctddSrr;rZrrZr[rnlcrNMrrrZrrZr[rnmrcrrrrZrrZr[rnnr)Zriiirz modulo(x, y))rr;rrrBrorrrZrr[test_fcode_functionshs rc Cst}td\}}}}}}}tt|t||t||dks$Jtt|d|||tdddks;Jtt|t||t||dddksSJt|t|ddd ksaJ|j|t||d d d ksrJ|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_)rIrvz( 2*X_**(7.0d0/2.0d0)*Y*x + exp(x__)F) 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,mTrgArUrVir 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>) obrUx_x__rVXX_Yrrmrrr rZrZr[ test_casets0"$&"    rcCstd}tdd}d}t|tdd|ksJt|tdd|ks)Jt|ttdd|ks9Jtttdd|ksGJttddtddksXJt|ttdd|kshJt|tttdd|kszJdS)NrUruz0.8340324452479558d0z x*%sd0z %sd0z x*%s)rrevalfr;rr)rUlog10_17 loglog10_17rZrZr["test_fcode_functions_with_integerss " (rcCsd}t}ttdt|ksJttdt|ksJttdt|ks,Jttdt|ks9Jttdt|ksFJttdddtdksUJttd d t|t|fht d fkslJttd d t|t|fht d fksJttd d t|t|fht d fksJttd d t|t|fht dfksJttd 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 rr]set)precrPrZrZr[test_fcode_NumberSymbolsH       rcCsttdksJtd}tdtdksJtddtdks"Jtddt|dks0Jtt|dks:Jtddt|d ksHJtdd d }td |d ksXJtt|dksbJtd|dkslJdS)Nz cmplx(0,1)rUz cmplx(0,4)rsz cmplx(3,4)z cmplx(3,4) + xz cmplx(0,1)*xz cmplx(3,4) - xT) imaginaryrz 5*xz x + 3)r;r rrmrZrZr[test_fcode_complexs rcCsNtd\}}tt|dksJtt||dksJtt|dks%JdS)Nrqz sin(x)z atan2(x, y)z conjg(x))rr;rrrrrZrZr[ test_implicits rcCstd}td}tttt|Wdn1swYttt|dddks0Jttt||WddS1sFwYdS)NrUgF)strictzHC Not supported in Fortran: C Integral Integral(sin(x), x))rrrBror;rr r)rUrrZrZr[test_not_fortrans  "rcCstd}tt|ddiddksJtd}tt|ddiddks$Jtd }t||d d idd ks6Jtd d d}tt|ddiddksJJdS)NrUrzsinrLz zsin(x)rmygammaz mygamma(x)rgreatz great(x)rrTrgrfctz fct(n))rr;rrrr)rUrrrrZrZr[test_user_functionss     rcCstd}tdt|d|}t||dksJtdt|dt|}t||dtdks3Jtd}tdtd d d }tdt||d |d|}t|||||d dks`JdS)NrUrrIz 2*xz( parameter (pi = %sd0) 2*pi/xrrrrrTrgrtrzI do i = 1, n A(i) = (A(i) + 1)*(A(i) + 2)*A(i) end do)rrArr;rrr=r>)rUrrrrZrZr[test_inline_functions    rcCs$td}tt|dddksJdS)NrUsrz s = sin(x))rr;rrmrZrZr[test_assign_tosrcsZtd\}t|ddddksJfddtdD}tt|d ks+JdS) Nrqruvarrz 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|]}|qSrZrZ).0rrmrZr[ 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;expandranger)rVerZrmr[test_line_wrappings   rcCstd\}}tt||k||dkdddksJtt||k||dkdddks,Jtt||k||dkddddd ksAJtt||k||dkddd ksTJdS) Nx yrtrCrDzx < y .and. y < x + 1zx < y .or. y < x + 1Fevaluatezx < y .neqv. y < x + 1zx < y .eqv. y < x + 1)rr;r4r5r8r7rrZrZr[test_fcode_precedence s  rcCsttd\}}}tt|dddksJtt||dddks Jtt|t|dddks/Jttt||dddks>Jttt|t|dddksOJttt||d d ddd ks`Jtt||ddd ksmJtt|t|ddd ks|Jttt||dddksJttt|t|dddksJttt||d d dddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddksJtt|||dddksJtt||t|dddks.Jtt|t||dddks?Jttt|||dddksPJttt|||d d dddkscJtt|||dddksrJtt||t|dddksJtt|t||dddksJttt|||dddksJttt|||d d ddd ksJdS)!NrFrCrDz.not. xz x .and. yzx .and. .not. yzy .and. .not. xz.not. x .and. .not. yFrz.not. (x .and. y)zx .or. yzx .or. .not. yzy .or. .not. xz.not. x .or. .not. yz.not. (x .or. y)zx .and. (y .or. z)zy .and. (x .or. z)zz .and. (x .or. y)zx .or. y .and. zzy .or. x .and. zzz .or. x .and. yzx .and. y .and. zzx .and. y .and. .not. zzx .and. z .and. .not. yzy .and. z .and. .not. xz.not. (x .and. y .and. z)zx .or. y .or. zzx .or. y .or. .not. zzx .or. z .or. .not. yzy .or. z .or. .not. xz.not. (x .or. y .or. z))rr;r6r4r5rUrVrWrZrZr[test_fcode_Logicalsh     "       rcCs(td\}}}tt||dddddksJtt|t|dddddks'Jttt||dddddks8Jttt|t|ddddd ksKJttt||ddddddd ks^Jtt||ddd kskJtt|t|ddd kszJttt||ddd ksJttt|t|dddksJttt||dddddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddksJttt|||dddks Jttt|||dddksJttt|||dddks.Jttt|||dddks?Jttt|||dddksPJttt|||dddksaJttt|||dddksrJttt||dd|dddksJttt||dd|dddksJttt||dd|dddksJttt|||dddddksJttt|||ddddd ksJttt|||ddddd!ksJttt|||ddddd"ksJttt|||ddddd#ks Jttt|||ddddd$ksJttt||dd|ddd%ks0Jttt||dd|ddd&ksCJttt||dd|ddd'ksVJttt|||ddddd(ksiJttt|||ddddd)ks|Jttt|||ddddd*ksJttt||dd|ddd+ksJttt||dd|ddd,ksJttt||dd|ddd-ksJtt|||ddddd.ksJtt||t|ddddd/ksJtt|t||ddddd0ksJttt|||ddddd1ksJdS)2NrFFrrCrDz x .neqv. yzx .neqv. .not. yzy .neqv. .not. xz.not. x .neqv. .not. yz.not. (x .neqv. y)z x .eqv. yzx .eqv. .not. yzy .eqv. .not. xz.not. x .eqv. .not. yz.not. (x .eqv. y)zx .eqv. y .and. zzy .eqv. x .and. zzz .eqv. x .and. yzx .and. (y .eqv. z)zy .and. (x .eqv. z)zz .and. (x .eqv. y)zx .eqv. y .or. zzy .eqv. x .or. zzz .eqv. x .or. yzx .or. (y .eqv. z)zy .or. (x .eqv. z)zz .or. (x .eqv. y)zx .eqv. (y .neqv. z)zy .eqv. (x .neqv. z)zz .eqv. (x .neqv. y)zx .neqv. (y .eqv. z)zy .neqv. (x .eqv. z)zz .neqv. (x .eqv. y)zx .neqv. y .and. zzy .neqv. x .and. zzz .neqv. x .and. yzx .and. (y .neqv. z)zy .and. (x .neqv. z)zz .and. (x .neqv. y)zx .neqv. y .or. zzy .neqv. x .or. zzz .neqv. x .or. yzx .or. (y .neqv. z)zy .or. (x .neqv. z)zz .or. (x .neqv. y)zx .neqv. y .neqv. zzx .neqv. y .neqv. .not. zzx .neqv. z .neqv. .not. yzy .neqv. z .neqv. .not. x)rr;r8r6r7r4r5rrZrZr[test_fcode_XlogicalGs*                            rcCstd\}}tt||ddddksJtt||ddddks"Jtt||dddd ks0Jtt||d ddd ks>Jtt||d ddd ksLJtt||ddddksZJdS)Nrz==rCrDzx == yz!=zx /= yz>=zx >= yz<=zx <= y>zx > y.rirz merge(x, x**2, x < 1)rrzR if (x < 1) then var = x else var = x**2 end ifrua/ 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 ifr weird_namez1 merge(x, merge(x**2, sin(x), x > 1), x < 1)crrlrrZrrZr[rnr) rrrBror;rrrr ValueError)rUcodeexpectedabrrZrr[test_fcode_Piecewises.         0 ,rcCsrt}gd}||}gd}|D] }t|dksJqt||D] \}}||ks,Jq"t|t|ks7JdS)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`lines wrapped_linesexpected_lineslinewrrZrZr[test_wrap_fortrans  rcCs,t}gd}gd}|||ksJdS)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`rrrZrZr[test_wrap_fortran_keep_d0s rcCsttdddS)NcSsttdddS)Nrgarbage)method)r;rrZrZrZr[rn6rztest_settings..)rB TypeErrorrZrZrZr[ test_settings5srcCs0td\}}tt|t|dddksJdS)NrqrCrDzsin(y) + cos(x))rr;rrrrZrZr[test_free_form_code_line9s $rcCs@td\}}tt|t|ddd}d}||ksJdS)NrqrvrCrDzsin(y)**7 + 7*sin(y)**6*cos(x) + 21*sin(y)**5*cos(x)**2 + 35*sin(y)**4* & cos(x)**3 + 35*sin(y)**3*cos(x)**4 + 21*sin(y)**2*cos(x)**5 + 7* & sin(y)*cos(x)**6 + cos(x)**7)rr;rrr)rUrVresultrrZrZr[ test_free_form_continuation_line>s  rcCs0tddi}dg}ddg}|||ksJdS)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 outputrrrZrZr[test_free_form_comment_lineIs rc Cstddd\}}td}td}td}td|}td|}d }t|||f||||d d }||d d iksS||d diksU||d diksW||d diksYJdSdSdSdS)NrTrgrrUrVrjz_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;) rrrrrUrVrrrrrZrZr[ test_loopsRs"  " rcCsftddtd\}}td}td}t||}d|jj|jd}t||||dd }||ks1JdS) Nzi mT)r/clsrUrVzMdo i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do)icountmcountrCr)rrr=r>label dummy_indexr;)rrrUrVrrrZrZr[test_dummy_loopsls rcCsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}t|j|jdd }|d sIJdS) NrrVshaperUDyrtrF)rcontract+Dy(i) = (y(i + 1) - y(i))/(x(i + 1) - x(i)))r=r>rr;rlhsendswithlen_yrVrUrrrcode0rZrZr[2test_fcode_Indexed_without_looking_for_contraction{s2rcsd}td|fd}td|fd}td|dfd}td|d}t||||d||||d||}tt|j|j}|dsIJGd d d tt | fd d | t D}t||||d||||d||}tt|j|j}|dsJdS) NrrVrrUrrtrrc@s eZdZdS)z.test_element_like_objects..ElementExprN)rbrcrdrZrZrZr[ ElementExprsr c3s"|] }||j|jfVqdSrl)nameindices)rrr rZr[ s z,test_element_like_objects..) r?r>rr;r#rrrr1rsubsatomsr@rrZr r[test_element_like_objectss22rcCs:Gdddt}|}td}|t|ddksJdS)Nc@seZdZejZdS)z1test_derived_classes..MyFancyFCodePrinterN)rbrcrdr<_default_settingscopyrZrZrZr[MyFancyFCodePrintersrrUborkz bork = sin(x))r<rrr)rr`rUrZrZr[test_derived_classessrcCs.d}d}tddi}||}||ksJdS)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< indent_code) codelinesrrPrrZrZr[ test_indents  rc Cs,td\}}}t||td||dkf|dft|g}tddd}t||dks,Jtd|d |dkf|d dft|d |d }t|d d dksQJtddd}tddd}tt|d dt|d g|d |d |ddgd|d|d t|d ddgg}t||dksJdS)Nzx,y,zrIrTrrsrtz 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)rtrrrrirz? merge(2*A(3, 1), A(3, 1), x > 0) + sin(A(2, 1)) + A(1, 1)rQrM)rsr)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) rUrVrWmatrrTrQrrrZrZr[test_Matrix_printings&, 6   $ rcCsDtd\}}t|tdddt|||g}t|}|dks JdS)NrrrurIz2 do x = 0, 9, 2 y = x*y end do)rr"r!r#r;)rUrVfsolrZrZr[test_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)NrirCrjr)rTrefkwargsrZrZr[checks"z%test_fcode_Declaration..checkrTrgzinteger*4 :: ir*)valueattrszreal*4, parameter :: x = 42typez logical :: yzreal*4zreal*8) type_aliases) rr&deducedr%r'r)r+r(r*) r"rvar1dcl1rUrVvar2dcl2bvar3dcl3rZrZr[test_fcode_Declarations          r0cCsxtddd}tddd}tddd}t|ddksJtd|ddks(J|d|||}t|d ks:JdS) NrrtrsBCrz A(1, 1)z 3*A(1, 1)z (A - B)(1, 1))r:r;r)rr1r2FrZrZr[test_MatrixElement_printing's   r4cCs(td}tt|dddddksJdS)NrU+rtrCrDz x = x + 1)rr;r$rmrZrZr[test_aug_assign4s r6cCs8td}ttt|dkt|ddgdddksJdS)NrUrt-rCrDz)do while (abs(x) > 1) x = x - 1 end do)rr;r,rNr$rmrZrZr[ test_While9s" r8csRtd}tddd}t|td}t|td}ttd||gttfdddS) NrUrrTrgr&powercrrlrrZfp1rZr[rnJrz.test_FunctionPrototype_print..)rr&r*r/r-rBro)rUrrvxvnrZr:r[test_FunctionPrototype_printBs    r>csjtd}tddd}t|td}t|td}t|||t|g}ttd||g|ttfdddS) NrUrrTrgr&r9crrlrrZfd1rZr[rnVrz/test_FunctionDefinition_print..) rr&r*r/r#r0r.rBro)rUrrr<r=bodyrZr?r[test_FunctionDefinition_printMs   rBN)sympy.core.addrsympy.core.exprrsympy.core.functionrrrsympy.core.modr sympy.corerr r sympy.core.numbersr r r rrrsympy.core.relationalrsympy.core.singletonrsympy.core.symbolrr(sympy.functions.combinatorial.factorialsr$sympy.functions.elementary.complexesrr&sympy.functions.elementary.exponentialrr(sympy.functions.elementary.miscellaneousr$sympy.functions.elementary.piecewiser(sympy.functions.elementary.trigonometricrrr'sympy.functions.special.gamma_functionsrsympy.integrals.integralsr sympy.sets.fancysetsr! sympy.codegenr"r#r$sympy.codegen.astr%r&r'r(r)r*r+r,r-r.r/r0r1r2r3sympy.logic.boolalgr4r5r6r7r8sympy.matricesr9r:sympy.printing.fortranr;r< sympy.tensorr=r>sympy.tensor.array.expressionsr?r@sympy.utilities.lambdifyrAsympy.testing.pytestrBr\rfrprwr{rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr0r4r6r8r>rBrZrZrZr[s           <             1c +<    ?#