y.worker% s3 #A&&DAq q!$$B777777 ) mykind_z"1234" // "ABCD"z
"1234" // z"ABCD"z"1234" // 'ABCD'z'ABCD'z "1\"2'AB'34"za = z '1\'2"AB"34' )selfr* kinds r) test_eliminate_quotesz"TestSymbolic.test_eliminate_quotes$ s
$ 6 6D F4,, - - - F4,&-8 9 9 9 F4.. / / / F4,&-8 9 9 9 F4++ , , , F6D=#44 5 5 5 5
6 6r+ c ( t d }t d }t d }|j t j k sJ t | dk sJ ||k sJ ||k sJ t | J t
d }t
d }|j t j k sJ t | dk sJ ||k sJ ||k sJ t | J t
d }t
d }|j t j k sJ t | d
k sJ ||k sJ ||k sJ t | J t d d }t d
d } |j t j
k sJ t | dk sJ ||k sJ || k sJ t | J t d }
t d } |
j t j k sJ t |
dk sJ t |
|
|
k sJ |
| k sJ t ||f } t |f }
| j t j k sJ t | dk sJ | | k sJ | |
k sJ t | }t | }|j t j k sJ t | dk sJ ||k sJ ||k sJ t | J t# | }t# | }|j t j k sJ t | dk sJ ||k sJ ||k sJ t | J t' ||| }t' ||| }|j t j k sJ ||k sJ ||k sJ t | J t+ || }t- || }|j t j k sJ ||k sJ ||k sJ t | J d S )NxyzzExpr(Op.SYMBOL, 'x'){ zExpr(Op.INTEGER, (123, 4))g(@gF@zExpr(Op.REAL, (12.3, 4))r zFExpr(Op.COMPLEX, (Expr(Op.INTEGER, (1, 4)), Expr(Op.INTEGER, (2, 4))))z'123'"ABC"z Expr(Op.STRING, ("'123'", 1))zHExpr(Op.ARRAY, (Expr(Op.INTEGER, (123, 4)), Expr(Op.INTEGER, (456, 4))))z)Expr(Op.TERMS, {Expr(Op.SYMBOL, 'x'): 1})z+Expr(Op.FACTORS, {Expr(Op.SYMBOL, 'x'): 1}))r opr SYMBOLreprhashr INTEGERREALr COMPLEXr STRINGr
ARRAYr TERMSr
FACTORSr TERNARYr r
RELATIONAL)r/ r3 r4 r5 nmfnfmcc2r% s2abtuvwefs r) test_sanityzTestSymbolic.test_sanity2 s[ cNN cNN cNNtry Aww00000AvvvvAvvvvAww""" cNN cNNtrz!!!!Aww66666AvvvvAvvvvAww"""
t__
t__u Bxx55555RxxxxRxxxxBxx### q!
1
trz!!!!Aww 9 : : : :AvvvvBwwwwAww""" g
w
try Aww;;;;T!WW;;;AvvvvBwwww aV aUOOtrx Aww ; < < < <AvvvvAvvvv QKK QKKtrx AwwEEEEEAvvvvAvvvvAww""" qMM qMMtrz!!!!AwwGGGGGAvvvvAvvvvAww""" q!Q q!Q trz!!!!AvvvvAvvvvAww""" !QKK !QKKtr}$$$$AvvvvAvvvvAww"""""r+ c $ t d }t d }t d }t d }t d }t ||f }t || }t | dk sJ t | dk sJ t | dk sJ t | dk sJ t t t j |d i dk sJ t t t j |d
i d k sJ t t t j |d i d
k sJ t t t j |di dk sJ t t t j |d |d i dk sJ t t t j |d |d i dk sJ t t t j |d
|di dk sJ t t t j |d|di dk sJ t t t j |d
|di dk sJ t t t j |d i dk sJ t t t j |d
i dk sJ t t t j |d i dk sJ t t t j |di dk sJ t t t j |d |d i dk sJ t t t j |d
|di dk sJ t t j |d
t t j |d |d i di }t | d k sJ t | t t j |d
t t j |d |d i di }t | d k sJ t | t t t j d d i f d k sJ t t t j d |fi f d!k sJ t t t j d ||fi f d"k sJ t t t j
d |f d#k sJ t t ||| d$k sJ t t || d%k sJ t t || d&k sJ t t || d'k sJ t t || d(k sJ t t! || d)k sJ t t# || d*k sJ d S )+Nr3 r4 r5 r6 r7 123z
[123, 456]z
(123, 456)r r8 z2 * xz-xz-2 * xx + yz-x - yr9 z
2 * x + 3 * yz-2 * x + 3 * yz
2 * x - 3 * yx ** 2zx ** -1zx ** -2x * yzx ** 2 * y ** 3zx ** 2 * (x + y) ** 3zx ** 2 * (x * y) ** 3rW r. zf()f(x)zf(x, y)f[x]zmerge(y, z, x)x .eq. yx .ne. yx .lt. yx .le. yx .gt. yx .ge. y)r r r
r strr r rE rF APPLYINDEXINGr r r r r r r ) r/ r3 r4 r5 rI rJ rP rM rT s r) test_tostring_fortranz"TestSymbolic.test_tostring_fortran s~ cNN cNN cNN cNN cNN aV q! 1vv}}}}1vv1vv %%%%1vv %%%%41a&))**c111141a&))**g555541b'**++t333341b'**++x777741aA,//00G;;;;41b!R.1122h>>>>41aA,//00OCCCC41b!Q-00115EEEEE41aB-0011_DDDD4
QF++,,33334
QF++,,88884
QG,,--::::4
QG,,--::::4
Q1aL1122g====4
Q1aL11226GGGGG aDAq!Q<$@$@! D E E1vv0000#a&&000 aDaAq\$B$BA F G G1vv0000#a&&00043B-0011U::::43r"23344>>>>43A"34455BBBB4 c1X..//69999:aA&&''+;;;;;5A;; :----5A;; :----5A;; :----5A;; :----5A;; :----5A;; :------r+ c t j }t d }t d }t d }t d }t t
j |di | dk sJ t t
j ||z di | dk sJ t t
j |d i | d
k sJ t t j
|| | d k sJ t t j
|||z | d k sJ t t j
||z
||z | d
k sJ |||z
||z z z |z | dk sJ t ||| | dk sJ t || | dk sJ t || | dk sJ t || | dk sJ t || | dk sJ t! || | dk sJ t# || | dk sJ d S )Nr3 r4 r5 r6 r8 languagezx * xz(x + y) * (x + y) z
pow(x, 12)x / yz x / (x + y)z(x - y) / (x + y)z123 + x + (x - y) / (x + y)z(x?y:z)x == yx != yx < yx <= yx > yx >= y)r Cr r r r rF tostringr r DIVr r r r r r r )r/ ro r3 r4 r5 rI s r) test_tostring_cz TestSymbolic.test_tostring_c s6 : cNN cNN cNN cNNBJA''00(0CCwNNNNRZ
E1"
8X8 & &*=> > > > BJ
r!
8X8 & &,7 7 7 7 Q #8X866'B B B BaQ (( ; ;}M M M Ma!eQ. $HhH 7 7;NO O O OQUq1u%%8X8..2OP P P P !Q""++X+>>)KKKKQ{{##X#66(BBBBQ{{##X#66(BBBBQ{{##X#66'AAAAQ{{##X#66(BBBBQ{{##X#66'AAAAQ{{##X#66(BBBBBBr+ c t d }t d }t d }||z t t j |di k sJ ||z
t t j d k sJ ||z t t j |d|di k sJ ||z
t t j |d|di k sJ ||z t t j |di k sJ ||z t t j |d|di k sJ |
|k sJ | t t j |di k sJ t
| d|z t t j |di k sJ d|z t t j |dt d di k sJ d|z d|z z t t j |d|di k sJ ||z dz t t j |d|di k sJ |dz t t j |di k sJ ||z dz t t j t t j |di dt t j |di dt t j |d|di di k sJ ||z |z |dz ||z z k sJ ||z dz |dz d|z |z z |dz z k sJ ||z dz ||z
dz z d|dz z d|dz z z k sJ ||z |z ||z ||z z k sJ |||z z ||z ||z z k sJ |dz t t j
|t d k sJ d|z dz |k sJ d|z dz t t j
d|z t d k sJ d |z dz d|z k sJ d
|z dz t t j
d
|z t d k sJ d |z dz d|z k sJ d |z d z t t j
d
|z t d k sJ d
|dz z |d z z d|dz z |dz z z t t j
d
|z d |z k sJ d |z d z d
z t t j
|t d k sJ d |z d z d
z |d
|z z t t j
|dz t d
k sJ |dz t t j |di k sJ t d }t d }||z t t j d k sJ ||z t t j
||f k sJ ||z t t j
||f k sJ t dd }| t dd k sJ ||z t d k sJ ||z t d k sJ d S )Nr3 r4 r5 r8 )r r: r r[ r9 r: g @g ?r; "123")z"ABC123"r g ?g g y @ @y @)r r r rE r@ rF r> r r r rz r rC CONCATr r )r/ r3 r4 r5 r% rR rM s r) test_operationsz TestSymbolic.test_operations s cNN cNN cNN1uRX1v......1uRZ0000001uRX1a|4444441uRX1a}5555551uRZ!Q0000001uRZ!Q1 666666rQwwwwrT"(QG,,,,,dA2hh,,,1uRX1v......1uRX1illA'>??????1uq1u}RX1a| < <<<<<A{d28aAq\ : :::::!ttBJA//////AzT HRZ!Q((!RZ!Q((!RZqq"
A{adQUl****AzQTAEAI -1 44444AzQUQJ&!ad(QAX*=====A{a!ea!em++++AE{a!ea!em++++A(7;9Q<<@@@@@@A a A hw{AE9Q<< H HHHHHA a!e####A hw{AE9Q<< H HHHHHA a!e####! aHW[!a%1 $N$NNNNNQT AqD BAI1$45( KQA;' ;' ' ' ' 'a!q Xgk1.7ll&< &< < < <>@1fqjA=M< < <QU adIaLL I IIIIICDAs8 4 44444 g g Avbi 999999Avbi!Q000000Avbi!Q000000 sC rZd++++++1u ------1u ,,,,,,,,r+ c ~ t d }t d }t d }t ||f }| ||i |k sJ ||z ||i ||z k sJ ||z ||i ||z k sJ |dz ||i |dz k sJ ||z ||i ||z k sJ | |||z i ||z k sJ | |||z i t ||z |f k sJ t ||| |||z i t ||z || k sJ t || |||z i t ||z | k sJ d S )Nr3 r4 r5 r: )r r
substituter r )r/ r3 r4 r5 rP s r) test_substitutez TestSymbolic.test_substitute s cNN cNN cNN aV ||QF##q((((A!!1a&))QU2222A!!1a&))QU22221 !Q((AqD0000A!!1a&))QU2222||QAJ''1q50000||QAJ''8QUAJ+?+?????!Q 'ZAE
3 3z!a%A7N7NO O O OQ{{%%q!a%j11U1q5!__DDDDDDr+ c t d }t d }t d }t d }t d }t d }t ||f }t d |k sJ t d |k sJ t d | k sJ t d ||z k sJ t d
|d z k sJ t d ||z k sJ t d
|dz k sJ t d ||z k sJ t dt j |dz k sJ t dt j |dz k sJ t d ||z |z k sJ t d || k sJ t d ||| k sJ t d || k sJ t d || | k sJ t d |k sJ t
t dt j ||z k sJ t d || k sJ t d t dd k sJ t d |k sJ t d t d || k sJ t d t ||z |z f k sJ t d t d! k sJ t d" t d!d k sJ t d# t d!d$ k sJ t d% t d&d' k sJ t d( t d&d' k sJ t d) t d&d k sJ t d* t d&d' k sJ t d+ t d&d k sJ t d, t d&d k sJ t d- t d&d' k sJ t d. t d&d k sJ t d/ t d&d0 k sJ t d1 t d2d' k sJ t d3 t d d k sJ t d4 t t d5 t d6 k sJ t d7 t t d t d f k sJ t d8 t t d9 |t d : k sJ t d; t t d< t d= t d> t t d? t d@ f A k sJ t dB t ||| k sJ t dC t | k sJ t dD t t | k sJ t dE t | k sJ t dF t | t | z k sJ t dG t | t | z k sJ t dH t | t | z k sJ t dI t | t | z k sJ t dJ t || k sJ t dK t || k sJ t dL t! || k sJ t dM t# || k sJ t dN t% || k sJ t dO t' || k sJ t dPt j t || k sJ t dQt j t || k sJ t dRt j t! || k sJ t dSt j t# || k sJ t dTt j t% || k sJ t dUt j t' || k sJ d S )VNr3 r4 r5 rW r; r z+ xz- xr] zx + 1r r` zx * 2r8 rq r_ rn z x ** 2 ** 3 z (x + y) * zra zf(x,y)rb zf[x][y]z"ABC" // "123" zf("ABC")zMYSTRKIND_"ABC" MYSTRKINDz(/x, y/)z f((/x, y/))z (/(x+y)*z/)rZ r6 123_2
123_myintkind myintkindz123.0g ^@r: z123.0_4z123.0_8z123.0e0z123.0d0123d0z123e-0z123d+0z123.0_myrealkind
myrealkind3E4g L@z(1, 2)z (1e2, PI)g Y@PIz[1, 2]z
POINT(x, y=1)POINT)r4 z-PERSON(name="John", age=50, shape=(/34, 23/))PERSONz"John"2 " )nameageshapezx?y:zz*xz**xz&xz (*x) * (*y)z (*x) * *yz*x * *yz*x**yrr rs rt rv ru rw rc rd re rg rf rh )r r r
r r Pythonr Fortranr r r r r r r r r r r r )r/ r3 r4 r5 rW r% rR rP s r) test_fromstringz TestSymbolic.test_fromstring& sP cNN cNN cNN cNN g g aV #!####% A%%%%&!!aR'''''""a!e++++'""a!e++++'""a!e++++'""a!e++++'""a!e++++(X_===AEEEE-(/BBBagMMMM-((QUaK7777&!!QQqTT))))(##qqAww....&!!QqT)))))$$!Q////'""a'''' ( ( 0
2
2
23 3671f= = = = *%%1----+,, ';0O0OOOOO*%%***Jz,B,B***-((AAaDD0000-((Hq1uk_,E,EEEEE% IcNN2222'""iQ&7&77777/**i[.I.IIIII'""iq&9&99999)$$ %(;(;;;;;)$$ %(;(;;;;;)$$ %(;(;;;;;)$$ %(;(;;;;;'""iq&9&99999(##y':':::::(##y':':::::,--5,1O1OOOOO% Igq$9$99999(##z!Q'7'77777+&&*Yu5E5E5>t__+F +F F F F F (##x1 y||0L'M'MMMMM/**hy7I7I789B1 /G /G /G G G G G ;= =@H(## x(( bMM "
y}} =>> A A A '""jAq&9&99999$ 8A;;....% HXa[[$9$99999$ 6!99,,,,-((HQKK(1++,EEEEE+&&(1++ *CCCCC)$$ hqkk(AAAAA'""hqkkHQKK&?????(##uQ{{2222(##uQ{{2222'""eAqkk1111'""eAqkk1111(##uQ{{2222(##uQ{{2222*x/?@@@E!QKKOOOO*x/?@@@E!QKKOOOO*x/?@@@E!QKKOOOO*x/?@@@E!QKKOOOO*x/?@@@E!QKKOOOO*x/?@@@E!QKKOOOOOOr+ c
t d
t d }t d }t d }|f
fd }
| |k sJ | | |k sJ | | |k sJ || | || k sJ |
| || k sJ || | || k sJ || | || k sJ
|z |z | d|z |z k sJ
||
|z
z | | ||t d z k sJ t
| | t || k sJ t t fd}
||
|z
z | |hk sJ
||hk sJ d }t
||
|z
z |
|||hk sJ d
}t
||
|z
z |
hk sJ d S ) Nr3 r4 r5 rW c | k r|S d S r$ r. )r% r& r3 s r)
replace_visitz1TestSymbolic.test_traverse..replace_visit s Avv vr+ r8 r c | j t j u r?| j d } | |v r | d S d S | j t j u r| vr | d S d S d S )Nr )r< r rj dataaddremover= )r% operfunction_symbolssymbolss r) collect_symbolsz3TestSymbolic.test_traverse..collect_symbols s trx vay $$T***7??NN4((((( #?""q0@'@'@ A #"'@'@r+ c Z | j t j u r| | d S d S r$ )r< r r= r exprr s r) collect_symbols2z4TestSymbolic.test_traverse..collect_symbols2 s1 w")## D!!!!! $#r+ c | j t j u r| S | j t j u r| | d S d S r$ )r< r rj r= r r s r) collect_symbols3z4TestSymbolic.test_traverse..collect_symbols3 sE w"("" w")## D!!!!! $#r+ )r traverser r set) r/ r4 r5 rW r r r r r r r3 s @@@r)
test_traversezTestSymbolic.test_traverse sC cNN cNN cNN cNN ! zz-((A----zz-((A----zz-((A----!}--15555!}--15555!}--15555!}--15555A ##M22q1uqyAAAA!QU %Xm 4 49:1ill9K9K:LM M M M Q{{##M22eAqkkAAAA 55%%
QQq!a%[[ ""?333 A3&&&&1a)#### " " " %%
QQq!a%[[ ""#3W===1aA,&&&& " " " %%
QQq!a%[[ ""#3W===1#~~~~~~r+ c t d t d }t d } t d t d fk sJ dz t d t d fk sJ dz t d t d fk sJ dz dz t d t d fk sJ t d t d t d fk sJ | t d |fk sJ ||z t d ||z fk sJ |z t d |fk sJ |z |z ||fk sJ ||z z |z ||z |fk sJ ||z z |z ||z |fk sJ t j t
fd d S ) Nr3 r4 r5 r r r8 r9 c 4 z S r$ ) linear_solve)r3 s r) z0TestSymbolic.test_linear_solve.. s QU,@,@,C,C r+ )r r r pytestraises RuntimeError)r/ r4 r5 r3 s @r) test_linear_solvez TestSymbolic.test_linear_solve sc cNN cNN cNN~~a Yq\\9Q<<$@@@@@A##A&&9Q<<1 *FFFFFA##A&&9Q<<1 *FFFFFA ''**y||Yq\\.JJJJJ||((++ ! ill/KKKKK~~a Yq\\1$55555A##A&&9Q<<Q*?????A##A&&9Q<<*;;;;;A ''**q!f4444Q! a--a00QUAJ>>>>A A
++A..1q5!*<<<<
l$C$C$C$CDDDDDr+ c < t d }t d }t d }t | |t d fk sJ t ||z ||fk sJ t ||z ||fk sJ t ||z ||fk sJ t ||z ||z t d fk sJ t |||z z |||z z |fk sJ t |||||z z
z z ||dz z ||z |z
fk sJ d S )Nr3 r4 r6 r r8 )r r r r/ r3 r4 rI s r) test_as_numer_denomz TestSymbolic.test_as_numer_denom sH cNN cNN cNN a Q ! $55555 a!e$$A.... a!e$$A.... a!e$$A.... a!e$$Q ! (===== a!a%i((QQYN:::: a!q1q5y/ 122q1a4xQ6KKKKKKKr+ c t d }t d }t d }| |hk sJ | t k sJ || || hk sJ || || hk sJ || |z || |hk sJ || || z || || hk sJ || |z || hk sJ d S )Nr3 r4 r6 )r r polynomial_atomsr r s r) test_polynomial_atomsz"TestSymbolic.test_polynomial_atoms sd cNN cNN cNN!!##s****!!##suu,,,,!&&((QqTF2222!&&((QQqTTF2222!q**,,1q 9999!qt --//AAaDD!A$<????!a))++!v555555r+ N)__name__
__module__ __qualname__r1 rX rl r{ r r r r r r r r. r+ r) r! r! # s 6 6 6 O# O# O#b/. /. /.b C C C@>- >- >-@E E E$\P \P \P|< < < |E E E( L L L 6 6 6 6 6r+ r! )!r numpy.f2py.symbolicr 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 F2PyTestr! r. r+ r) r s
: K6 K6 K6 K6 K64= K6 K6 K6 K6 K6r+