U
|
P ±dI ã @ s d dl Z d dlmZmZmZmZmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl m Z G dd de j Z dS )é N)ÚExprÚOpÚArithOpÚLanguageÚ as_symbolÚ as_numberÚ as_stringÚas_arrayÚ
|
as_complexÚas_termsÚ
|
as_factorsÚeliminate_quotesÚ insert_quotesÚ
|
fromstringÚas_exprÚas_applyÚas_numer_denomÚ
|
as_ternaryÚas_refÚas_derefÚ normalizeÚas_eqÚas_neÚas_ltÚas_gtÚas_leÚas_geé )Úutilc @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )Ú TestSymbolicc C sr dd }dD ]`}||d ||d | d ||d ||d | d ||d |d
|
| d q d S ) Nc S s& t | \}}t||}|| ks"td S ©N)r r ÚAssertionError)ÚsÚrÚdÚs1© r& úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/f2py/tests/test_symbolic.pyÚworker% s
|
z2TestSymbolic.test_eliminate_quotes.<locals>.worker)Ú Zmykind_z"1234" // "ABCD"z
|
"1234" // z"ABCD"z"1234" // 'ABCD'z'ABCD'z "1\"2'AB'34"za = z '1\'2"AB"34'r& )Úselfr( Úkindr& r& r' Útest_eliminate_quotes$ s z"TestSymbolic.test_eliminate_quotesc C s t d}t d}t d}|jtjks(tt|dks8t||ksDt||ksPtt|d k s`ttd}td}|jtjkstt|dkst||kst||ks¨tt|d k s¸ttd}td }|jtj ksØtt|d
|
ksèt||ksôt||kstt|d k stt
|
d d }t
|
d d} |jtj ks:tt|dksLt||ksZt|| kshtt|d k sztt d}
|
t d} |
|
jtj kstt|
|
dks¶tt|
|
|
|
|
|
ksÄt|
|
| ksÒtt||f} t|f} | jtjksútt| dks t| | kst| | ks(tt|}t|}|jtjksJtt|dks\t||ksjt||ksxtt|d k stt|}t|}|jtjks¬tt|dks¾t||ksÌt||ksÚtt|d k sìtt|||}t|||}|jtjkst||ks$t||ks2tt|d k sDtt||}t||}|jtjksjt||ksxt||kstt|d k std S )NÚxÚyÚzzExpr(Op.SYMBOL, 'x')é{ éÈ zExpr(Op.INTEGER, (123, 4))g(@gÍÌÌÌÌÌF@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 ÚSYMBOLr! ÚreprÚhashr ÚINTEGERZREALr
|
ZCOMPLEXr ÚSTRINGr ÚARRAYr ÚTERMSr ÚFACTORSr ZTERNARYr r Z
|
RELATIONAL)r* r- r. r/ ÚnÚmÚfnZfmÚcÚc2r" Ús2ÚaÚbÚtÚuÚvÚwÚeÚfr& r& r' Ú test_sanity2 s
|
|
|
|
|
zTestSymbolic.test_sanityc
|
C s t d}t d}t d}td}td}t||f}t||}t|dksNtt|dks^tt|dksntt|dks~ttttj|d idksttttj|d
|
id ks¶ttttj|d id ksÒttttj|didksîttttj|d |d idksttttj|d |d idks2ttttj|d
|
|didksTttttj|d|didksvttttj|d
|
|didksttttj |d idks¶ttttj |d
|
idksÔttttj |d idksòttttj |didksttttj |d |d idks2ttttj |d
|
|didksTtttj |d
|
ttj|d |d idi}t|d kstt|ttj |d
|
ttj |d |d idi}t|d ksÐtt|tttj
|
d d i fd ksðttttj
|
d |fi fd!ksttttj
|
d ||fi fd"ks6ttttj d |fd#ksTttt |||d$ksnttt ||d%ksttt||d&ksttt||d'ks¶ttt||d(ksÎttt||d)ksættt||d*ksþtd S )+Nr- r. r/ r0 r1 Ú123z
|
[123, 456]z
|
(123, 456)r r2 z2 * xéÿÿÿÿz-xéþÿÿÿz-2 * xúx + yz-x - yr3 z 2 * x + 3 * yz-2 * x + 3 * yéýÿÿÿz 2 * x - 3 * yúx ** 2zx ** -1zx ** -2úx * yzx ** 2 * y ** 3zx ** 2 * (x + y) ** 3zx ** 2 * (x * y) ** 3rL r& zf()úf(x)zf(x, y)úf[x]zmerge(y, z, x)úx .eq. yúx .ne. yúx .lt. yúx .le. yúx .gt. yúx .ge. y)r r r r
|
Ústrr! r r r= r> ÚAPPLYZINDEXINGr r r r r r r ) r* r- r. r/ r? r@ rE rB rI r& r& r' Útest_tostring_fortran sR
|
""""" ""$$ "$ z"TestSymbolic.test_tostring_fortranc C sä t j}td}td}td}td}ttj|dij|ddksFtttj|| dij|ddksjtttj|d ij|dd
|
kstt t
|
j ||j|dd ks¨tt t
|
j ||| j|dd ksÊtt t
|
j || || j|dd ksðt||| || | j|ddkstt |||j|ddks8tt ||j|ddksTtt||j|ddksptt||j|ddkstt||j|ddks¨tt||j|ddksÄtt||j|ddksàtd S )Nr- r. r/ r0 r2 ©Ú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 == yúx != yúx < yúx <= yúx > yúx >= y)r ÚCr r r r r> Útostringr! r r ÚDIVr r r r r r r )r* ra r- r. r/ r? r& r& r' Útest_tostring_c´ sr ÿþþ ÿþþÿÿÿÿÿÿÿÿÿÿÿÿ
|
z TestSymbolic.test_tostring_cc C s" t d}t d}t d}|| ttj|diks4t|| ttjdksLt|| ttj|d|dikslt|| ttj|d|dikst|| ttj|diks¨t|| ttj|d|diksÈt|
|
|ksÖt| ttj|diksútt| d| ttj|dikstd| ttj|dtddiks>td| d| ttj|d|diksht|| d ttj|d|dikst|d ttj|diks¬t|| d ttjttj|didttj|didttj|d|didiksþt|| | |d || ks t|| d |d d| | |d ksNt|| d || d d|d d|d kst|| | || || ks¦t||| || || ksÈt|d t t
|
j |tdksètd| d |ksþtd| d t t
|
j d| tdks&td | d d| ks@td
|
| d t t
|
j d
|
| tdkshtd | d d| kstd | d t t
|
j d
|
| tdksªtd |d |d d|d |d t t
|
j d
|
| d | ksîtd | d d
|
t t
|
j |tdks&td | d d
|
|d
|
| t t
|
j |d td
|
ksNt|d ttj|diksltt d}t d}|| ttj dkst|| ttj||fks´t|| ttj||fksÒttdd}| tddksòt|| tdkst|| tdks td S )Nr- r. r/ r2 )r r4 r rO r3 r4 é é é é é g @g à?r5 ú"123")z"ABC123"r g ð?g ð¿g Ày @ @y À @)r r r r= r! r: r> r8 r r r rl r r; ZCONCATr
|
r )r* r- r. r/ r" rG rB r& r& r' Útest_operationsÔ s $ &*& þúþ ".6"" (((( ÿ ÿ
|
ÿ(
|
z TestSymbolic.test_operationsc C sV t d}t d}t d}t||f}| ||i¡|ks:t|| ||i¡|| ksXt|| ||i¡|| ksvt|d ||i¡|d kst|| ||i¡|| ks²t| ||| i¡|| ksÐt| ||| i¡t|| |fksött||| ||| i¡t|| ||ks&tt|| ||| i¡t|| |ksRtd S )Nr- r. r/ r4 )r r Ú
|
substituter! r r )r* r- r. r/ rE r& r& r' Útest_substitute s& &ÿ
|
ÿÿ
|
z TestSymbolic.test_substitutec C s> t d}t d}t d}t d}td}td}t||f}td|ksLttd|ks\ttd| ksnttd || ksttd
|
|d ksttd || ksªttd |d ks¾ttd|| ksÒttdtjd|d ksìttdtjd|d ksttd|| | ks"ttd||ks8ttd|||ksPttd|| ksfttd|| | ksttd|kstttdtjd|| ks²ttd||ksÈttdtdd ksàttd |ksúttd td ||ksttd t|| | fks0ttd t d!ksFttd"t d!dks^ttd#t d!d$ksvttd%t d&d'ksttd(t d&d'ks¦ttd)t d&dks¾ttd*t d&d'ksÖttd+t d&dksîttd,t d&dksttd-t d&d'ks ttd.t d&dks6ttd/t d&d0ksNttd1t d2d'ksfttd3t
|
d dks~ttd4t
|
t d5t d6ksttd7tt d t dfksÀttd8t t d9|t d d:ksättd;t t d<td=t d>tt d?t d@fdAks ttdBt |||ks8ttdCt |ksNttdDt t |kshttdEt|ks~ttdFt |t | ksttdGt |t | ksºttdHt |t | ksØttdIt |t | ksöttdJt||ksttdKt||ks&ttdLt||ks>ttdMt||ksVttdNt||ksnttdOt||ksttdPtjdt||ks¤ttdQtjdt||ksÂttdRtjdt||ksàttdStjdt||ksþttdTtjdt||ks ttdUtjdt||ks:td S )VNr- r. r/ rL r5 rs z+ xz- xrQ zx + 1r rT zx * 2r2 rc rS r` z x ** 2 ** 3é z (x + y) * zrU zf(x,y)rV zf[x][y]z"ABC" // "123" zf("ABC")zMYSTRKIND_"ABC"Z MYSTRKINDz(/x, y/)z f((/x, y/))z (/(x+y)*z/)rN r0 Z123_2Z 123_myintkindZ myintkindz123.0g À^@r4 z123.0_4z123.0_8z123.0e0z123.0d0Z123d0z123e-0z123d+0z123.0_myrealkindZ
|
myrealkindZ3E4g LÝ@z(1, 2)z (1e2, PI)g Y@ÚPIz[1, 2]z POINT(x, y=1)ÚPOINT)r. z-PERSON(name="John", age=50, shape=(/34, 23/))ZPERSONz"John"é2 é" é )ÚnameZageÚshapezx?y:zz*xz**xz&xz (*x) * (*y)z (*x) * *yz*x * *yz*x**yrd re rf rh rg ri rW rX rY r[ rZ r\ )r r r r r! r ÚPythonr ZFortranr r
|
r r r r r r r r r r )r* r- r. r/ rL r" rG rE r& r& r' Útest_fromstring& s® ÿÿþ
|
ÿ "þÿüÿ
|
z TestSymbolic.test_fromstringc s t dt d}t d}t d}|ffdd } |¡|ksBt| |¡|ksTt| |¡|ksft|| |¡||kst| |¡||kst|| |¡|| ks´t|| |¡|| ksÎt| | |¡d| | ksðt||| |¡|||td ks tt| |¡t||ks@tt t fd d
|
}||| |¡ |hkst||hkstd d }t ||| |¡ |||hksÔtd d }t ||| |¡ hks td S )Nr- r. r/ rL c s | kr |S d S r r& )r" r# ©r- r& r' Ú replace_visit s z1TestSymbolic.test_traverse.<locals>.replace_visitr2 r c sV | j tjkr4| jd } |¡ |krR |¡ n | j tjkrR| krR | ¡ d S )Nr )r6 r r^ ÚdataÚaddÚremover7 )r" Úoper)Úfunction_symbolsÚsymbolsr& r' Úcollect_symbols s
|
|
z3TestSymbolic.test_traverse.<locals>.collect_symbolsc S s | j tjkr| | ¡ d S r )r6 r r7 r ©Úexprr r& r& r' Úcollect_symbols2® s z4TestSymbolic.test_traverse.<locals>.collect_symbols2c S s* | j tjkr| S | j tjkr&| | ¡ d S r )r6 r r^ r7 r r r& r& r' Úcollect_symbols3· s z4TestSymbolic.test_traverse.<locals>.collect_symbols3)r Ztraverser! r r Úset)r* r. r/ rL r r r r r& )r r r- r' Ú test_traverse sJ " ÿÿ ÿÿ
|
zTestSymbolic.test_traversec s¦ t d t d}t d} ¡tdtdfks6t d ¡tdtdfksXtd ¡tdtdfksztd d ¡tdtdfks ttd ¡tdtdfksÂt| ¡td|fksÜt|| ¡td|| fksþt | ¡td|fks t| | ¡||fks>t|| | ¡|| |fksft|| | ¡|| |fkstt t fdd ¡ d S )
|
Nr- r. r/ r r r2 r3 c s ¡S r )Ú linear_solver& r r& r' Ú<lambda>Ô ó z0TestSymbolic.test_linear_solve.<locals>.<lambda>)r r r r! ÚpytestZraisesÚ RuntimeError)r* r. r/ r& r r' Útest_linear_solve s ""&"" ((z TestSymbolic.test_linear_solvec C sô t d}t d}td}t||tdfks0tt|| ||fksHtt|| ||fks`tt|| ||fksxtt|| || tdfkstt||| ||| |fks¼tt||||| ||d || | fksðtd S )Nr- r. r0 r r2 )r r r r! ©r* r- r. r? r& r& r' Útest_as_numer_denomÖ s $z TestSymbolic.test_as_numer_denomc C sÚ t d}t d}td}| ¡ |hks*t| ¡ t ks<t|| ¡ || hksVt|| ¡ ||hkspt||| ¡ |||hkst|||| ¡ |||| hks¸t||| ¡ ||hksÖtd S )Nr- r. r0 )r r Zpolynomial_atomsr! r r r& r& r' Útest_polynomial_atomsã s (z"TestSymbolic.test_polynomial_atomsN)Ú__name__Ú
|
__module__Ú __qualname__r, rM r_ rm rt rv r r r r r r& r& r& r' r # s Q1 @^> r )!r Znumpy.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 ZF2PyTestr r& r& r& r' Ú<module> s t
|
