OtjddlmZddlmZddlmZmZmZddlm 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~dAZdBZdCZdDZdEZdFZdGZdHZdIS)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 |ddksJtd\}}}t||z}tt||z} t | d d d ksJt | d d idksJdS)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 _/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExprr]#sgD)))GAq! !a% C s1S5zz??D V , , ,0B B B B BgGAq! !a% C AcE OOE e $ $ $Q ' . . 0 04L L L L L j'9 : : : A A C CGk k k k k k kctd}Gddt}t||dksJdS)NrWceZdZdZdS)test_printmethod..nintcHd||jdzS)Nznint(%s)r)_printargs)selfprinters r\_fcodez%test_printmethod..nint._fcode3sty| < << >_ , , , , , ,r^cxtdtdd}tdd}ttdd d ksJtt|dd d ksJtt|dd d ksJttfd dS)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..?sd1ggr^)rr<rrCNotImplementedError)rXrYrWs @r\test_fcode_signry8s cllA c4   A c4   A a2V < < <@e e e e e a2V < < <@_ _ _ _ _ a2V < < <@q q q q q  6 6 6 677777r^ctd\}}tdd}t|dzdksJt||dzzdksJtdt|d z|||zz zz |d z|zz d ksJtt|d ksJtt|d ksJt|dzd ksJtt|d ksJttddksJt|dzdksJt|dzdddksJt|t ddzdksJdS)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_PowrBs 5>>DAqT"""A A;;, & & & & QT  0 0 0 0 0 CFF3J!ad(++QTAX6 7 76 7 7 7 7 a>>_ , , , , a>>2 2 2 2 2 C==O + + + + a>>_ , , , , b??2 2 2 2 2 D>>] * * * * D#V 4 4 48I I I I I HQNN" # #'? ? ? ? ? ? ?r^ctd}ttdddksJttdddksJttddd ksJttd ddksJt|tddzd ksJttdd|zd ksJdS) 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 7 7 7 7 "a ! !Y . . . . !R ! !%9 9 9 9 9 "b!! " "&9 9 9 9 9 Xa^^# $ $(? ? ? ? ? !Q! " "&= = = = = = =r^cttddksJttddksJdS)NCz 67z -1)r<rrkr^r\test_fcode_Integerr^sF    + + + +    + + + + + +r^cttddksJttddksJdS)NgE@z 42.0000000000000d0g@xz -1.00000000000000d+20)r<r rkr^r\test_fcode_FloatrcsH t  !; ; ; ; ; u  "? ? ? ? ? ? ?r^ctd\tttzdksJt t fdt t fdt t fdt t fddD]A}tt |d ksJtz|d ksJBdS) Nr{z sin(x)**cos(y)cBttdSNBrsr<rrWrXsr\rwz&test_fcode_functions..kc!Qii"(E(E(Er^c,tzdSrr<rsr\rwz&test_fcode_functions..la!eb(A(A(Ar^cBttdSNMrrrsr\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 Q3q66! " "&< < < < <  E E E E EFFF  A A A A ABBB  E E E E EFFF  A A A A ABBB(GGSAYY2226JJJJJQUX...2FFFFFFGGr^c Tt}td\}}}}}}}tt|t ||zzt ||zzdksJtt|d|z|z|t ddzzzdksJtt|t ||zzt ||zzddksJt|t |z dd ksJ||t |z|zd d ksJ||t |zd dksJ||d dksJtdd\}} td} td}td}td| } td|} t| | | f|| z|| ddksJdS)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_casertsO B 344AbQqA R3qs88#c!A#hh. / /@ A A A A SAaCE"hq!nn"444 5 5B C C C C R3qs88#c!A#hh.e D D D= > > > > SVV5 1 1 15G G G G G ::aAhmt: 4 48T T T T T ::aAh$: / /3I I I I I ::bD: ) )-= = = = = 5$ ' ' 'DAqCACACA C A C A 1a411V D D D-5 7 7 7 7 7 7r^c Rtd}tdd}d}t|tdzd|zksJt|tdzd|zksJt|tt dzd|zksJttt dd|zksJtt ddt ddzksJt|ttdzd|zksJt|ttt dzd|zksJdS)NrWrz0.8340324452479558d0z x*%sd0z %sd0z x*%s)rrevalfr<rr)rWlog10_17 loglog10_17s r\"test_fcode_functions_with_integersrsos||A2ww}}R  H(K SWW  (!: : : : : SWW  (!: : : : : S2ZZ NX$= = = = = QrUU   x 7 7 7 7 7 R>>\CGGMM",=,== = = = = SR\\! " "l[&@ @ @ @ @ SQrUU__$ % % )C C C C C C Cr^cd}t}ttdtj|zksJttdt j|zksJtt dt j|zksJtt dt j|zksJttdtj|zksJttddtjdzksJttd t|tj|fhtd fksJttd t|t j|fhtd fksJtt d t |t j|fhtd fksJtt d t |t j|fhtdfksJttd t|tj|fhtdfksJttdd t|tjdfhtdfksJdS)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_NumberSymbolrs0 DA >>NQXQ^_cQdQdd d d d d    WZdZjkoZpZp p p p p p 88 > > > > % QXXagdmm $ $ %&y%: : : : : E * * *ahh$?!?!1"0#$'EE+>/@ @ @ @ @ 5 ! ! ! ahhrx~~&& '(#%%&= = = = = q . . . ahhrx{{## $%suuj3: : : : : : :r^cTttdksJtd}tdtzdksJtddtzzdksJtddtzz|zdksJtt|zdksJtddtzz|z d ksJtdd }td |zd ksJtt|zdksJtd|zdksJdS)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_complexrs3 88) ) ) ) ) A 1::+ + + + + QqS>>/ / / / / QqS1  !7 7 7 7 7 1::- - - - - QqS1  !7 7 7 7 7t$$$A 1:: $ $ $ $ 1::- - - - - Q<<= ( ( ( ( ( (r^ctd\}}tt|dksJtt||dksJtt |dksJdS)Nr{z sin(x)z atan2(x, y)z conjg(x))rr<rrrrs r\ test_implicitrsv 5>>DAq Q==N * * * * q!  !4 4 4 4 4 1  "2 2 2 2 2 2 2r^ctd}td}tt5t t |dddn #1swxYwYt t t|ddksJtt5t ||ddddS#1swxYwYdS)NrWgF)strictzHC Not supported in Fortran: C Integral Integral(sin(x), x))rrrCrxr<r r!r)rWrs r\test_not_fortranrsO A A # $ $ eAhh #a&&!!% 0 0 05A A A A A # $ $ aadd s#AA #A ,CCCctd}tt|ddidksJtd}tt|ddidksJt d }t||d d id ksJtd d }tt |ddidksJdS)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 8 8 8O K K K K A  a'9!5 7 7 7:L M M M M A 1sGn 5 5 59I I I I IT"""A ! k5%9 ; ; ;>L M M M M M Mr^c btd}tdt|d|z}t||dksJtdt|dtz|z }t||dt jdzksJt d}tdtd d }tdt||d |zzd|zz}t||||| dksJdS)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..11A 1;;+ % % % %S&AbDF"3"344A 1;;        CA Cd+++,,AS&Aq1uIq1u,=">">??A 1Q4AaD ) ) )       r^cltd}tt|ddksJdS)NrWsrz s = sin(x))rr<rrvs r\test_assign_tors8 A Q3 ' ' '+= = = = = = =r^ctd\}t|zdzddksJfdtdD}tt |dksJdS) 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**10cg|]}|zSrkrk).0rrWs r\ z&test_line_wrapping..s!!!!A!!!r^ zR x**10 + x**9 + x**8 + x**7 + x**6 + x**5 + x**4 + x**3 + x**2 + x @ + 1)rr<expandranger)rXerWs @r\test_line_wrappingrs 5>>DAq 1q52+%%''5 9 9 9 )     "!!!uRyy!!!A a>>       r^ctd\}}tt||k||dzkddksJtt||k||dzkddksJtt ||k||dzkddd ksJtt ||k||dzkdd ksJdS) 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 < < <     QUAAI666   !9 : : : : AE1q1u9--V D D D      r^c  td\}}}tt|ddksJtt||ddksJtt|t|ddksJttt||ddksJttt|t|ddksJttt||d dd ksJtt ||dd ksJtt |t|dd ksJtt t||ddksJtt t|t|ddksJttt ||d ddksJttt |||ddksJttt |||ddksJttt |||ddksJtt t|||ddksJtt t|||ddksJtt t|||ddksJtt|||ddksJtt||t|ddksJtt|t||ddksJttt|||ddksJttt|||d ddksJtt |||ddksJtt ||t|ddksJtt |t||ddksJtt t|||ddksJttt |||d dd ksJdS)!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 1 1[ @ @ @ @ QAv 6 6 6:K K K K K SVVQv 6 6 6:K K K K K SVVSVV$$F ; ; ;    SAYY///v F F F     Aq 0 0 0J > > > > As1vvf 5 5 59I I I I I CFFAf 5 5 59I I I I I CFFCFF##6 : : :     R1XX...f E E E     R1XXq!! 8 8 8=zx >= yz<=zx <= y>zx > y>DAq Aq$''v > > >( J J J J Aq$''v > > >( J J J J Aq$''v > > >( J J J J Aq$''v > > >( J J J J Aq#&&f = = = H H H H Aq#&&f = = = H H H H H Hr^c Vtd}t||dkf|dzdfttfdt d}d}||ksJt t||dkf|dzdfd d ksJt ||z }t ||z }td D]"}t||}t||}#d }t t||dkf|dfd }||ksJt t||dkf|dz|dkft |dfd}d}||ksJt||dkf|dz|dkft ||dkfttfddS)NrWr~rKTc"tSrurrVsr\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)c"tSrurrsr\rwz&test_fcode_Piecewise..s uT{{r^) rrrCrxr<rrrr ValueError)rWcodeexpectedabrrVs @r\test_fcode_Piecewisers A aQZ!Q$ . .D  3 3 3 3444  # # #D,H 8     Aq1u:1d|44 F F F      AqA AqA 2YY AJJ AJJ   Aq1u:4y11\ J J JD 8     Aq1u:1a!e}s1vvtnEEPR S S SDBH 8     aQZ!Q$AQQ @ @D :****+++++r^ct}gd}||}gd}|D]}t|dksJt||D] \}}||ksJt|t|ksJdS)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_fortranrsnnG   E"))%00MN@4yyBM>221Avvvvv }  ^!4!4 4 4 4 4 4 4r^cht}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 =r zF this_variable_is_very_long_because_we_try_to_test_line_break =r zG this_variable_is_very_long_because_we_try_to_test_line_break =r rz @ 10.0d0r=rrfrrs r\test_wrap_fortran_keep_d0r sTnnG   E   H   ' '8 3 3 3 3 3 3r^c2ttddS)Nc>ttddS)Nrgarbage)method)r<rrkr^r\rwztest_settings..6seAaDD;;;r^)rC TypeErrorrkr^r\ test_settingsr5s 9;;<<<<>DAq Q#a&& 7 7 7;L L L L L L Lr^ctd\}}tt|t|zdzd}d}||ksJdS)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>sg 5>>DAq SVVc!ff_*2244F K K KF - X      r^cltddi}dg}ddg}|||ksJdS)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 outputr r s r\test_free_form_comment_linerIsPOV455Gn oEQ"$H   ' '8 3 3 3 3 3 3r^ctdd\}}td}td}td}td|}td|}d }t|||f||z||d }||d d izks#||d dizks||d dizks||d dizksJdSdSdSdS)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 D D DD H'<== = = H'<== = = H'<== = = H'<== = = = > = = = = = = = =r^ctddt\}}td}td}t||}d|jj|jdz}t ||||d }||ksJdS) 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 2 2 2DAqCACA Aq A 7&!-@@ AH 11V < < r?rr<rlhsendswith)len_yrXrWr(rrcode0s r\2test_fcode_Indexed_without_looking_for_contractionr/{s ECx(((ACx(((A T%' , , ,B CqAA1Q3!q1vad{+,,A !%155 9 9 9E >>G H HHH HHHr^c4d}td|f}td|f}td|dz f}td|dz }t||||dz||z ||dz||z z }tt |j|j}|dsJGd d tt| fd | tD}t||||dz||z ||dz||z z }tt |j|j}|dsJdS) NrrXr&rWr(r~rr*ceZdZdS).test_element_like_objects..ElementExprN)rhrirjrkr^r\ ElementExprr2s r^r3c3HK|]}||j|jfVdSru)nameindices)rrr3s r\ z,test_element_like_objects..s8TTq;;qvqy112TTTTTTr^) r@r?rr<r$r+rr,r2rsubsatomsrA)r-rXrWr(rrr.r3s @r\test_element_like_objectsr:s ECx(((ACx(((A T%' , , ,B CqAA1Q3!q1vad{+,,A *QUAE** + +E >>G H HHH H     gt    TTTTAGGL>G H HHH HHHr^cGddt}|}td}|t|ddksJdS)Nc@eZdZejZdS)1test_derived_classes..MyFancyFCodePrinterN)rhrirjr=_default_settingscopyrkr^r\MyFancyFCodePrinterr=s%(:??AAr^r@rWborkz bork = sin(x))r=rrr)r@rfrWs r\test_derived_classesrBs{BBBBBlBBB"!##G A ??3q666 * *.C C C C C C Cr^cjd}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 rGrE)r= indent_code) codelinesrrRrs r\ test_indentrFsR <  : ov.//A ]]9 % %F X      r^c td\}}}t||ztd|z|dkf|dft|g}t ddd}t ||dksJtd|d z|dkf|d dft|d z|d z}t |d dksJt ddd}t ddd}tt|d dt |d g|d |d z|ddgd|dz|d z t|d dzdgg}t ||dksJdS)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) rWrXrYmatrrVrSrIrs r\test_Matrix_printingrKsgGAq! !A#y!a%14y993q66B C CCS!QA a== ! " " " " a$iQ'!D'4 9 9C$LL H1T7 RD  # # #I K K K K S!QAS!QAQsVaQsV- 3!C&!D'1% 1T71S6 4#<>DAq AuQAAq1u!5!5 677A ((C " # # # # # #r^cDd}tdd}tj|}t|}||dtd\}}t|tdt h}t|}||d t|t }t|} || d |td |td |td tti|td ttidS)Nc2t|fddd||ksJdS)NrqrErrr)rVrefkwargss r\checkz%test_fcode_Declaration..checks0TGBfGGGG3NNNNNNr^rTrozinteger*4 :: ir*)valueattrszreal*4, parameter :: x = 42typez logical :: yzreal*4zreal*8) type_aliases) rr'deducedr&r(r*r,r)r+) rTrvar1dcl1rWrXvar2dcl2bvar3dcl3s r\test_fcode_Declarationrbs2OOO T"""A  A  D t  D E$ !!! 5>>DAq Awb > > >D   E E%./// AE " " "D t  D E$ E'8 E'8 E$g7777 E$g777777r^cFtddd}tddd}tddd}t|ddksJtd|dzdksJ|d|||z }t|d ksJdS) Nrr~r}BCrHz A(1, 1)z 3*A(1, 1)z (A - B)(1, 1))r;r<r8)rrdreFs r\test_MatrixElement_printingrg'sS!QAS!QAS!QA 4>>_ , , , , QtW  !2 2 2 2 2 $ QAA 88, , , , , , ,r^cptd}tt|ddddksJdS)NrW+r~rErFz x = x + 1)rr<r%rvs r\test_aug_assignrj4s= A AsA&&f = = = L L L L L Lr^c td}ttt|dkt |ddgddksJdS)NrWr~-rErFz)do while (abs(x) > 1) x = x - 1 end do)rr<r-rPr%rvs r\ test_Whilerm9sc A s1vvzJq#q$9$9#:;;6 R R R       r^ctd}tdd}t|t}t|t}t td||gt t fddS)NrWr|TrorXpowerc"tSrur)fp1sr\rwz.test_FunctionPrototype_print..J c r^)rr'r+r0r.rCrx)rWr|vxvnrqs @r\test_FunctionPrototype_printruBs| AT"""A !$   B !' " " "B D'B8 4 4C  2 2 2 233333r^cFtd}tdd}t|t}t|t}t |||zt |g}t td||g|ttfddS)NrWr|TrorXroc"tSrur)fd1sr\rwz/test_FunctionDefinition_print..Vrrr^) rr'r+r0r$r1r/rCrx)rWr|rsrtbodyrxs @r\test_FunctionDefinition_printrzMs AT"""A !$   B !' " " "B q!Q$   +D T7RHd ; ;C  2 2 2 233333r^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/r:rBrFrKrOrbrgrjrmrurzrkr^r\rs 88888888889999999999CCCCCCCCCCCCCCCC$$$$$$""""""........>>>>>>BBBBBBBB========999999::::::FFFFFFFFFF999999......&&&&&&5555555555!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!,+++++,,,,,,==============////////66666666))))))))DDDDDDDD999999'''''' l l l---888@@@$>>>,,, @@@ G G G777> D D D:::0 ) ) )333 M M M&>>>       ."."."b`$`$`$FIII(,(,(,V959595x4444===MMM 444???4   IIIIII&DDD===~   F###8882 . . .MMM 444 4 4 4 4 4r^