U L?hW@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@r&xrAr(r(r) test_Rational6s$ rLc Csttjgd}ttttd|ddks.tttttd|ddksJttt ttd|ddksfttt ttd|ddksttt ttd|ddksttt ttd|dd kstW5QRXdS) N FrJ(assert (= x y))(assert (not (= x y)))(assert (<= x y))(assert (< x y))(assert (> x y))(assert (>= x y))) r;rr&rrrKyr@rrrrrrAr(r(r)test_RelationalCsrUc sttjgdttttdddks0ttt ttdddksNttt ttdddkslttt ttdddksttt ttdddksttt ttddd kstW5QRXttfd d dS) NrMFrJrNrOrQrPrRrScstttdSNr>)rrcomplexrKr(rAr(r)Vz,test_AppliedBinaryRelation..)r;rr&rreqrKrTr@neltlegtger ValueErrorr(r(rAr)test_AppliedBinaryRelationMs(rac Csttjgd}tttd|ddks.ttttd|ddksJttt td|ddksfttt td|ddksttt td|ddksttt td|dd kstW5QRXdS) NFrJz(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&rrZpositiverKr@negativezero nonpositiveZ nonnegativeZnonzerorAr(r(r)test_AppliedPredicateYsrfc Csttjtjg(}tttttd|ddks4tW5QRXttjg6}tt ttt t t it dit jdi|ddks|tW5QRXtd}ttjg.}t|t|tt gtfi|d d kstW5QRXtg}t|t|tt gt fi|d d ksttt|ttt |tt t gt fi|d d ks6ttt|ttt |tt t gt fi|d tdi|ddksttW5QRXttjgd2}tt|ttt |d tdi|ddkstW5QRXdS)NFrJz(pow (sin x) (cos x))Z INTEGER_TYPEZABSOLUTE_VALUE_OF) symbol_tableZ known_typesknown_functionsr6z4(declare-const x INTEGER_TYPE) (ABSOLUTE_VALUE_OF x)f1rgr6z:(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==)rgrhr6zf(declare-const x Int) (declare-const y Bool) (declare-const z Bool) (assert (== (MY_KNOWN_FUN x z) y))rD)rhr6zg(declare-const x Real) (declare-const y Real) (declare-const z Real) (assert (== (MY_KNOWN_FUN x z) y)))r;rr&r'rrrKrr@absintrTboolsympyZAbsr rfloatrz)r9Zmy_fun1r(r(r) test_Functionbsh,        rqc Csttjtjg }ttdd|ddks,tW5QRXttjtjtjg$}tttdd|ddksjtW5QRXttjtjgv}tttddd|ddkstt dd d }t d d d }t d }dd}d||d||||d|}W5QRXtgL}tt |dk|t ||k||d @t |dtddg|ddksPtW5QRXttjtjg<}tt d|t t ||ddddddd|ddkstW5QRXdS)NrDFrJz (pow x 3)z(pow x (pow y 3))r<z(pow x (/ 2 3))aTintegerbrealccSsd|S)Nr<r()rKr(r(r)grYztest_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'rrKr@rTr rrrr)r9rrrurxryexprr(r(r)test_Pows:$(   2   "rc Csttjtjtjg }tttd|ddks0tW5QRXttjtjtjg }tttd|ddksjtW5QRXttjtjg}tt d|ddkstW5QRXdS)NFrJz(* x y)z(+ x y)z(* -1 x))r;rr&r'rrKrTr@rAr(r(r)test_basic_opss $$rc sddlmddlmmmGfdddtd}ttj gD}t t ddft |t |t |t tgtfi|d d kstW5QRXttj gd X}t t ddftd d ftt t tt |t |t|t tgtfi|d dkstW5QRXtddd}tddd}td}tgJ}t |d df|d dft||kt||k|B|ddksvtW5QRXdS)Nr)Boolean)IntervalTuplesympifycsBeZdZfddZeddZeddZfddZd S) z*test_quantifier_extensions..ForAllcs0fdd|jD}dd||jgS)Nc s6g|].\}}}|jjj|||gqSr()_s_exprr2Z _known_typesrg).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) szKtest_quantifier_extensions..ForAll.bound_symbols..)rrr(r(r) bound_symbolssz8test_quantifier_extensions..ForAll.bound_symbolscs&dd|jDfdd|jjDS)NcSsh|] }|jqSr(r2rrr(r(r)rszJtest_quantifier_extensions..ForAll.free_symbols..csh|]}|jkr|qSr(rrZbound_symbol_namesr(r)rs )rr free_symbolsrr(rr)rs z7test_quantifier_extensions..ForAll.free_symbolscsfdd|D}fdd|D}t|t|t|ksDtt|dksTt|d}t|rjf||j|jfS|}t||g|_||_||_|S)Ncs"g|]}t|tfr|qSr() isinstancetuplerrr)rrr(r)rszFtest_quantifier_extensions..ForAll.__new__..csg|]}t|r|qSr()rr)rrr(r)rs rzr)lenr@r__new__rrr_args)clsargsrrinst)rForAllrrr(r)rs"  z2test_quantifier_extensions..ForAll.__new__N)r r!r"rpropertyrrrr(rrrrrr(r)rs    rfirjz,(assert (forall ( (x Real [-42, 21])) true))r<irDzs(declare-fun f (Real) Real) (assert (forall ( (x Real [-42, 21]) (y Real [-100, 3])) (=> (= x y) (= (f x) (f y)))))rrTrsrurvrxdr>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.boolalgrrnrrrr r;rr&rrKrrror@rTrrr)rr9rrrurxr(rr)test_quantifier_extensionssP +   rc Csttjg$}tdttdi|ddks,tW5QRXttjg2}tttddddtgtdi|ddksptW5QRXttjg\}tttj tj tj tj tj tjtjddtjd tjd tj d itd td id|ddkstW5QRXttjgR}tttj tj tj tj tj tjtjddtjd itd td id|ddksBtW5QRXttjgF}tttj tj tj tj tj tjtjddtd id|ddkstW5QRXttjgN}tttj tj tj tj tj tjtjddtj d itd id|ddkstW5QRXdS)NrzZMY_PI)known_constantsr6z(pow MY_PI -1)gQ @Fr|z&(assert (= MY_PI 3.14)) (pow MY_PI -1)pryr:plusrrD)rrh precisionr6z!(plus 0 1 -1 (/ 1 2) (exp 1) p g)z$(plus 0 1 -1 (/ 1 2) (exp 1) p 1.62))rhrr6z$(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 r@rrr ZZeroZOneZ NegativeOneZHalfZExp1PiZ GoldenRatiorrAr(r(r)test_mix_number_mult_symbols7s    rc Cstgh}ttt@|ddks"ttttB|ddks:ttt|ddksPtttt@t@|ddksltW5QRXttjg(}ttt@tdkB|ddkstW5QRXtd}td }td }ttjgR}tt |ttt t|tg|t t gt f|t t gt fi|d d ks"tW5QRXtgR}tt|ttt t|tg|t t gt f|t t gt fi|d d kstW5QRXtgx}tt|ttt||ttt|||ttg|t tgt f|t t gt f|t t gtfi|d dkstW5QRXdS)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))rDzj(declare-const x Bool) (declare-const y Bool) (declare-const z Real) (assert (or (> z 3) (and x (not y))))rryhrjz(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;rrKrTr@rprr&r rrrrmrlrro)r9rryrr(r(r) test_boolean}sn &,               rc sttjtjg2ttttdkftddfdddks>tW5QRXttjtjgRtttdtdkftdtdkftdtdkftd dfddd kstW5QRXtttdkftdtdkftttd kfttjtjgt tfd d W5QRXdS)Nrzr<TFrJz(ite (< x 1) x (pow x 2))rDrCzQ(ite (< x 1) (pow x 2) (ite (< x 2) (pow x 3) (ite (< x 3) (pow x 4) (pow x 5))))rcs tdSrVrr(r~r9r(r)rXrYz'test_smtlib_piecewise..) r;rr&r'rrrKr@rrr(r(rr)test_smtlib_piecewises2    ,rc Cstttdkftddf}ttjtjg}td||ddksDtW5QRXttjtjg}t|t|ddksxtW5QRXttjtjtjg"}t|tt|ddkstW5QRXttjtjg}t|d|dd kstW5QRXdS) Nrzr<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)))rDz<(declare-const x Real) (* (/ 1 3) (ite (< x 1) x (pow x 2)))) rrKr;rr&r'rr@rT)pwr9r(r(r)!test_smtlib_piecewise_times_consts""&rc Cstg}tdd|ddks ttd|ddks4tttj|ddksJtttj|ddks`ttd|ddkstttdd|ddkstW5QRXdS) NTF)Z auto_assertr6truer>z (assert true)z(assert false)false)r;rr@r rrrAr(r(r)test_smtlib_booleans rc sftdttjtjgttfddW5QRXttjgttfddW5QRXdS)Nrcs&tttttgtfidS)Nrj)rrKdiffrror(rr9r(r)rX!rYz$test_not_supported..csttjdSrV)rr ZComplexInfinityr(rAr(r)rX#rY)r r;rr&r'rKeyErrorr(r(rr)test_not_supporteds rcCs4tddksttddks ttddks0tdS)Ngz0.0geK contextlibr0r#typingenumrrrnrrrZ sympy.corerrr r r r r rrrrrrrrZsympy.functionsrrrrZsympy.assumptions.askrZsympy.printing.smtlibrZsympy.testing.pytestrrrKrTrprcontextmanagerIterabler;rBrLrUrarfrqrrrrrrrrrrr(r(r(r)sB  <       @.XFX#