gWddlZddlZddlZddlZddlmZddlmZddlZddlmZm 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 eZ*ej+dej,e*fdZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8dZ9dZ:dZ;dZ>A666)CCCC8B??Q7773>>>>8Ar??Q777:EEEE8B++a888IEEEEE FFFFFFFFFFFFFFF r-r/ABQF G GN11x1~~-EANNNRaaaaa8Aq>>A-:::?NNNNNNNNNNNNNNNNNNNNNNNNs%BCCC3AEE!$E!cttjgdz5}tt t t d|dksJttt t d|dksJttt t d|dksJttt t d|dksJttt t d|dksJttt t d|d ksJ ddddS#1swxYwYdS) N FrR(assert (= x y))(assert (not (= x y)))(assert (<= x y))(assert (< x y))(assert (> x y))(assert (>= x y))) rAr!r(rrrTyrrrrrrHs r-test_Relationalr_Cs r-.3 4 4\2a88%!DDDHZZZZZ2a88%!DDDH`````2a88%!DDDH[[[[[2a88%!DDDHZZZZZ2a88%!DDDHZZZZZ2a88%!DDDH[[[[[[ \\\\\\\\\\\\\\\\\\sD!E  EEc ttjgdz5tt jt tddksJtt jt tddksJtt j t tddksJtt j t tddksJtt j t tddksJtt j t tdd ksJ dddn #1swxYwYttfd dS) NrWFrRrXrYr[rZr\r]cTttjtSNrE)rrcomplexrTrHsr-z,test_AppliedBinaryRelation..Vs{19Q< x 0))z(assert (< x 0))z(assert (= x 0))z(assert (<= x 0))z(assert (>= x 0))z(assert (not (= x 0)))) rAr!r(rrpositiverTnegativezero nonpositive nonnegativenonzerorHs r-test_AppliedPredicateruYs r-.2 3 3eq1:a==uqIIIM_____1:a==uqIIIM_____16!9951EEEI[[[[[1=++%!LLLPccccc1=++%!LLLPccccc19Q<?   ' '''' wwq!}}a !8S$K,=#>?$nb$?    2 22222'222222222222222: r-.2 3 32q wwq!}}a $nb$?   2 22222222222222222222222s^AA33A7:A7AC==DD1AE??FFD J//J36J3A L..L25L2cHttjtjg5}t t dzd|dksJ dddn #1swxYwYttjtjtjg5}t t t dzzd|dksJ dddn #1swxYwYttjtjg5}t t tddzd|dksJtdd }td d }td }d}d||dz|||zz zz |dz|zz }dddn #1swxYwYtg5}t t|dk|t||k||d zt|dtddzg|dksJ dddn #1swxYwYttjtjg5}t td|tt||dddd|dksJ ddddS#1swxYwYdS)NrLFrRz (pow x 3)z(pow x (pow y 3))rCz(pow x (/ 2 3))aTintegerbrealcc d|zS)NrCr+)rTs r-gztest_Pow..gs Q,r,g @rEz(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)))evaluaterGzG(declare-const b Real) (declare-const c Real) (* -2 c (pow (* b b) -1))) rAr!r(r*rrTr^rrrrr )r?rrrrexprs r-test_Powrs r-r/AB C CRq16BBBkQQQQQRRRRRRRRRRRRRRR r-r/ErGYZ [ [a_`1a=uqIIIM``````aaaaaaaaaaaaaaa r-r/AB C C ?q1A.UQOOOSddddd 3 % % % 3T " " " 3KKAAaDD3JAQJ//16A:> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? b  Q1q5! 1q5! D4Xa^^+,,       ( r-r/AB C C)q As3q!e444b5IIITY Z Z Z   ))))))))))))))))))))))))s\!AAA)CC  C 5BFF F !A"HHHA JJJcxttjtjtjg5}t t t zd|dksJ dddn #1swxYwYttjtjtjg5}t t t zd|dksJ dddn #1swxYwYttjtjg5}t t d|dksJ ddddS#1swxYwYdS)NFrRz(* x y)z(+ x y)z(* -1 x))rAr!r(r*rrTr^rHs r-test_basic_opsrs r-r/ErGYZ [ [O_`1q5uqAAAYNNNNNOOOOOOOOOOOOOOO r-r/ErGYZ [ [O_`1q5uqAAAYNNNNNOOOOOOOOOOOOOOO r-r/AB C CMqA2EA>>>*LLLLLMMMMMMMMMMMMMMMMMMs5&A$$A(+A(&CCCD//D36D3c ddlmddlmmm G fddt d}ttj g5}ttddft|t|t|ttgtfi| d ksJ dddn #1swxYwYttj gd z5}ttddftd d ft!tttt|t|t|ttgtfi| dksJ dddn #1swxYwYt#dd}t#dd}t#d}tg5}t|d df|d dft!||kt%||k|z|dksJ ddddS#1swxYwYdS)Nr)Boolean)IntervalTuplesympifycVeZdZfdZedZedZfdZdS)*test_quantifier_extensions..ForAllcfd|jD}dd||jgS)Nc g|]C\}}}|jjj|||gDSr+)_s_exprr7 _known_typesry).0symstartendrprinters r- zFtest_quantifier_extensions..ForAll._smtlib..sj)))'sE3()=c)BCHUC((+)))r,forall)limitsrfunction)selfrbound_symbol_declarationsrs ` r-_smtlibz2test_quantifier_extensions..ForAll._smtlibsl)))))+/+ ))) % ??8$=>> . r,c$d|jDS)Nch|]\}}}| Sr+r+)rs_s r- zKtest_quantifier_extensions..ForAll.bound_symbols..s111'!QA111r,)r)rs r- bound_symbolsz8test_quantifier_extensions..ForAll.bound_symbolss11T[111 1r,cVd|jDfd|jjDS)Nch|] }|j Sr+r7)rrs r-rzJtest_quantifier_extensions..ForAll.free_symbols..s!E!E!EQ!&!E!E!Er,c&h|] }|jv |Sr+r)rrbound_symbol_namess r-rzJtest_quantifier_extensions..ForAll.free_symbols..s16!333333r,)rr free_symbols)rrs @r-rz7test_quantifier_extensions..ForAll.free_symbolssJ!E!E$2D!E!E!E =5 r,cfd|D}fd|D}t|t|zt|ksJt|dksJ|d}t|rjg||jz|jRS|}t ||gz|_||_||_|S)NcTg|]$}t|tf|%Sr+) isinstancetuple)rrrrs r-rzFtest_quantifier_extensions..ForAll.__new__..s4PPPQ*Q2O2OPggajjPPPr,cFg|]}t||Sr+)r)rrrrs r-rzFtest_quantifier_extensions..ForAll.__new__..s0KKKqJq'4J4JK KKKr,rr)lenr__new__rrr_args) clsargsrrinstrForAllrrs r-rz2test_quantifier_extensions..ForAll.__new__sPPPPP$PPPFKKKKKDKKKHv;;X.#d));;;;x==A%%%%{H(F++ NFN5 8?255=5F555.??3''Dv 233DJ DK$DMKr,N)r"r#r$rpropertyrrr)rrrrrsr-rrs       2 2  2               r,rfir}z,(assert (forall ( (x Real [-42, 21])) true))rCirLzs(declare-fun f (Real) Real) (assert (forall ( (x Real [-42, 21]) (y Real [-100, 3])) (=> (= x y) (= (f x) (f y)))))rTrrrrdrEzx(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.boolalgrrrrrr rAr!r(rrTrrrr^rrr) rr?rrrrrrrrs @@@@@r-test_quantifier_extensionsrs++++++..........''''''''''''''V  A r-. / /<1 FAsC="QQqTT11Q44.. 1 1Xugun56   < <<<<<<<<<<<<<<<<<<<< r-.2 3 3 q FC 4|1a"QQqTT11Q44..11  Xugun56                         sD!!!AsAs A b   Q FAs VV3KAE477Q;//!3                              s9A/CCC7BF""F&)F&-AIIIc ttjg5}tdtz tdi|dksJ dddn #1swxYwYttjg5}tt tdddtz gtdi|dksJ dddn #1swxYwYttjg5}tt tjtj tj tj tj tj tjdtj d tjd tj d it d td id|dksJ dddn #1swxYwYttjg5}tt tjtj tj tj tj tj tjdtj d it d td id|dksJ dddn #1swxYwYttjg5}tt tjtj tj tj tj tj tjdt d id|dksJ dddn #1swxYwYttjg5}tt tjtj tj tj tj tj tjdtj d it d id|dksJ ddddS#1swxYwYdS)NrMY_PI)known_constantsr<z(pow MY_PI -1)gQ @Frz&(assert (= MY_PI 3.14)) (pow MY_PI -1)prr@plusrrL)rr{ precisionr<z!(plus 0 1 -1 (/ 1 2) (exp 1) p g)z$(plus 0 1 -1 (/ 1 2) (exp 1) p 1.62))r{rr<z$(plus 0 1 -1 (/ 1 2) 2.72 3.14 1.62)z!(plus 0 1 -1 (/ 1 2) e 3.14 1.62))rAr!r*rr rrr ZeroOne NegativeOneHalfExp1Pi GoldenRatiorrHs r-test_mix_number_mult_symbolsr7s r)* + +q FM     r)* + + q2te,,,B  M                    r)* + +1q q}afame = = =c1=# VU    1 1 1 1 1 1111111111111111 r)* + + 4q q}afame = = =cVU    4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 r)* + +4q q}afame = = = &M    4 44444444444444444444 r)* + +1q q}afame = = =VSM &M    111111111111111111111111so(AAA5?CCC&B(FF"FBII!$I!A z 3) (and x (not y))))rrhr}z(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)))rArrTr^rr!r(r rrrrrrr)r?rrrs r- test_booleanr}s) b   LQ1q51---2FFFFF1q51---2EEEEEA2***/AAAAA1q519q1116LLLLLL L L L L L L L L L L L L L L L r-. / /f1AFq1u-:::?fffffffffffffffffffff  A A A r-. / / $1 !a[[ 11Q44[[ 8TFCK(!Xtfck-B    $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ b   $Q !a[[ 11Q44[[ 8TFCK(!Xtfck-B    $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ b  ,Q !a[[ !!A$$^^ !!AAaDD'' A   8UGSL)8SE4K(8TFEM*     ,,,,,,,,,,,,,,,,,,,,,,,,s]BB//B36B32DDD%BHH HBJ;;J?J?C(O  OOc ttjtjg5t t t t dkft dzdfddksJ dddn #1swxYwYttjtjg5t t t dzt dkft dzt dkft dzt dkft d zdfdd ksJ dddn #1swxYwYt t t dkft dzt dkftt t d kfttjtjg5ttfd ddddS#1swxYwYdS) NrrCTFrRz(ite (< x 1) x (pow x 2))rLrKzQ(ite (< x 1) (pow x 2) (ite (< x 2) (pow x 3) (ite (< x 3) (pow x 4) (pow x 5))))rc&tSrbrrr?sr-rdz'test_smtlib_piecewise..s{4!'D'D'Dr,) rAr!r(r*rrrTrrAssertionErrorrs@@r-test_smtlib_piecewisers r-r/AB C C)q q!a%jAvtn & &    ) )))))))))))))))))))) r-r/AB C C q qAvq1uoAvq1uoAvq1uoAvtn & &                         aQZ!q&!a%3q661q5/ B BD r-r/AB C CFq~DDDDDEEEFFFFFFFFFFFFFFFFFFs7AA66A:=A:&A,DD#&D# GG G cVtttdkftdzdf}ttjtjg5}t d|z|dksJ dddn #1swxYwYttjtjg5}t |tz |dksJ dddn #1swxYwYttjtjtjg5}t |ttzz |dksJ dddn #1swxYwYttjtjg5}t |dz |d ksJ ddddS#1swxYwYdS) NrrCTrEz6(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)))rLz<(declare-const x Real) (* (/ 1 3) (ite (< x 1) x (pow x 2))))rrTrAr!r(r*rr^)pwr?s r-!test_smtlib_piecewise_times_constrs Aq1u:Q~ . .B r-r/AB C Clq1r6A...2kkkkkklllllllllllllll r-r/AB C Cuq26A...2ttttttuuuuuuuuuuuuuuu r-r/ErGYZ [ [^_`2Q.!s=QQqTTYYq\\HV[U\^cUcLdHepq!r!r!rr,c:ttjSrb)rr ComplexInfinityrHsr-rdz$test_not_supported..#sQ->!K!K!Kr,)r rAr!r(r*rKeyErrorrs@@r-test_not_supportedrsI A r-r/AB C Ctqxrrrrrsssttttttttttttttt r)* + +MqxKKKKLLLMMMMMMMMMMMMMMMMMMs#AA!$A!B((B,/B,ctddksJtddksJtddksJdS)Ngz0.0geK contextlibr5r%typingenumrrrrrr sympy.corerr r r r r rrrrrrrrrsympy.functionsrrrrsympy.assumptions.askrsympy.printing.smtlibrsympy.testing.pytestrrrTr^rr!contextmanagerIterablerArIrUr_rlrurrrrrrrrrrrr+r,r-rs%  $$$$$$$$$$888888888888888888888888888888444444444444######------//////// ''  1a{{{{{{{{  /6?2.////"000 N N N\\\ F F Feee=2=2=2@+)+)+)\ M M MUUUpC1C1C1LI,I,I,pFFF8 r r rFLLLMMM%%%%%r,