a khX@sddlZddlZddlZddlZddlmZddlmZddlZddlmZm Z m Z ddl m Z m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZddl m!Z!dd l"m#Z#dd l$m%Z%m&Z&ed \Z'Z(Z)Gd d d eZ*ej+ej,e*dddZ-ddZ.ddZ/ddZ0ddZ1ddZ2ddZ3ddZ4dd Z5d!d"Z6d#d$Z7d%d&Z8d'd(Z9d)d*Z:d+d,Z;d-d.Zz(/ 3 7) 2iz(/ -3 7)r<FZ auto_declarer6z (+ (/ 3 7) x)z$(declare-const x Real) (* (/ 3 7) x))r;rr'rr r&xr@r(r(r) test_Rational6s8 rKcCsttjgd}ttttd|ddks.Jttttd|ddksJJttttd|ddksfJtt ttd|ddksJtt ttd|ddksJtt ttd|dd ksJWdn1s0YdS) N FrI(assert (= x y))(assert (not (= x y)))(assert (<= x y))(assert (< x y))(assert (> x y))(assert (>= x y))) r;rr&rrrJyrrrrrr@r(r(r)test_RelationalCsrTcsttjgdĉttttdddks0JttttdddksNJtt ttdddkslJtt ttdddksJtt ttdddksJtt ttddd ksJWdn1s0Yt tfd d dS) NrLFrIrMrNrPrOrQrRcstttdSNr>)rrcomplexrJr(r@r(r)Vz,test_AppliedBinaryRelation..)r;rr&rreqrJrSneltlegtger ValueErrorr(r(r@r)test_AppliedBinaryRelationMs<r`cCsttjgd}tttd|ddks.Jtttd|ddksJJtttd|ddksfJtt td|ddksJtt td|ddksJtt td|dd ksJWdn1s0YdS) NFrIz(assert (> x 0))z(assert (< x 0))z(assert (= x 0))z(assert (<= x 0))z(assert (>= x 0))z(assert (not (= x 0)))) r;rr&rrZpositiverJnegativeZzero nonpositiveZ nonnegativeZnonzeror@r(r(r)test_AppliedPredicateYsrdcCs4ttjtjg2}tttttd|ddks4JWdn1sH0Yttjg@}ttttt t t it dit j di|ddksJWdn1s0Ytd}ttjg8}t|t|tt gtfi|d d ksJWdn1s0Ytg}t|t|tt gt fi|d d ks@Jtt|ttt |tt t gt fi|d d kstJtt|ttt |tt t gt fi|d tdi|ddksJWdn1s0Yttjgd<}tt|ttt |d tdi|ddksJWdn1s&0YdS)NFrIz(pow (sin x) (cos x))Z INTEGER_TYPEZABSOLUTE_VALUE_OF) symbol_tableZ known_typesknown_functionsr6z4(declare-const x INTEGER_TYPE) (ABSOLUTE_VALUE_OF x)f1rer6z:(declare-const x Bool) (declare-fun f1 (Bool) Real) (f1 x)zC(declare-const x Bool) (declare-fun f1 (Bool) Bool) (assert (f1 x))z|(declare-const x Int) (declare-const y Bool) (declare-const z Bool) (declare-fun f1 (Int Bool) Bool) (assert (= (f1 x z) y))Z MY_KNOWN_FUNz==)rerfr6zf(declare-const x Int) (declare-const y Bool) (declare-const z Bool) (assert (== (MY_KNOWN_FUN x z) y))rC)rfr6zg(declare-const x Real) (declare-const y Real) (declare-const z Real) (assert (== (MY_KNOWN_FUN x z) y)))r;rr&r'rrrJrabsintrSboolsympyZAbsr rfloatrz)r9Zmy_fun1r(r(r) test_Functionbsh@ & (    *  roc Csttjtjg*}ttdd|ddks,JWdn1s@0Yttjtjtjg.}tttdd|ddks~JWdn1s0Yttjtjg}tttddd|ddksJtdd d }td d d }td }dd}d||d||||d|}Wdn1s40YtgV}tt |dk|t ||k||d @t |dtddg|ddksJWdn1s0YttjtjgF}tt d|t t ||ddddddd|ddksJWdn1s 0YdS)NrCFrIz (pow x 3)z(pow x (pow y 3))r<z(pow x (/ 2 3))aTintegerbrealccSsd|S)Nr<r()rJr(r(r)grXztest_Pow..gg @r>z(declare-const a Int) (declare-const b Real) (declare-const c Bool) (assert (= (< a 2) c)) (assert (= (> b a) c)) (assert c) (assert (= (* (pow (* 7.0 a) (+ (pow b a) (* -1 a))) (pow (+ b (pow a 2)) -1)) (/ 7 3)))evaluater?zG(declare-const b Real) (declare-const c Real) (* -2 c (pow (* b b) -1))) r;rr&r'rrJrSr rrrr)r9rprsrvrwexprr(r(r)test_Pows:8<   H   *"r}cCsttjtjtjg*}tttd|ddks0JWdn1sD0Yttjtjtjg*}tttd|ddks~JWdn1s0Yttjtjg(}tt d|ddksJWdn1s0YdS)NFrIz(* x y)z(+ x y)z(* -1 x))r;rr&r'rrJrSr@r(r(r)test_basic_opss 88r~c sddlmddlmmmGfdddtd}ttj gN}t t ddft |t |t |t tgtfi|d d ksJWdn1s0Yttj gd d}t t ddftd d ftt t tt |t |t|t tgtfi|d dksJWdn1s*0Ytddd}tddd}td}tgT}t |d df|d dft||kt||k|B|ddksJWdn1s0YdS)Nr)Boolean)IntervalTuplesympifycsBeZdZfddZeddZeddZfddZd S) z*test_quantifier_extensions..ForAllcs0fdd|jD}dd||jgS)Nc s6g|].\}}}|jjj|||gqSr()_s_exprr2Z _known_typesre).0symstartend)rprinterr(r) s zFtest_quantifier_extensions..ForAll._smtlib..forall)limitsrfunction)selfrZbound_symbol_declarations)r)rr)_smtlibs  z2test_quantifier_extensions..ForAll._smtlibcSsdd|jDS)NcSsh|]\}}}|qSr(r()rs_r(r(r) rXzKtest_quantifier_extensions..ForAll.bound_symbols..)rrr(r(r) bound_symbolssz8test_quantifier_extensions..ForAll.bound_symbolscs&dd|jDfdd|jjDS)NcSsh|] }|jqSr(r2rrr(r(r)rrXzJtest_quantifier_extensions..ForAll.free_symbols..csh|]}|jvr|qSr(rrZbound_symbol_namesr(r)rs )rr free_symbolsrr(rr)rs z7test_quantifier_extensions..ForAll.free_symbolscsfdd|D}fdd|D}t|t|t|ksDJt|dksTJ|d}t|rjg||j|jRS|}t||g|_||_||_|S)Ncs"g|]}t|tfr|qSr() isinstancetuplerrp)rrr(r)rrXzFtest_quantifier_extensions..ForAll.__new__..csg|]}t|r|qSr()rr)rrr(r)rrXrxr)lenr__new__rrr_args)clsargsrrinst)rForAllrrr(r)rs" z2test_quantifier_extensions..ForAll.__new__N)r r!r"rpropertyrrrr(rrrrrr(r)rs    rfirhz,(assert (forall ( (x Real [-42, 21])) true))r<irCzs(declare-fun f (Real) Real) (assert (forall ( (x Real [-42, 21]) (y Real [-100, 3])) (=> (= x y) (= (f x) (f y)))))rpTrqrsrtrvdr>zx(declare-const c Bool) (assert (forall ( (a Int [2, 100]) (b Real [2, 100])) (or c (=> (< a b) (< (pow a (/ 1 2)) b))))))Zsympy.logic.boolalgrrlrrrr r;rr&rrJrrrmrSrrr)rr9rprsrvr(rr)test_quantifier_extensionssL +&*    rc Csttjg.}tdttdi|ddks,JWdn1s@0Yttjg<}tttddddtgtdi|ddksJWdn1s0Yttjgh}tttjtj tj tj tj tj tjddtj d tjd tj d itd td id|ddksJWdn1s0Yttjg\}tttjtj tj tj tj tj tjddtj d itd td id|ddksJWdn1s0YttjgP}tttjtj tj tj tj tj tjddtd id|ddksJWdn1s0YttjgX}tttjtj tj tj tj tj tjddtj d itd id|ddksfJWdn1s|0YdS)NrxZMY_PI)known_constantsr6z(pow MY_PI -1)gQ @Frzz&(assert (= MY_PI 3.14)) (pow MY_PI -1)prwr:plusrrC)rrf precisionr6z!(plus 0 1 -1 (/ 1 2) (exp 1) p g)z$(plus 0 1 -1 (/ 1 2) (exp 1) p 1.62))rfrr6z$(plus 0 1 -1 (/ 1 2) 2.72 3.14 1.62)z!(plus 0 1 -1 (/ 1 2) e 3.14 1.62))r;rr'rr rrr ZZeroZOneZ NegativeOneZHalfZExp1PiZ GoldenRatiorr@r(r(r)test_mix_number_mult_symbols7s& &   * **rc Cstgr}ttt@|ddks"JtttB|ddks:Jtt|ddksPJttt@t@|ddkslJWdn1s0Yttjg2}ttt@tdkB|ddksJWdn1s0Ytd}td }td }ttjg\}tt|ttt t|tg|t t gt f|t t gt fi|d d ksJJWdn1s`0Ytg\}tt |ttt t|tg|t t gt f|t t gt fi|d d ksJWdn1s0Ytg}tt |ttt ||ttt |||ttg|t tgt f|t t gt f|t t gtfi|d dks\JWdn1sr0YdS)Nr>z@(declare-const x Bool) (declare-const y Bool) (assert (and x y))z?(declare-const x Bool) (declare-const y Bool) (assert (or x y))z'(declare-const x Bool) (assert (not x))zY(declare-const x Bool) (declare-const y Bool) (declare-const z Bool) (assert (and x y z))rCzj(declare-const x Bool) (declare-const y Bool) (declare-const z Real) (assert (or (> z 3) (and x (not y))))rrwhrhz(declare-const x Bool) (declare-const y Real) (declare-const z Bool) (declare-fun f (Bool) Int) (declare-fun g (Bool) Int) (assert (> (f x) y)) (assert (< y (g z)))z(declare-const x Bool) (declare-const y Int) (declare-const z Bool) (declare-fun f (Bool) Int) (declare-fun g (Bool) Int) (assert (= (f x) y)) (assert (< y (g z)))z(declare-const x Real) (declare-const y Int) (declare-const z Bool) (declare-fun f (Real) Int) (declare-fun g (Int) Bool) (declare-fun h (Bool) Real) (assert (= (f x) y)) (assert (= (g (f x)) z)) (assert (= (h (g (f x))) x)))r;rrJrSrnrr&r rrrrkrjrrm)r9rrwrr(r(r) test_boolean}s\ :@  *   *   rcsNttjtjg<ttttdkftddfdddks>JWdn1sR0Yttjtjg\tttdtdkftdtdkftdtdkftd dfddd ksJWdn1s0Ytttdkftdtdkftttd kfttjtjg$tt fd d Wdn1s@0YdS)Nrxr<TFrIz(ite (< x 1) x (pow x 2))rCrBzQ(ite (< x 1) (pow x 2) (ite (< x 2) (pow x 3) (ite (< x 3) (pow x 4) (pow x 5))))rcs tdSrUrr(r|r9r(r)rWrXz'test_smtlib_piecewise..) r;rr&r'rrrJrrAssertionErrorr(r(rr)test_smtlib_piecewises2  & & ,rcCsJtttdkftddf}ttjtjg(}td||ddksDJWdn1sX0Yttjtjg(}t|t|ddksJWdn1s0Yttjtjtjg,}t|tt|ddksJWdn1s0Yttjtjg*}t|d|dd ks&JWdn1s<0YdS) Nrxr<Tr>z6(declare-const x Real) (* 2 (ite (< x 1) x (pow x 2)))z?(declare-const x Real) (* (pow x -1) (ite (< x 1) x (pow x 2)))za(declare-const x Real) (declare-const y Real) (* (pow x -1) (pow y -1) (ite (< x 1) x (pow x 2)))rCz<(declare-const x Real) (* (/ 1 3) (ite (< x 1) x (pow x 2))))rrJr;rr&r'rrS)pwr9r(r(r)!test_smtlib_piecewise_times_consts66:rcCstg}tdd|ddks Jtd|ddks4Jttj|ddksJJttj|ddks`Jtd|ddkstJtdd|ddksJWdn1s0YdS) NTF)Z auto_assertr6truer>z (assert true)z(assert false)false)r;rr rrr@r(r(r)test_smtlib_booleans rcstdttjtjg$ttfddWdn1sB0Yttjg"ttfddWdn1s0YdS)Nrcs&tttttgtfidS)Nrh)rrJdiffrrmr(rr9r(r)rW!rXz$test_not_supported..csttjdSrU)rr ZComplexInfinityr(r@r(r)rW#rX)r r;rr&r'rKeyErrorr(r(rr)test_not_supporteds 2rcCs4tddksJtddks Jtddks0JdS)Ngz0.0geK contextlibr0r#typingenumrrrlrrrZ sympy.corerrr r r r r rrrrrrrrZsympy.functionsrrrrZsympy.assumptions.askrZsympy.printing.smtlibrZsympy.testing.pytestrrrJrSrnrcontextmanagerIterabler;rArKrTr`rdror}r~rrrrrrrrr(r(r(r)sB  <       @.XFX#