o o 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+dej,e*fddZ-ddZ.ddZ/ddZ0ddZ1ddZ2ddZ3ddZ4dd Z5d!d"Z6d#d$Z7d%d&Z8d'd(Z9d)d*Z:d+d,Z;d-d.ZF auto_declarer8z (+ (/ 3 7) x)z$(declare-const x Real) (* (/ 3 7) x))r=rr(rr r&xrCr)r)r* test_Rational6s  "rOcCsttjgd]}ttttd|ddksJttttd|ddks%Jttttd|ddks3Jtt ttd|ddksAJtt ttd|ddksOJtt ttd|dd ks]JWddS1shwYdS) N FrL(assert (= x y))(assert (not (= x y)))(assert (<= x y))(assert (< x y))(assert (> x y))(assert (>= x y))) r=rr&rrrNyrrrrrrCr)r)r*test_RelationalC"rXcsttjgdbttttdddksJttttdddks'Jtt ttdddks6Jtt ttdddksEJtt ttdddksTJtt ttddd kscJWdn1smwYt tfd d dS) NrPFrLrQrRrTrSrUrVcstttdSNr@)rrcomplexrNr)rCr)r*Vz,test_AppliedBinaryRelation..)r=rr&rreqrNrWneltlegtger ValueErrorr)r)rCr*test_AppliedBinaryRelationMs recCsttjgd]}tttd|ddksJtttd|ddks%Jtttd|ddks3Jtt td|ddksAJtt td|ddksOJtt td|dd ks]JWddS1shwYdS) NFrLz(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&rrpositiverNnegativezero nonpositive nonnegativenonzerorCr)r)r*test_AppliedPredicateYrYrmcCs,ttjtjg}tttttd|ddksJWdn1s$wYttjg }ttttt t t it dit j di|ddksHJWdn1sRwYtd}ttjg}t|t|tt gtfi|d d ksvJWdn1swYtgS}t|t|tt gt fi|d d ksJtt|ttt |tt t gt fi|d d ksJtt|ttt |tt t gt fi|d tdi|ddksJWdn1swYttjgd}tt|ttt |d tdi|ddksJWddS1swYdS)NFrLz(pow (sin x) (cos x)) INTEGER_TYPEABSOLUTE_VALUE_OF) symbol_table known_typesknown_functionsr8z4(declare-const x INTEGER_TYPE) (ABSOLUTE_VALUE_OF x)f1rpr8z:(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)) MY_KNOWN_FUNz==)rprrr8zf(declare-const x Int) (declare-const y Bool) (declare-const z Bool) (assert (== (MY_KNOWN_FUN x z) y))rF)rrr8zg(declare-const x Real) (declare-const y Real) (declare-const z Real) (assert (== (MY_KNOWN_FUN x z) y)))r=rr&r(rrrNrabsintrWboolsympyAbsr rfloatrz)r;my_fun1r)r)r* test_Functionbsr$          $r~c Csttjtjg}ttdd|ddksJWdn1s wYttjtjtjg}tttdd|ddks?JWdn1sIwYttjtjg@}tttddd|ddksgJtdd d }td d d }td }dd}d||d||||d|}Wdn1swYtg*}tt |dk|t ||k||d @t |dtddg|ddksJWdn1swYttjtjg#}tt d|t t ||ddddddd|ddksJWddS1swYdS)NrFFrLz (pow x 3)z(pow x (pow y 3))r>z(pow x (/ 2 3))aTintegerbrealccSsd|S)Nr>r))rNr)r)r*gsztest_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)))evaluaterBzG(declare-const b Real) (declare-const c Real) (* -2 c (pow (* b b) -1))) r=rr&r(rrNrWr rrrr)r;rrrrexprr)r)r*test_PowsD    *    " $rcCsttjtjtjg}tttd|ddksJWdn1s"wYttjtjtjg}tttd|ddks?JWdn1sIwYttjtjg}tt d|ddkscJWddS1snwYdS)NFrLz(* x y)z(+ x y)z(* -1 x))r=rr&r(rrNrWrCr)r)r*test_basic_opss"rc sddlmddlmmmGfdddtd}ttj g'}t t ddft |t |t |t tgtfi|d d ksHJWdn1sRwYttj gd 1}t t ddftd d ftt t tt |t |t|t tgtfi|d dksJWdn1swYtddd}tddd}td}tg*}t |d df|d dft||kt||k|B|ddksJWddS1swYdS)Nr)Boolean)IntervalTuplesympifycsBeZdZfddZeddZeddZfddZd S) z*test_quantifier_extensions..ForAllcs0fdd|jD}dd||jgS)Nc s6g|]\}}}|jjj|||gqSr))_s_exprr3 _known_typesrp).0symstartend)rprinterr)r* szFtest_quantifier_extensions..ForAll._smtlib..forall)limitsrfunction)selfrbound_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)r3rrr)r)r*rr]zJtest_quantifier_extensions..ForAll.free_symbols..csh|] }|jvr|qSr)rrbound_symbol_namesr)r*rs  )rr free_symbolsrr)rr*rs z7test_quantifier_extensions..ForAll.free_symbolscsfdd|D}fdd|D}t|t|t|ks"Jt|dks*J|d}t|rBjg||j|jRS|}t||g|_||_||_|S)Ncs"g|] }t|tfr|qSr)) isinstancetuplerr)rrr)r*rs"zFtest_quantifier_extensions..ForAll.__new__..csg|] }t|r|qSr))rr)rrr)r*rsrr)lenr__new__rrr_args)clsargsrrinst)rForAllrrr)r*rs" z2test_quantifier_extensions..ForAll.__new__N)r r!r"rpropertyrrrr)rrrrrr)r*rs   rfirtz,(assert (forall ( (x Real [-42, 21])) true))r>irFzs(declare-fun f (Real) Real) (assert (forall ( (x Real [-42, 21]) (y Real [-100, 3])) (=> (= x y) (= (f x) (f y)))))rTrrrrdr@zx(declare-const c Bool) (assert (forall ( (a Int [2, 100]) (b Real [2, 100])) (or c (=> (< a b) (< (pow a (/ 1 2)) b))))))sympy.logic.boolalgrryrrrr r=rr&rrNrrr{rWrrr)rr;rrrr)rr*test_quantifier_extensionssR +       "rc Csttjg}tdttdi|ddksJWdn1s wYttjg}tttddddtgtdi|ddksBJWdn1sLwYttjg3}tttjtj tj tj tj tj tjddtj d tjd tj d itd td id|ddksJWdn1swYttjg-}tttjtj tj tj tj tj tjddtj d itd td id|ddksJWdn1swYttjg'}tttjtj tj tj tj tj tjddtd id|ddksJWdn1swYttjg-}tttjtj tj tj tj tj tjddtj d itd id|ddks-JWddS1s9wYdS)NrMY_PI)known_constantsr8z(pow MY_PI -1)gQ @Frz&(assert (= MY_PI 3.14)) (pow MY_PI -1)prr<plusrrF)rrr precisionr8z!(plus 0 1 -1 (/ 1 2) (exp 1) p g)z$(plus 0 1 -1 (/ 1 2) (exp 1) p 1.62))rrrr8z$(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 ZeroOne NegativeOneHalfExp1Pi GoldenRatiorrCr)r)r*test_mix_number_mult_symbols7s          $rc Csztg9}ttt@|ddksJtttB|ddksJtt|ddks(Jttt@t@|ddks6JWdn1s@wYttjg}ttt@tdkB|ddks]JWdn1sgwYtd}td }td }ttjg-}tt|ttt t|tg|t t gt f|t t gt fi|d d ksJWdn1swYtg-}tt |ttt t|tg|t t gt f|t t gt fi|d d ksJWdn1swYtgB}tt |ttt ||ttt |||ttg|t tgt f|t t gt f|t t gtfi|d dks*JWddS1s6wYdS)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))rFzj(declare-const x Bool) (declare-const y Bool) (declare-const z Real) (assert (or (> z 3) (and x (not y))))rrhrtz(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=rrNrWr|rr&r rrrrxrwrr{)r;rrrr)r)r* test_boolean}sf $          $rcsNttjtjgttttdkftddfdddksJWdn1s)wYttjtjg.tttdtdkftdtdkftdtdkftd dfddd ks]JWdn1sgwYtttdkftdtdkftttd kfttjtjgtt fd d WddS1swYdS)Nrr>TFrLz(ite (< x 1) x (pow x 2))rFrEzQ(ite (< x 1) (pow x 2) (ite (< x 2) (pow x 3) (ite (< x 3) (pow x 4) (pow x 5))))rcs tdSrZrr)rr;r)r*r\s z'test_smtlib_piecewise..) r=rr&r(rrrNrrAssertionErrorr)r)rr*test_smtlib_piecewises8     ,"rcCsHtttdkftddf}ttjtjg}td||ddks"JWdn1s,wYttjtjg}t|t|ddksFJWdn1sPwYttjtjtjg}t|tt|ddksnJWdn1sxwYttjtjg}t|d|dd ksJWddS1swYdS) Nrr>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)))rFz<(declare-const x Real) (* (/ 1 3) (ite (< x 1) x (pow x 2))))rrNr=rr&r(rrW)pwr;r)r)r*!test_smtlib_piecewise_times_consts"rcCstgI}tdd|ddksJtd|ddksJttj|ddks%Jttj|ddks0Jtd|ddks:Jtdd|ddksEJWddS1sPwYdS) NTF) auto_assertr8truer@z (assert true)z(assert false)false)r=rr rrrCr)r)r*test_smtlib_booleans "rcstdttjtjgttfddWdn1s!wYttjgttfddWddS1sAwYdS)Nrcs&tttttgtfidS)Nrt)rrNdiffrr{r)rr;r)r*r\!s&z$test_not_supported..csttjdSrZ)rr ComplexInfinityr)rCr)r*r\#s)r r=rr&r(rKeyErrorr)r)rr*test_not_supporteds"rcCs4tddksJtddksJtddksJdS)Ngz0.0geK contextlibr1r#typingenumrrryrrr sympy.corerrr r r r r rrrrrrrrsympy.functionsrrrrsympy.assumptions.askrsympy.printing.smtlibrsympy.testing.pytestrrrNrWr|rcontextmanagerIterabler=rDrOrXrermr~rrrrrrrrrrr)r)r)r*sD  <       @.XFX#