
    sg                        d dl mZ d dlmZ d dlmZmZmZ d dlm	Z	 d dl
mZmZmZ d dlmZmZmZmZmZmZ d dlmZ d dlmZ d d	lmZmZ d d
lmZ d dlmZm Z  d dl!m"Z"m#Z# d dl$m%Z% d dl&m'Z' d dl(m)Z)m*Z*m+Z+ d dl,m-Z- d dl.m/Z/ d dl0m1Z1 d dl2m3Z3m4Z4m5Z5 d dl6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZC d dlmDZD d dlmEZE d dlFmGZGmHZHmIZImJZJmKZK d dlLmMZMmNZN d dlOmPZPmQZQ d dlRmSZSmTZT d dlUmVZVmWZW d dlXmYZY d dlZm[Z[ d Z\d Z]d  Z^d! Z_d" Z`d# Zad$ Zbd% Zcd& Zdd' Zed( Zfd) Zgd* Zhd+ Zid, Zjd- Zkd. Zld/ Zmd0 Znd1 Zod2 Zpd3 Zqd4 Zrd5 Zsd6 Ztd7 Zud8 Zvd9 Zwd: Zxd; Zyd< Zzd= Z{d> Z|d? Z}d@ Z~dA ZdB ZdC ZdD ZdE ZdF ZdG ZdH ZyI)J    )Add)Expr)FunctionLambdadiff)Mod)Catalan
EulerGammaGoldenRatio)EFloatIIntegerRationalpi)Eq)S)Dummysymbols)	factorial)	conjugatesign)explog)sqrt)	Piecewise)atan2cossin)gamma)Integral)Range)For
Assignment
aug_assign)DeclarationVariablefloat32float64value_constrealbool_WhileFunctionPrototypeFunctionDefinitionintegerReturnElement)UnevaluatedExpr)
Relational)AndOrNot
EquivalentXor)MatrixMatrixSymbol)fcodeFCodePrinter)IndexedBaseIdx)ArraySymbolArrayElement)implemented_function)raisesc                     t        dd      \  } }}t        ||z         }t        t        | |z               }t	        |d      dk(  sJ t        d      \  }}}t        ||z         }t        t        ||z               }	t	        |	d	      d
   j                         dk(  sJ t	        |	ddi      j                         dk(  sJ y )Nzp q rT)r+   freesource_formatzexp(p + (q + r))x y zFhuman   zexp(re(x) + re(y + z))rerealpartuser_functionsz"exp(realpart(x) + realpart(y + z)))r   r3   absr   r<   lstrip)
pqrq_rexprxyzy_zexpr2s
             T/var/www/html/venv/lib/python3.12/site-packages/sympy/printing/tests/test_fortran.pytest_UnevaluatedExprr]   #   s    gD)GAq!
!a%
 Cs1S5z?DV,0BBBBgGAq!
!a%
 CAcE
OEe$Q'..04LLLLj'9:AACGkkkk    c                  d    t        d      }  G d dt              }t         ||             dk(  sJ y )NrW   c                       e Zd Zd Zy)test_printmethod.<locals>.nintc                 D    d|j                  | j                  d         z  S )Nznint(%s)r   )_printargs)selfprinters     r\   _fcodez%test_printmethod.<locals>.nint._fcode3   s    tyy| <<<r^   N)__name__
__module____qualname__rg    r^   r\   nintra   2   s    	=r^   rl   z      nint(x))r   r   r<   )rW   rl   s     r\   test_printmethodrm   /   s/    A=x = a>_,,,r^   c                     t        d      t        dd      } t        dd      }t        t              dd	      d
k(  sJ t        t        |       dd	      dk(  sJ t        t        |      dd	      dk(  sJ t        t        fd       y )NrW   rX   Tr0   rY   )complex_   rE   standardrG   z#merge(0d0, dsign(1d0, x), x == 0d0)zmerge(0, isign(1, y), y == 0)z/merge(cmplx(0d0, 0d0), z/abs(z), abs(z) == 0d0)c                  ,    t        t                     S N)r<   r   rW   s   r\   <lambda>z!test_fcode_sign.<locals>.<lambda>?   s    d1g r^   )r   r<   r   rC   NotImplementedError)rX   rY   rW   s     @r\   test_fcode_signry   8   s    clAc4 Ac4 Aa2V<@eeeea2V<@____a2V<@qqqq
 67r^   c                  h   t        d      \  } }t        dd      }t        | dz        dk(  sJ t        | |dz  z        dk(  sJ t        dt        |       d	z  | || z  z
  z  z  | d
z  |z   z        dk(  sJ t        t        |             dk(  sJ t        t        |            dk(  sJ t        | dz        dk(  sJ t        t        |             dk(  sJ t        t        d            dk(  sJ t        | dz        dk(  sJ t        | dz  dd      dk(  sJ t        | t	        dd      z        dk(  sJ y )Nx,ynTro      z
      x**3z      x**(y**3)   g      @rK   z,      (3.5d0*sin(x))**(-x + y**x)/(x**2 + y)z      sqrt(x)z      sqrt(dble(n))g      ?
   z      sqrt(10.0d0)g      z      1d0/xg       rX   rE   rF   zy = x**(-2.0d0)   z      x**(3.0d0/7.0d0))r   r<   r   r   r   )rW   rX   r|   s      r\   test_fcode_Powr   B   sO   5>DAqT"AA;,&&&QT0000CF3J!ad(++QTAX6767 7 7a>_,,,a>2222C=O+++a>_,,,b?2222D>]***D#V48IIIIHQN"#'????r^   c                  ^   t        d      } t        t        dd            dk(  sJ t        t        dd            dk(  sJ t        t        dd            d	k(  sJ t        t        d
d            dk(  sJ t        | t        dd      z         dk(  sJ t        t        dd      | z        dk(  sJ y )NrW   r}   r   z      3.0d0/7.0d0   	   z      2iz      -3.0d0/7.0d0z      x + 3.0d0/7.0d0z      (3.0d0/7.0d0)*x)r   r<   r   rv   s    r\   test_fcode_Rationalr   T   s    A!Q $7777"a!Y...!R!%9999"b!"&9999Xa^#$(????!Q!"&====r^   c                  h    t        t        d            dk(  sJ t        t        d            dk(  sJ y )NC   z      67z      -1)r<   r   rk   r^   r\   test_fcode_Integerr   ^   s0    ++++++r^   c                  h    t        t        d            dk(  sJ t        t        d            dk(  sJ y )Ng      E@z      42.0000000000000d0g@xz      -1.00000000000000d+20)r<   r   rk   r^   r\   test_fcode_Floatr   c   s2    t!;;;;u"????r^   c                  ~   t        d      \  t        t              t              z        dk(  sJ t	        t
        fd       t	        t
        fd       t	        t
        fd       t	        t
        fd       dD ]3  } t        t              |       d	k(  sJ t        z  |       d	k(  r3J  y )
Nr{   z      sin(x)**cos(y)c                  2    t        t               d      S NB   rs   r<   r   rW   rX   s   r\   rw   z&test_fcode_functions.<locals>.<lambda>k       c!Qi"(E r^   c                  $    t         z  d      S r   r<   r   s   r\   rw   z&test_fcode_functions.<locals>.<lambda>l       a!eb(A r^   c                  2    t        t               d      S NM   r   r   r   s   r\   rw   z&test_fcode_functions.<locals>.<lambda>m   r   r^   c                  $    t         z  d      S r   r   r   s   r\   rw   z&test_fcode_functions.<locals>.<lambda>n   r   r^   )Z   rq   i  i  r   z      modulo(x, y))r   r<   r   r   rC   rx   r   )rs   rW   rX   s    @@r\   test_fcode_functionsr   h   s    5>DAqQ3q6!"&<<<<
 EF
 AB
 EF
 AB( GSAY26JJJJQUX.2FFFFGr^   c            	      X   t               } t        d      \  }}}}}}}t        t        |      t	        ||z        z   t        ||z        z         dk(  sJ t        t        |      d|z  |z  |t        dd      z  z  z         dk(  sJ t        t        |      t	        ||z        z   t        ||z        z   d      dk(  sJ t        |t        |      z
  d      d	k(  sJ | j                  |t	        |      z  |z   d
      dk(  sJ | j                  |t	        |      z  d      dk(  sJ | j                  |d      dk(  sJ t        dd      \  }}	t        d      }
t        d      }t        d      }t        d|	      }t        d|      }t        |
||f   ||   z  ||   d      dk(  sJ y )Nzx,x_,x__,y,X,X_,Yz&      exp(x_) + sin(x*y) + cos(X__*Y_)rK   r   z(      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,mTro   ArW   rX   ir   rE   r   rG   zndo 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=   r   r<   r   r   r   r   doprintr>   r?   )obrW   x_x__rX   XX_Yr|   mr   r   r   s                r\   	test_caser   t   s   	B 34AbQqAR3qs8#c!A#h./@A A ASAaCE"hq!n"4445BC C CR3qs8#c!A#h.eD=> > >SV515GGGG::aAhmt:48TTTT::aAh$:/3IIII::bD:)-====5$'DAqCACACACACA1a411VD-57 7 7r^   c            
         t        d      } t        d      j                  d      }d}t        | t        d      z        d|z  k(  sJ t        | t        d      z        d|z  k(  sJ t        | t        t	        d            z        d|z  k(  sJ t        t        t	        d                  d|z  k(  sJ t        t        d            dt        d      j                  d      z  k(  sJ t        | t        t        d            z        d|z  k(  sJ t        | t        t        t	        d                  z        d|z  k(  sJ y )NrW   r      z0.8340324452479558d0z      x*%sd0z
      %sd0z
      x*%s)r   r   evalfr<   r   r   )rW   log10_17loglog10_17s      r\   "test_fcode_functions_with_integersr      s   s|A2w}}R H(KSW(!::::SW(!::::S2Z NX$====QrUx 7777R>\CGMM",=====SR\!"l[&@@@@SQrU_$%)CCCCr^   c                     d} t               }t        t              dt        j                  |       z  k(  sJ t        t              dt	        j                  |       z  k(  sJ t        t
              dt        j                  |       z  k(  sJ t        t              dt        j                  |       z  k(  sJ t        t              dt        j                  |       z  k(  sJ t        t        d      dt        j                  d      z  k(  sJ t        t        d	
      t        |j                  t        j                  |             fht               dfk(  sJ t        t        d	
      t        |j                  t	        j                  |             fht               dfk(  sJ t        t
        d	
      t
        |j                  t        j                  |             fht               dfk(  sJ t        t        d	
      t        |j                  t        j                  |             fht               dfk(  sJ t        t        d	
      t        |j                  t        j                  |             fht               dfk(  sJ t        t        dd	      t        |j                  t        j                  d            fht               dfk(  sJ y )Nr   z.      parameter (Catalan = %sd0)
      Catalanz4      parameter (EulerGamma = %sd0)
      EulerGammaz"      parameter (E = %sd0)
      Ez6      parameter (GoldenRatio = %sd0)
      GoldenRatioz$      parameter (pi = %sd0)
      pi   )	precisionFrI   z      Catalanz      EulerGammaz      Ez      GoldenRatioz      pi)r   rJ   )
r=   r<   r	   r   r
   r   r   r   rc   set)precrR   s     r\   test_fcode_NumberSymbolr      s   DA>NQXQ^Q^_cQddddd WZdZjZjkoZp pppp8<qwwt}LLLL!Z]h]n]nos]t!tttt9?"((4.PPPP
aCbhhqkQR R R&	!((7==./0,2365/+K K K K5)
AHH=  0! /"#&5*<.> > > >% 
QXXaggdm$	%&y%: : : :E*ahh$?! 1" 0#$'E+>/@ @ @ @5!
ahhrxx~&	'(#%&= = = =q.
ahhrxx{#	$%suj3: : : :r^   c                     t        t              dk(  sJ t        d      } t        dt        z        dk(  sJ t        ddt        z  z         dk(  sJ t        ddt        z  z   | z         dk(  sJ t        t        | z        dk(  sJ t        ddt        z  z   | z
        d	k(  sJ t        dd
      } t        d| z        dk(  sJ t        t        | z        dk(  sJ t        d| z         dk(  sJ y )Nz      cmplx(0,1)rW      z      cmplx(0,4)r}   z      cmplx(3,4)z      cmplx(3,4) + xz      cmplx(0,1)*xz      cmplx(3,4) - xT)	imaginaryr   z	      5*xz      x + 3)r<   r   r   rv   s    r\   test_fcode_complexr      s    8))))A1:++++QqS>////QqS1!77771:----QqS1!7777t$A1:$$$1:----Q<=(((r^   c                      t        d      \  } }t        t        |             dk(  sJ t        t        | |            dk(  sJ t        t	        |             dk(  sJ y )Nr{   z      sin(x)z      atan2(x, y)z      conjg(x))r   r<   r   r   r   r   s     r\   test_implicitr      sW    5>DAqQ=N***q!!44441"2222r^   c                  R   t        d      } t        d      }t        t              5  t	        t        |              d d d        t	        t        t        |             d      dk(  sJ t        t              5  t	         ||              d d d        y # 1 sw Y   WxY w# 1 sw Y   y xY w)NrW   gF)strictzHC     Not supported in Fortran:
C     Integral
      Integral(sin(x), x))r   r   rC   rx   r<   r    r!   r   )rW   r   s     r\   test_not_fortranr      s    AA	#	$ eAh#a&!%0  5A  A  A  A	#	$ ad   s   B6BBB&c                  B   t        d      } t        t        |       ddi      dk(  sJ t        d      } t        t        |       ddi      dk(  sJ t	        d	      }t         ||       d	d
i      dk(  sJ t        dd      }t        t        |      ddi      dk(  sJ y )NrW   r   zsinrN   z      zsin(x)r    mygammaz      mygamma(x)r   greatz      great(x)r|   Tro   r   fctz      fct(n))r   r<   r   r    r   r   )rW   r   r|   s      r\   test_user_functionsr      s    AQ8OKKKAa'9!57:LM M MA1sGn59IIIIT"A!k5%9;>LM M Mr^   c            	         t        d      } t        dt        | d| z              }t         ||             dk(  sJ t        dt        | dt        z  | z              }t         ||             dt	        j
                  d      z  k(  sJ t        d      }t        dt        d	d
            }t        dt        | | d| z   z  d| z   z              }t         |||         ||         dk(  sJ y )NrW   r   rK   z	      2*xz(      parameter (pi = %sd0)
      2*pi/xr   r   r   r|   Tro   r~   r   zI      do i = 1, n
         A(i) = (A(i) + 1)*(A(i) + 2)*A(i)
      end do)r   rB   r   r<   r   r   r>   r?   )rW   r   r   r   s       r\   test_inline_functionr      s    AS&AaC.1A1;+%%%S&AbDF"34A1;	
    	CACd+,AS&Aq1uIq1u,=">?A1Q4AaD)	  r^   c                  P    t        d      } t        t        |       d      dk(  sJ y )NrW   sr   z      s = sin(x))r   r<   r   rv   s    r\   test_assign_tor      s%    AQ3'+====r^   c                      t        d      \  } }t        | |z   dz  j                         d      dk(  sJ t        d      D cg c]  }| |z  	 }}t        t	        |       dk(  sJ y c c}w )Nr{   r   varr   z      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)r   r<   expandranger   )rW   rX   r   es       r\   test_line_wrappingr      s    5>DAq1q52+%%'59	)  
 Ry!!A!A!a>	   	"s   A'c                  L   t        d      \  } }t        t        | |k  || dz   k        d      dk(  sJ t        t        | |k  || dz   k        d      dk(  sJ t        t	        | |k  || dz   k  d      d      d	k(  sJ t        t        | |k  || dz   k        d      d
k(  sJ y )Nx yr~   rE   rF   zx < y .and. y < x + 1zx < y .or. y < x + 1Fevaluatezx < y .neqv. y < x + 1zx < y .eqv. y < x + 1)r   r<   r5   r6   r9   r8   r   s     r\   test_fcode_precedencer   
  s    5>DAqQUAAI&f=     AE1q1u9%V<  QUAAI6!9: : :AE1q1u9-VD     r^   c            	         t        d      \  } }}t        t        |       d      dk(  sJ t        t        | |      d      dk(  sJ t        t        | t        |            d      dk(  sJ t        t        t        |       |      d      dk(  sJ t        t        t        |       t        |            d      dk(  sJ t        t        t        | |      d	
      d      dk(  sJ t        t	        | |      d      dk(  sJ t        t	        | t        |            d      dk(  sJ t        t	        t        |       |      d      dk(  sJ t        t	        t        |       t        |            d      dk(  sJ t        t        t	        | |      d	
      d      dk(  sJ t        t        t	        ||      |       d      dk(  sJ t        t        t	        ||       |      d      dk(  sJ t        t        t	        | |      |      d      dk(  sJ t        t	        t        ||      |       d      dk(  sJ t        t	        t        ||       |      d      dk(  sJ t        t	        t        | |      |      d      dk(  sJ t        t        | ||      d      dk(  sJ t        t        | |t        |            d      dk(  sJ t        t        | t        |      |      d      dk(  sJ t        t        t        |       ||      d      dk(  sJ t        t        t        | ||      d	
      d      dk(  sJ t        t	        | ||      d      dk(  sJ t        t	        | |t        |            d      dk(  sJ t        t	        | t        |      |      d      dk(  sJ t        t	        t        |       ||      d      dk(  sJ t        t        t	        | ||      d	
      d      d k(  sJ y )!NrH   rE   rF   z.not. xz	x .and. yzx .and. .not. yzy .and. .not. xz.not. x .and. .not. yFr   z.not. (x .and. y)zx .or. yzx .or. .not. yzy .or. .not. xz.not. x .or. .not. yz.not. (x .or. y)zx .and. (y .or. z)zy .and. (x .or. z)zz .and. (x .or. y)zx .or. y .and. zzy .or. x .and. zzz .or. x .and. yzx .and. y .and. zzx .and. y .and. .not. zzx .and. z .and. .not. yzy .and. z .and. .not. xz.not. (x .and. y .and. z)zx .or. y .or. zzx .or. y .or. .not. zzx .or. z .or. .not. yzy .or. z .or. .not. xz.not. (x .or. y .or. z))r   r<   r7   r5   r6   rW   rX   rY   s      r\   test_fcode_Logicalr     s   gGAq!Qv.);;;Q&1[@@@QAv6:KKKKSVQv6:KKKKSVSV$F;     SAY/vF   Aq0J>>>As1vf59IIIICFAf59IIIICFCF#6:  R1X.fE   R1Xq!8<PPPPR1Xq!8<PPPPR1Xq!8<PPPPC1Iq!8<NNNNC1Iq!8<NNNNC1Iq!8<NNNNQ1V48KKKKQ3q6"&9!" " "QA"&9!" " "SVQ"&9!" " "SAq\E2&I#$ $ $ Aq!F37HHHHAq#a&!8     As1vq!8     CFAq!8     R1a[51H!" " "r^   c            	         t        d      \  } }}t        t        | |d      d      dk(  sJ t        t        | t        |      d      d      dk(  sJ t        t        t        |       |d      d      dk(  sJ t        t        t        |       t        |      d      d      d	k(  sJ t        t        t        | |d      d      d      d
k(  sJ t        t	        | |      d      dk(  sJ t        t	        | t        |            d      dk(  sJ t        t	        t        |       |      d      dk(  sJ t        t	        t        |       t        |            d      dk(  sJ t        t        t	        | |      d      d      dk(  sJ t        t	        t        ||      |       d      dk(  sJ t        t	        t        ||       |      d      dk(  sJ t        t	        t        | |      |      d      dk(  sJ t        t        t	        ||      |       d      dk(  sJ t        t        t	        ||       |      d      dk(  sJ t        t        t	        | |      |      d      dk(  sJ t        t	        t        ||      |       d      dk(  sJ t        t	        t        ||       |      d      dk(  sJ t        t	        t        | |      |      d      dk(  sJ t        t        t	        ||      |       d      dk(  sJ t        t        t	        ||       |      d      dk(  sJ t        t        t	        | |      |      d      dk(  sJ t        t	        t        ||d      |       d      dk(  sJ t        t	        t        || d      |      d      dk(  sJ t        t	        t        | |d      |      d      dk(  sJ t        t        t	        ||      | d      d      dk(  sJ t        t        t	        ||       |d      d      d k(  sJ t        t        t	        | |      |d      d      d!k(  sJ t        t        t        ||      | d      d      d"k(  sJ t        t        t        ||       |d      d      d#k(  sJ t        t        t        | |      |d      d      d$k(  sJ t        t        t        ||d      |       d      d%k(  sJ t        t        t        || d      |      d      d&k(  sJ t        t        t        | |d      |      d      d'k(  sJ t        t        t        ||      | d      d      d(k(  sJ t        t        t        ||       |d      d      d)k(  sJ t        t        t        | |      |d      d      d*k(  sJ t        t        t        ||d      |       d      d+k(  sJ t        t        t        || d      |      d      d,k(  sJ t        t        t        | |d      |      d      d-k(  sJ t        t        | ||d      d      d.k(  sJ t        t        | |t        |      d      d      d/k(  sJ t        t        | t        |      |d      d      d0k(  sJ t        t        t        |       ||d      d      d1k(  sJ y )2NrH   Fr   rE   rF   z
x .neqv. yzx .neqv. .not. yzy .neqv. .not. xz.not. x .neqv. .not. yz.not. (x .neqv. y)z	x .eqv. yzx .eqv. .not. yzy .eqv. .not. xz.not. x .eqv. .not. yz.not. (x .eqv. y)zx .eqv. y .and. zzy .eqv. x .and. zzz .eqv. x .and. yzx .and. (y .eqv. z)zy .and. (x .eqv. z)zz .and. (x .eqv. y)zx .eqv. y .or. zzy .eqv. x .or. zzz .eqv. x .or. yzx .or. (y .eqv. z)zy .or. (x .eqv. z)zz .or. (x .eqv. y)zx .eqv. (y .neqv. z)zy .eqv. (x .neqv. z)zz .eqv. (x .neqv. y)zx .neqv. (y .eqv. z)zy .neqv. (x .eqv. z)zz .neqv. (x .eqv. y)zx .neqv. y .and. zzy .neqv. x .and. zzz .neqv. x .and. yzx .and. (y .neqv. z)zy .and. (x .neqv. z)zz .and. (x .neqv. y)zx .neqv. y .or. zzy .neqv. x .or. zzz .neqv. x .or. yzx .or. (y .neqv. z)zy .or. (x .neqv. z)zz .or. (x .neqv. y)zx .neqv. y .neqv. zzx .neqv. y .neqv. .not. zzx .neqv. z .neqv. .not. yzy .neqv. z .neqv. .not. x)r   r<   r9   r7   r8   r5   r6   r   s      r\   test_fcode_Xlogicalr   G  sv   gGAq!QE*&A  QA/vF  SVQ/vF  SVSVe4!9: : :SA.?!56 6 6 Aq!8KGGGAs1v&f=  CFA&f=  CFCF+6B     Z1%6!45 5 5 C1Iq)@  C1Iq)@  C1Iq)@  Z1%q)@  Z1%q)@  Z1%q)@   Bq!Ha(?  Bq!Ha(?  Bq!Ha(?  Jq!$a(?  Jq!$a(?  Jq!$a(?   C1u5q9!78 8 8C1u5q9!78 8 8C1u5q9!78 8 8Z1%q59!78 8 8Z1%q59!78 8 8Z1%q59!78 8 8 SAYE2&I  SAYE2&I  SAYE2&I  SA.2&I  SA.2&I  SA.2&I   R1Xq51H  R1Xq51H  R1Xq51H  C1u-q1H  C1u-q1H  C1u-q1H   Q1u-VD  Q3q6E2&I#$ $ $QAE2&I#$ $ $SVQE2&I#$ $ $r^   c                  |   t        d      \  } }t        t        | |d      d      dk(  sJ t        t        | |d      d      dk(  sJ t        t        | |d      d      d	k(  sJ t        t        | |d
      d      dk(  sJ t        t        | |d      d      dk(  sJ t        t        | |d      d      dk(  sJ y )Nr   z==rE   rF   zx == yz!=zx /= yz>=zx >= yz<=zx <= y>zx > y<zx < y)r   r<   r4   r   s     r\   test_fcode_Relationalr     s    5>DAqAq$'v>(JJJAq$'v>(JJJAq$'v>(JJJAq$'v>(JJJAq#&f=HHHAq#&f=HHHr^   c            	         t        d      } t        | | dk  f| dz  df      t        t        fd       t	        d      }d}||k(  sJ t	        t        | | dk  f| dz  df      d	
      dk(  sJ t        |       | z  }t        |       | z  }t        d      D ]  }t        ||       }t        ||       } d}t	        t        || dk  f|df      d
      }||k(  sJ t	        t        | | dk  f| dz  | dkD  ft        |       df      d      }d}||k(  sJ t        | | dk  f| dz  | dkD  ft        |       | dkD  f      t        t        fd       y )NrW   r~   rK   Tc                      t               S ru   r   rV   s   r\   rw   z&test_fcode_Piecewise.<locals>.<lambda>  s    d r^   rq   r   z      merge(x, x**2, x < 1)r   r   zR      if (x < 1) then
         var = x
      else
         var = x**2
      end ifr   a/        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                      t               S ru   r   r   s   r\   rw   z&test_fcode_Piecewise.<locals>.<lambda>  s    uT{ r^   )
r   r   rC   rx   r<   r   r   r   r   
ValueError)rW   codeexpectedabr   rV   s         @r\   test_fcode_Piecewiser     s   AaQZ!Q$.D
 34#D,H8Aq1u:1d|4F	   	AqAAqA2Y AJAJ
	  Aq1u:4y1\JD8Aq1u:1a!e}s1vtnEPRSDBH8aQZ!Q$AQQ@D
:*+r^   c                      t               } g d}| j                  |      }g d}|D ]  }t        |      dk  rJ  t        ||      D ]  \  }}||k(  rJ  t        |      t        |      k(  sJ y )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 *r   G      this = is + a + long + and + nasty + fortran + statement +   thatz'     @ * must + be + wrapped + properlyzF      this = is + a + long + and + nasty + fortran + statement + that*r   zH      this = is + a + long + and + nasty + fortran + statement +   that*r   H      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**r   zH      this = is + a + long + and + nasty + fortran + statement +  that**r   r   '     @ **must + be + wrapped + properlyr   r   r   z+     @ that**must + be + wrapped + properlyzE      this = is + a + long + and + nasty + fortran + statement(that)/r   zH      this = is + a + long + and + nasty + fortran +     statement(that)z&     @ /must + be + wrapped + properlyH   )r=   _wrap_fortranlenzip)rf   lineswrapped_linesexpected_lineslinewr   s          r\   test_wrap_fortranr    s    nGE" ))%0MN@  4yBM>2 1Avv}^!4444r^   c                  T    t               } g d}g d}| j                  |      |k(  sJ y )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)r  D      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  r  z     @ 10.0d0r=   r   rf   r   r   s      r\   test_wrap_fortran_keep_d0r	    s2    nGEH   '8333r^   c                  &    t        t        d        y )Nc                  .    t        t        d      d      S )Nr   garbage)method)r<   r   rk   r^   r\   rw   ztest_settings.<locals>.<lambda>6  s    eAaD; r^   )rC   	TypeErrorrk   r^   r\   test_settingsr  5  s    
9;<r^   c                  n    t        d      \  } }t        t        |       t        |      z   d      dk(  sJ y )Nr{   rE   rF   zsin(y) + cos(x))r   r<   r   r   r   s     r\   test_free_form_code_liner  9  s1    5>DAqQ#a&7;LLLLr^   c                      t        d      \  } }t        t        |       t        |      z   dz  j	                         d      }d}||k(  sJ y )Nr{   r   rE   rF   zsin(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)r   r<   r   r   r   )rW   rX   resultr   s       r\    test_free_form_continuation_liner  >  sO    5>DAqSVc!f_*224FKF	- 
 Xr^   c                  X    t        ddi      } dg}ddg}| j                  |      |k(  sJ y )NrG   rE   z^! 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_liner  I  sA    OV45GnoEQ"$H   '8333r^   c                  ,   t        dd      \  } }t        d      }t        d      }t        d      }t        d|      }t        d|       }d	}t        |||f   ||   z  ||   d
      }||ddiz  k(  s#||ddiz  k(  s||ddiz  k(  s||ddiz  k(  sJ y y y y )Nr   Tro   r   rW   rX   r   jz_do i = 1, m
   y(i) = 0
end do
do i = 1, m
   do j = 1, n
      y(i) = %(rhs)s
   end do
end dorE   r   rhszy(i) + A(i, j)*x(j)zy(i) + x(j)*A(i, j)zx(j)*A(i, j) + y(i)zA(i, j)*x(j) + y(i))r   r>   r?   r<   )	r|   r   r   rW   rX   r   r  r   r   s	            r\   
test_loopsr  R  s    5$'DAqCACACACACA	  1a411VDDH'<===H'<===H'<===H'<===? > > > > >r^   c                      t        ddt              \  } }t        d      }t        d      }t        | |      } d| j                  j
                  |j
                  dz  }t        ||    ||    d	      }||k(  sJ y )
Nzi mT)r0   clsrW   rX   zMdo i_%(icount)i = 1, m_%(mcount)i
   y(i_%(icount)i) = x(i_%(icount)i)
end do)icountmcountrE   r   )r   r   r>   r?   labeldummy_indexr<   )r   r   rW   rX   r   r   s         r\   test_dummy_loopsr!  l  s~    5$E2DAqCACAAq	A	 77&&!--@	AH
 11V<D8r^   c                  >   d} t        d| f      }t        d| f      }t        d| dz
  f      }t        d| dz
        }t        ||   ||dz      ||   z
  ||dz      ||   z
  z        }t        |j                  |j
                  d	      }|j                  d
      sJ y )Nr   rX   shaperW   Dyr~   r   F)r   contract+Dy(i) = (y(i + 1) - y(i))/(x(i + 1) - x(i)))r>   r?   r   r<   r  lhsendswith)len_yrX   rW   r%  r   r   code0s          r\   2test_fcode_Indexed_without_looking_for_contractionr,  {  s    ECx(ACx(A	T%'	,BCqAA1Q3!q1vad{+,A!%%15559E>>GHHHr^   c                     d} t        d| f      }t        d| f      }t        d| dz
  f      }t        d| dz
        }t        ||   ||dz      ||   z
  ||dz      ||   z
  z        }t        t	        |j
                  |j                              }|j                  d      sJ  G d	 d
t        t              |j                  fd|j                  t              D              }t        ||   ||dz      ||   z
  ||dz      ||   z
  z        }t        t	        |j
                  |j                              }|j                  d      sJ y )Nr   rX   r#  rW   r%  r~   r   r'  c                       e Zd Zy).test_element_like_objects.<locals>.ElementExprN)rh   ri   rj   rk   r^   r\   ElementExprr/    s    r^   r0  c              3   \   K   | ]#  }| |j                   |j                        f % y wru   )nameindices).0r   r0  s     r\   	<genexpr>z,test_element_like_objects.<locals>.<genexpr>  s%     Tq;qvvqyy12Ts   ),)r@   r?   r   r<   r$   r(  r  r)  r2   r   subsatomsrA   )r*  rX   rW   r%  r   r   r+  r0  s          @r\   test_element_like_objectsr8    s5   ECx(ACx(A	T%'	,BCqAA1Q3!q1vad{+,A*QUUAEE*+E>>GHHHgt  	
TAGGL<QTTAA1Q3!q1vad{+,A*QUUAEE*+E>>GHHHr^   c                       G d dt               }  |        }t        d      }|j                  t        |      d      dk(  sJ y )Nc                   @    e Zd Zej                  j                         Zy)1test_derived_classes.<locals>.MyFancyFCodePrinterN)rh   ri   rj   r=   _default_settingscopyrk   r^   r\   MyFancyFCodePrinterr;    s    (::??Ar^   r>  rW   borkz      bork = sin(x))r=   r   r   r   )r>  rf   rW   s      r\   test_derived_classesr@    sA    Bl B "#GA??3q66*.CCCCr^   c                  V    d} d}t        ddi      }|j                  |       }||k(  sJ y )Nau  subroutine 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
a  subroutine 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
rG   rE   )r=   indent_code)	codelinesr   rR   r  s       r\   test_indentrD    sC    	 <	 : 	ov./A]]9%FXr^   c            	      B   t        d      \  } }}t        | |z  t        d| z   |dkD  f|df      t        |      g      }t	        ddd      }t        ||      dk(  sJ t        d|d	   z  | dkD  f|d	   df      t        |d
         z   |d   z   }t        |d      dk(  sJ t	        ddd      }t	        ddd      }t        t        |d
         dt        |d	         g|d
   |d	   z   |d   dgd|d   z  |d
   z  t        |d         dz   dgg      }t        ||      dk(  sJ y )Nzx,y,zrK   r   Tr   r}   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))rK   r   )r~   r   r   r   rq   r   z?      merge(2*A(3, 1), A(3, 1), x > 0) + sin(A(2, 1)) + A(1, 1)rS   r   M)r}   r   )r   r   r   z      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)r   r:   r   r   r;   r<   r   r   )	rW   rX   rY   matr   rV   rS   rG  r   s	            r\   test_Matrix_printingrI    sp   gGAq!
!A#y!a%14y93q6B
CCS!QAa=	!" " " a$iQ'!D'49C$LH1T7RD#IK K K 	S!QAS!QAQsVaQsV-	
3!C&!D'1%	
1T71S6	4#<!+Q/1 	2A A;		 	 	r^   c            	          t        d      \  } }t        | t        ddd      t        || |z        g      }t	        |      }|dk(  sJ y )Nr   r   r   rK   z2      do x = 0, 9, 2
         y = x*y
      end do)r   r#   r"   r$   r<   )rW   rX   fsols       r\   test_fcode_ForrM    sS    5>DAqAuQAAq1u!5 67A
(C " # # #r^   c                     d } t        dd      }t        j                  |      }t        |      } | |d       t        d      \  }}t        |t        dt
        h      }t        |      } | |d	       t        |t        
      }t        |      }	 | |	d        | t        d        | t        d        | t        dt        t        i        | t        dt        t        i       y )Nc                 ,    t        | fddd||k(  sJ y )Nrq   rE   rr   r   )rV   refkwargss      r\   checkz%test_fcode_Declaration.<locals>.check  s!    TGBfGG3NNNr^   r   Tro   zinteger*4 :: ir   *   )valueattrszreal*4, parameter :: x = 42typezlogical :: yzreal*4zreal*8)type_aliases)	r   r'   deducedr&   r(   r*   r,   r)   r+   )
rR  r   var1dcl1rW   rX   var2dcl2bvar3dcl3s
             r\   test_fcode_Declarationr`    s    O 	T"AADtD	$ ! 5>DAqAwb>DE	%./AE"DtD	$	'8	'8	$g7	$g7r^   c                      t        ddd      } t        ddd      }t        ddd      }t        | d         dk(  sJ t        d| d   z        dk(  sJ |d   j                  || |z
        }t        |      d	k(  sJ y )
Nr   r~   r}   BCrF  z      A(1, 1)z      3*A(1, 1)z      (A - B)(1, 1))r;   r<   r6  )r   rb  rc  Fs       r\   test_MatrixElement_printingre  '  s    S!QAS!QAS!QA4>_,-,QtW!2232	$QAA8,,-,r^   c                  T    t        d      } t        t        | dd      d      dk(  sJ y )NrW   +r~   rE   rF   z	x = x + 1)r   r<   r%   rv   s    r\   test_aug_assignrh  4  s)    AAsA&f=LLLr^   c            
          t        d      } t        t        t        |       dkD  t	        | dd      g      d      dk(  sJ y )NrW   r~   -rE   rF   z)do while (abs(x) > 1)
   x = x - 1
end do)r   r<   r-   rP   r%   rv   s    r\   
test_Whilerk  9  sD    As1vzJq#q$9#:;6R	  r^   c                      t        d      } t        dd      }t        | t              }t        |t              }t	        t        d||g      t        t        fd       y )NrW   r|   Tro   rV  powerc                      t               S ru   r   )fp1s   r\   rw   z.test_FunctionPrototype_print.<locals>.<lambda>J      c
 r^   )r   r'   r+   r0   r.   rC   rx   )rW   r|   vxvnro  s       @r\   test_FunctionPrototype_printrs  B  sP    AT"A	!$	B	!'	"B
D'B8
4C  23r^   c                      t        d      } t        dd      }t        | t              }t        |t              }t	        | | |z        t        |       g}t        t        d||g|      t        t        fd       y )NrW   r|   Tro   rV  rm  c                      t               S ru   r   )fd1s   r\   rw   z/test_FunctionDefinition_print.<locals>.<lambda>V  rp  r^   )	r   r'   r+   r0   r$   r1   r/   rC   rx   )rW   r|   rq  rr  bodyrv  s        @r\   test_FunctionDefinition_printrx  M  sk    AT"A	!$	B	!'	"Bq!Q$+D
T7RHd
;C  23r^   N)sympy.core.addr   sympy.core.exprr   sympy.core.functionr   r   r   sympy.core.modr   
sympy.corer	   r
   r   sympy.core.numbersr   r   r   r   r   r   sympy.core.relationalr   sympy.core.singletonr   sympy.core.symbolr   r   (sympy.functions.combinatorial.factorialsr   $sympy.functions.elementary.complexesr   r   &sympy.functions.elementary.exponentialr   r   (sympy.functions.elementary.miscellaneousr   $sympy.functions.elementary.piecewiser   (sympy.functions.elementary.trigonometricr   r   r   '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/   r0   r1   r2   r3   r4   sympy.logic.boolalgr5   r6   r7   r8   r9   sympy.matricesr:   r;   sympy.printing.fortranr<   r=   sympy.tensorr>   r?   sympy.tensor.array.expressionsr@   rA   sympy.utilities.lambdifyrB   sympy.testing.pytestrC   r]   rm   ry   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r	  r  r  r  r  r  r!  r,  r8  r@  rD  rI  rM  r`  re  rh  rk  rs  rx  rk   r^   r\   <module>r     sk      8 8  9 9 C C $ " . > B = 9 : F F 9 . & 5 5! ! ! ! , , = = / 6 ) D 9 '	l-8@$>,
@
	G7>
D:0)3
M&>
	 ."b`$FI(,V95x44=M
4?4II&D=~ F#82
.M
4	4r^   