sgddlmZddlmZddlmZmZmZddlm 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[dZ\dZ]d Z^d!Z_d"Z`d#Zad$Zbd%Zcd&Zdd'Zed(Zfd)Zgd*Zhd+Zid,Zjd-Zkd.Zld/Zmd0Znd1Zod2Zpd3Zqd4Zrd5Zsd6Ztd7Zud8Zvd9Zwd:Zxd;Zyd<Zzd=Z{d>Z|d?Z}d@Z~dAZdBZdCZdDZdEZdFZdGZdHZyI)J)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)raisesctdd\}}}t||z}tt||z}t |ddk(sJtd\}}}t||z}tt||z} t | d d j d k(sJt | d d ij dk(sJy)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)))rr3absrr<lstrip) pqrq_rexprxyzy_zexpr2s T/var/www/html/venv/lib/python3.12/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExprr]#sgD)GAq! !a% C s1S5z?D V ,0B BB BgGAq! !a% C AcE OE e $Q ' . . 04L LL L j'9 : A A CGk kk kcdtd}Gddt}t||dk(sJy)NrWceZdZdZy)test_printmethod..nintcDd|j|jdzS)Nznint(%s)r)_printargs)selfprinters r\_fcodez%test_printmethod..nint._fcode3styy| << _ ,, ,r^ctdtdd}tdd}ttdd d k(sJtt|dd d k(sJtt|dd d k(sJttfd y)NrWrXTr0rY)complex_rEstandardrGz#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)c,ttSN)r<rrWsr\z!test_fcode_sign..?sd1gr^)rr<rrCNotImplementedError)rXrYrWs @r\test_fcode_signry8s clA c4 A c4 A a2V <@e ee e a2V <@_ __ _ a2V <@q qq q  67r^chtd\}}tdd}t|dzdk(sJt||dzzdk(sJtdt|d z|||zz zz |d z|zz d k(sJtt|d k(sJtt|d k(sJt|dzd k(sJtt|d k(sJttddk(sJt|dzdk(sJt|dzdddk(sJt|t ddzdk(sJy)Nx,ynTroz x**3z x**(y**3)g @rKz, (3.5d0*sin(x))**(-x + y**x)/(x**2 + y)z sqrt(x)z sqrt(dble(n))g? z sqrt(10.0d0)gz 1d0/xgrXrErFzy = x**(-2.0d0)z x**(3.0d0/7.0d0))rr<rrr)rWrXr|s r\test_fcode_PowrBsO 5>DAqT"A A;, && & QT 0 00 0 CF3J!ad(++QTAX6 76 77 7 a>_ ,, , a>2 22 2 C=O ++ + a>_ ,, , b?2 22 2 D>] ** * D#V 48I II I HQN" #'? ?? ?r^c^td}ttdddk(sJttdddk(sJttddd k(sJttd ddk(sJt|tddzd k(sJttdd|zd k(sJy) NrWr}rz 3.0d0/7.0d0 z 2iz -3.0d0/7.0d0z x + 3.0d0/7.0d0z (3.0d0/7.0d0)*x)rr<rrvs r\test_fcode_RationalrTs A !Q $7 77 7 "a !Y .. . !R !%9 99 9 "b! "&9 99 9 Xa^# $(? ?? ? !Q! "&= == =r^chttddk(sJttddk(sJy)NCz 67z -1)r<rrkr^r\test_fcode_Integerr^s0   ++ +   ++ +r^chttddk(sJttddk(sJy)NgE@z 42.0000000000000d0g@xz -1.00000000000000d+20)r<r rkr^r\test_fcode_Floatrcs2 t !; ;; ; u "? ?? ?r^c~td\tttzdk(sJt t fdt t fdt t fdt t fddD]3}tt |d k(sJtz|d k(r3Jy) Nr{z sin(x)**cos(y)c2ttdSNBrsr<rrWrXsr\rwz&test_fcode_functions..kc!Qi"(Er^c$tzdSrr<rsr\rwz&test_fcode_functions..la!eb(Ar^c2ttdSNMrrrsr\rwz&test_fcode_functions..mrr^c$tzdSrrrsr\rwz&test_fcode_functions..nrr^)Zrqiirz modulo(x, y))rr<rrrCrxr)rsrWrXs @@r\test_fcode_functionsrhs 5>DAq Q3q6! "&< << <  EF  AB  EF  AB(GSAY26JJJJQUX.2FFFFGr^c Xt}td\}}}}}}}tt|t ||zzt ||zzdk(sJtt|d|z|z|t ddzzzdk(sJtt|t ||zzt ||zzddk(sJt|t |z dd k(sJ|j|t |z|zd d k(sJ|j|t |zd dk(sJ|j|d dk(sJtdd\}} td} td}td}td| } td|} t| | | f|| z|| ddk(sJy)Nzx,x_,x__,y,X,X_,Yz& exp(x_) + sin(x*y) + cos(X__*Y_)rKrz( 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,mTroArWrXirrErrGzndo 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?) obrWx_x__rXXX_Yr|mrrrs r\ test_caserts B 34AbQqA R3qs8#c!A#h. /@ AA A SAaCE"hq!n"444 5B CC C R3qs8#c!A#h.e D= >> > SV5 15G GG G ::aAhmt: 48T TT T ::aAh$: /3I II I ::bD: )-= == = 5$ 'DAqCACACA C A C A 1a411V D-5 77 7r^c td}tdjd}d}t|tdzd|zk(sJt|tdzd|zk(sJt|tt dzd|zk(sJttt dd|zk(sJtt ddt djdzk(sJt|ttdzd|zk(sJt|ttt dzd|zk(sJy)NrWrz0.8340324452479558d0z x*%sd0z %sd0z x*%s)rrevalfr<rr)rWlog10_17 loglog10_17s r\"test_fcode_functions_with_integersrss|A2w}}R H(K SW (!: :: : SW (!: :: : S2Z NX$= == = QrU  x 7 77 7 R>\CGMM",== == = SR\! "l[&@ @@ @ SQrU_$ % )C CC Cr^cd}t}ttdtj|zk(sJttdt j|zk(sJtt dt j|zk(sJtt dt j|zk(sJttdtj|zk(sJttddtjdzk(sJttd t|jtj|fhtd fk(sJttd t|jt j|fhtd fk(sJtt d t |jt j|fhtd fk(sJtt d t |jt j|fhtdfk(sJttd t|jtj|fhtdfk(sJttdd t|jtjdfhtdfk(sJy)Nrz. parameter (Catalan = %sd0) Catalanz4 parameter (EulerGamma = %sd0) EulerGammaz" parameter (E = %sd0) Ez6 parameter (GoldenRatio = %sd0) GoldenRatioz$ parameter (pi = %sd0) pi) precisionFrIz Catalanz EulerGammaz Ez GoldenRatioz pi)rrJ) r=r<r rr r r rrcset)precrRs r\test_fcode_NumberSymbolrs DA >NQXQ^Q^_cQdd dd d   WZdZjZjkoZp p pp p 8 >> > % QXXaggdm $ %&y%: :: : E *ahh$?!1"0#$'E+>/@ @@ @ 5 ! ahhrxx~& '(#%&= == = q . ahhrxx{# $%suj3: :: :r^cttdk(sJtd}tdtzdk(sJtddtzzdk(sJtddtzz|zdk(sJtt|zdk(sJtddtzz|z d k(sJtdd }td |zd k(sJtt|zdk(sJtd|zdk(sJy)Nz cmplx(0,1)rWz cmplx(0,4)r}z cmplx(3,4)z cmplx(3,4) + xz cmplx(0,1)*xz cmplx(3,4) - xT) imaginaryrz 5*xz x + 3)r<rrrvs r\test_fcode_complexrs 8) )) ) A 1:+ ++ + QqS>/ // / QqS1 !7 77 7 1:- -- - QqS1 !7 77 7t$A 1: $$ $ 1:- -- - Q<= (( (r^ctd\}}tt|dk(sJtt||dk(sJtt |dk(sJy)Nr{z sin(x)z atan2(x, y)z conjg(x))rr<rrrrs r\ test_implicitrsW 5>DAq Q=N ** * q! !4 44 4 1 "2 22 2r^cRtd}td}tt5t t |dddt t t|ddk(sJtt5t ||dddy#1swYWxYw#1swYyxYw)NrWgF)strictzHC Not supported in Fortran: C Integral Integral(sin(x), x))rrrCrxr<r r!r)rWrs r\test_not_fortranrs A A # $ eAh #a&!% 05A AA A # $ ad sB6BBB&cBtd}tt|ddidk(sJtd}tt|ddidk(sJt d }t||d d id k(sJtd d }tt |ddidk(sJy)NrWrzsinrNz zsin(x)r mygammaz mygamma(x)rgreatz great(x)r|Trorfctz fct(n))rr<rr rr)rWrr|s r\test_user_functionsrs A Q 8O KK K A  a'9!5 7:L MM M A 1sGn 59I II IT"A ! k5%9 ;>L MM Mr^c td}tdt|d|z}t||dk(sJtdt|dtz|z }t||dt j dzk(sJt d}tdtd d }tdt||d |zzd|zz}t||||| dk(sJy)NrWrrKz 2*xz( parameter (pi = %sd0) 2*pi/xrrrr|Tror~rzI do i = 1, n A(i) = (A(i) + 1)*(A(i) + 2)*A(i) end do)rrBrr<rrr>r?)rWrrrs r\test_inline_functionrs AS&AaC.1A 1;+ %% %S&AbDF"34A 1;      CA Cd+,AS&Aq1uIq1u,=">?A 1Q4AaD )   r^cPtd}tt|ddk(sJy)NrWsrz s = sin(x))rr<rrvs r\test_assign_tors% A Q3 '+= == =r^ctd\}}t||zdzjddk(sJtdDcgc]}||z }}tt |dk(sJycc}w)Nr{rvarrz 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**10 zR x**10 + x**9 + x**8 + x**7 + x**6 + x**5 + x**4 + x**3 + x**2 + x @ + 1)rr<expandranger)rWrXres r\test_line_wrappingrs 5>DAq 1q52+%%'5 9 )   Ry!!A!A! a>    "s A'cLtd\}}tt||k||dzkddk(sJtt||k||dzkddk(sJtt ||k||dzkddd k(sJtt ||k||dzkdd k(sJy) Nx yr~rErFzx < y .and. y < x + 1zx < y .or. y < x + 1Fevaluatezx < y .neqv. y < x + 1zx < y .eqv. y < x + 1)rr<r5r6r9r8rs r\test_fcode_precedencer s 5>DAq QUAAI&f =   AE1q1u9%V <   QUAAI6 !9 :: : AE1q1u9-V D   r^c td\}}}tt|ddk(sJtt||ddk(sJtt|t|ddk(sJttt||ddk(sJttt|t|ddk(sJttt||d dd k(sJtt ||dd k(sJtt |t|dd k(sJtt t||ddk(sJtt t|t|ddk(sJttt ||d ddk(sJttt |||ddk(sJttt |||ddk(sJttt |||ddk(sJtt t|||ddk(sJtt t|||ddk(sJtt t|||ddk(sJtt|||ddk(sJtt||t|ddk(sJtt|t||ddk(sJttt|||ddk(sJttt|||d ddk(sJtt |||ddk(sJtt ||t|ddk(sJtt |t||ddk(sJtt t|||ddk(sJttt |||d dd k(sJy)!NrHrErFz.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<r7r5r6rWrXrYs r\test_fcode_LogicalrsgGAq! Qv .) ;; ; Q& 1[ @@ @ QAv 6:K KK K SVQv 6:K KK K SVSV$F ;   SAY/v F   Aq 0J >> > As1vf 59I II I CFAf 59I II I CFCF#6 :   R1X.f E   R1Xq! 8=zx >= yz<=zx <= y>zx > yDAq Aq$'v >( JJ J Aq$'v >( JJ J Aq$'v >( JJ J Aq$'v >( JJ J Aq#&f = HH H Aq#&f = HH Hr^c td}t||dkf|dzdfttfdt d}d}||k(sJt t||dkf|dzdfd d k(sJt ||z }t ||z }td D]}t||}t||}d }t t||dkf|dfd }||k(sJt t||dkf|dz|dkDft |dfd}d}||k(sJt||dkf|dz|dkDft ||dkDfttfdy)NrWr~rKTctSrurrVsr\rwz&test_fcode_Piecewise..s d r^rqrz merge(x, x**2, x < 1)rrzR if (x < 1) then var = x else var = x**2 end ifra/ 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)ctSrurrsr\rwz&test_fcode_Piecewise..s uT{r^) rrrCrxr<rrrr ValueError)rWcodeexpectedabrrVs @r\test_fcode_Piecewisers A aQZ!Q$ .D  34  #D,H 8   Aq1u:1d|4 F    AqA AqA 2Y AJ AJ   Aq1u:4y1\ JD 8   Aq1u:1a!e}s1vtnEPR SDBH 8   aQZ!Q$AQQ @D :*+r^ct}gd}|j|}gd}|D]}t|dkrJt||D] \}}||k(r Jt|t|k(sJy)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)rflines wrapped_linesexpected_lineslinewrs r\test_wrap_fortranrsnG E"))%0MN@4yBM>21Av v } ^!4 44 4r^cTt}gd}gd}|j||k(sJy)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=rrfrrs r\test_wrap_fortran_keep_d0r s2nG E H   '8 33 3r^c&ttdy)Nc.ttddS)Nrgarbage)method)r<rrkr^r\rwztest_settings..6seAaD;r^)rC TypeErrorrkr^r\ test_settingsr5s  9;DAq Q#a& 7;L LL Lr^ctd\}}tt|t|zdzj d}d}||k(sJy)Nr{rrErFzsin(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)rWrXresultrs r\ test_free_form_continuation_liner>sO 5>DAq SVc!f_*224F KF - X  r^cXtddi}dg}ddg}|j||k(sJy)NrGrEz^! 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 outputrrs r\test_free_form_comment_linerIsAOV45Gn oEQ"$H   '8 33 3r^c,tdd\}}td}td}td}td|}td|}d }t|||f||z||d }||d d izk(s#||d dizk(s||d dizk(s ||d dizk(sJyyyy)NrTrorrWrXrjz_do i = 1, m y(i) = 0 end do do i = 1, m do j = 1, n y(i) = %(rhs)s end do end dorErrhszy(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<) r|rrrWrXrrrrs r\ test_loopsrRs 5$ 'DAqCACACA C A C A   1a411V DD H'<== = H'<== = H'<== = H'<== =? > > > > >r^ctddt\}}td}td}t||}d|jj |j dz}t ||||d }||k(sJy) Nzi mT)r0clsrWrXzMdo i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do)icountmcountrEr)rrr>r?label dummy_indexr<)rrrWrXrrs r\test_dummy_loopsr!ls~ 5$E 2DAqCACA Aq A 77&&!--@ AH 11V d}td|f}td|f}td|dz f}td|dz }t||||dz||z ||dz||z z }t|j|j d }|j d sJy) NrrXshaperWDyr~rF)rcontract+Dy(i) = (y(i + 1) - y(i))/(x(i + 1) - x(i)))r>r?rr<rlhsendswith)len_yrXrWr%rrcode0s r\2test_fcode_Indexed_without_looking_for_contractionr,{s ECx(ACx(A T%' ,B CqAA1Q3!q1vad{+,A !%%1555 9E >>G HH Hr^cd}td|f}td|f}td|dz f}td|dz }t||||dz||z ||dz||z z }tt |j |j }|jdsJGd d tt|jfd |jtD}t||||dz||z ||dz||z z }tt |j |j }|jdsJy) NrrXr#rWr%r~rr'c eZdZy).test_element_like_objects..ElementExprN)rhrirjrkr^r\ ElementExprr/s r^r0c3\K|]#}||j|jf%ywru)nameindices).0rr0s r\ z,test_element_like_objects..s%Tq;qvvqyy12Ts),) r@r?rr<r$r(rr)r2rsubsatomsrA)r*rXrWr%rrr+r0s @r\test_element_like_objectsr8s5 ECx(ACx(A T%' ,B CqAA1Q3!q1vad{+,A *QUUAEE* +E >>G HH H gt  TAGGL>G HH Hr^cGddt}|}td}|jt|ddk(sJy)Nc@eZdZejj Zy)1test_derived_classes..MyFancyFCodePrinterN)rhrirjr=_default_settingscopyrkr^r\MyFancyFCodePrinterr;s(::??Ar^r>rWborkz bork = sin(x))r=rrr)r>rfrWs r\test_derived_classesr@sABlB"#G A ??3q66 *.C CC Cr^cVd}d}tddi}|j|}||k(sJy)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 rGrE)r= indent_code) codelinesrrRrs r\ test_indentrDsC <  : ov./A ]]9 %F X  r^c Btd\}}}t||ztd|z|dkDf|dft|g}t ddd}t ||dk(sJtd|d z|dkDf|d dft|d z|d z}t |d dk(sJt ddd}t ddd}tt|d dt |d g|d |d z|ddgd|dz|d z t|d dzdgg}t ||dk(sJy)Nzx,y,zrKrTrr}r~z 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))rKr)r~rrrrqrz? merge(2*A(3, 1), A(3, 1), x > 0) + sin(A(2, 1)) + A(1, 1)rSrM)r}r)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)rr:rrr;r<rr) rWrXrYmatrrVrSrGrs r\test_Matrix_printingrIspgGAq! !A#y!a%14y93q6B CCS!QA a= ! "" " a$iQ'!D'4 9C$L H1T7 RD  #I KK K S!QAS!QAQsVaQsV- 3!C&!D'1% 1T71S6 4#DAq AuQAAq1u!5 67A (C " ## #r^cd}tdd}tj|}t|}||dtd\}}t|tdt h}t|}||d t|t }t|} || d |td |td |td tti|td ttiy)Nc,t|fddd||k(sJy)NrqrErrr)rVrefkwargss r\checkz%test_fcode_Declaration..checks!TGBfGG3NNNr^rTrozinteger*4 :: ir*)valueattrszreal*4, parameter :: x = 42typez logical :: yzreal*4zreal*8) type_aliases) rr'deducedr&r(r*r,r)r+) rRrvar1dcl1rWrXvar2dcl2bvar3dcl3s r\test_fcode_Declarationr`sO T"A   A D t D $ ! 5>DAq Awb >D  E %./ AE "D t D $ '8 '8 $g7 $g7r^ctddd}tddd}tddd}t|ddk(sJtd|dzdk(sJ|dj|||z }t|d k(sJy) Nrr~r}BCrFz A(1, 1)z 3*A(1, 1)z (A - B)(1, 1))r;r<r6)rrbrcFs r\test_MatrixElement_printingre'sS!QAS!QAS!QA 4>_ ,- , QtW !2 23 2 $ QAA 8, ,- ,r^cTtd}tt|ddddk(sJy)NrW+r~rErFz x = x + 1)rr<r%rvs r\test_aug_assignrh4s) A AsA&f = LL Lr^c td}ttt|dkDt |ddgddk(sJy)NrWr~-rErFz)do while (abs(x) > 1) x = x - 1 end do)rr<r-rPr%rvs r\ test_Whilerk9sD A s1vzJq#q$9#:;6 R   r^ctd}tdd}t|t}t|t}t td||gt t fdy)NrWr|TrorVpowerctSrur)fp1sr\rwz.test_FunctionPrototype_print..J c r^)rr'r+r0r.rCrx)rWr|vxvnros @r\test_FunctionPrototype_printrsBsP AT"A !$ B !' "B D'B8 4C  23r^ctd}tdd}t|t}t|t}t |||zt |g}t td||g|ttfdy)NrWr|TrorVrmctSrur)fd1sr\rwz/test_FunctionDefinition_print..Vrpr^) rr'r+r0r$r1r/rCrx)rWr|rqrrbodyrvs @r\test_FunctionDefinition_printrxMsk AT"A !$ B !' "B q!Q$  +D T7RHd ;C  23r^N)sympy.core.addrsympy.core.exprrsympy.core.functionrrrsympy.core.modr sympy.corer r r sympy.core.numbersr r rrrrsympy.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_functionsr sympy.integrals.integralsr!sympy.sets.fancysetsr" sympy.codegenr#r$r%sympy.codegen.astr&r'r(r)r*r+r,r-r.r/r0r1r2r3r4sympy.logic.boolalgr5r6r7r8r9sympy.matricesr:r;sympy.printing.fortranr<r= sympy.tensorr>r?sympy.tensor.array.expressionsr@rAsympy.utilities.lambdifyrBsympy.testing.pytestrCr]rmryrrrrrrrrrrrrrrrrrrrrrr rrrrrr!r,r8r@rDrIrMr`rerhrkrsrxrkr^r\rsk 8899CC$".>B=9:FF9.&55!!!!,,==/6)D9' l-8@$>, @ G7> D:0 )3 M&>   ."b`$FI(,V95x44=M 4?4 II&D=~ F#82 .M 4 4r^