K iddlmZddlmZddlZedr ddlmcmZndZ ddl Z ddl m Z m Z mZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZ y)) import_module)raisesNllvmliteT)abncJd}d}t||z ||t|zzkS)Ngh㈵>g:0yE>)abs)rrrtolatols g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/printing/tests/test_llvmjit.pyisclosers, D D qs8td3q6k) ))ctdz}tjtg|}t|j tdij }|d}t ||sJy)Ng?@)rg llvm_callablefloatsubsevalfrefresjit_ress r test_simple_exprrsV CA QA 3x &&( )CfG 7C  rcdtztzdz}tjttg|}t |j tdtdij }|dd}t||sJy)Nrg@rrrrrrrrrs r test_two_argrsg A CA A"A 33'(..0 1CSkG 7C  rcdtjt z}tjtg|}t |j tdij}|d}t||sJy)Nr?) sympyexprrrrrrrrs r test_funcr$(sa EIIqbMA QA 3x &&( )CfG 7C  rcHdtjt ztjtz}t j ttg|}t |jtdtdij}|dd}t||sJyNrr!@) r"r#rrrrrrrrrs r test_two_funcr(1sw EIIqbMEIIaL(A A"A 33'(..0 1CSkG 7C  rcFdtjtztjtz}t j ttg|}t |jtdtdij}|dd}t||sJyr&) r"sqrtrrrrrrrrrs r test_two_sqrtr+:su EJJqMEJJqM)A A"A 33'(..0 1CSkG 7C  rctdztdzz}tjttg|}t |j tdtdij }|dd}t||sJy)Nr!r'rrs r test_two_powr.Csg 3A A A"A 33'(..0 1CSkG 7C  rcLtdz}tjtg|d}tjd}tj dz}tdi}||t}|||}t |j|j}t||sJy)N333333?scipy.integrate.test callback_type皙@ rrrctypesc_intc_doublerrrrrrm array_typeinparrayrrs r test_callbackr?Ls CA Q.DEA QA1$J c(C s1v E5kG s !!# $C 7C  rcvtdz}tjtg|d}tjd}tj dz}tdi}||t}|d}|||d||}|dk(sJt |j|j}t|d|sJy)Nr0cubaturer2r4r5grr6) rrr;r<r=r> out_arrayjit_retrs r test_callback_cubaturerDZs CA Qj9A QA1$J c(C s1v E3I5$9-G a<< s !!# $C 9Q< %% %rcdtztz}tjttg|d}t j d}tj dz}tdtdi}||t|t}|||}t|j|j}t||sJy)Nr1r2皙?333333?) rrrrr7r8r9rrrrr:s r test_callback_tworJks !AA A1GHA QA1$J c1c C s1vs1v &E5kG s !!# $C 7C  rctjd}d|dz|dz}tjt|g|d}t j d}t jdz}|dd|dd i}|||d||d}|||}t|j|j}t||sJy) NdrFrr4r1r2rGrHrI) r" IndexedBaserrrr7r8r9rrrr) rLrrr;r<r=r>rrs r test_callback_alt_tworNys #A !A$qt A A1GHA QA1$J Q4adC C s1Q4y#ad) ,E5kG s !!# $C 7C  rc tdtzfgtdzgg}tjtg|}|dddj tdi}t |ddj t|ij }|d}t||sJtjtg|d}tjd}tjdz}|d}|||} t| |sJy)Nrrr4r!r1r2) rrrrrrrrr7r8r9) rrb_valrr f_callbackr;r<r>jit_callback_ress r test_multiple_statementsrTs c!e*AwA QA aDGAJOOQH %E !Q aZ(..0 1CfG 7C  !a7MNJ QA1$J sOE!!U+ #S )) )rcttzttzztjt tzttzz z}tj|}t j ttg|}t|jtdtdij}|dd}t||sJy)Ngffffff@皙?) rrr"r#cserrrrrr)re2rrrs r test_cserYs !ac EIIqbdQqSj))A 1B A#A 33'(..0 1CSkG 7C  rci}|dD],\}}|j|}|j|}|||<.|dDcgc]"}|j|j|$c}Scc}w)Nrr4)r)rsub_dicttmp_dicttmp_nametmp_exprrXe3s r eval_cser`ssHd ( ]]8 $ WWX  67qT :AFF8  ! !( + :: :s'A(cttz}ttzttzz}tj||gt t fdt jttg}|dd}t|dk(sJttdtdi}t|d|dsJt|d|dsJy)NcHtjttgdS)Nzscipy.integrater2)rrrr)r_sr z#test_cse_multiple..s1??Aq62=NOrrVr!rGrr4) rrr"rWrNotImplementedErrorrrlenr`r)e1rXrrrr_s @r test_cse_multiplergs 1B 1qsB B8 B  OQ A#ASkG w<1   233' (C 3q671: && & 3q671: && &rcttz}ttzttzz}tj||d|zg}t j ttg|d}d}d}t j|}t j|}t j|z}t j|z} tdtdi} || t| t} | } ||| d|| } | dk(sJt|| }t| d|dsJt| d |d sJt| d|dsJy) NrAr2rGrFrHr!rr4) rrr"rWrrr7r8r9r`r)rfrXr_rndimoutdimr;fdimr<out_array_typer=r>rBrCrs r test_callback_cubature_multiplerns* 1B 1qsB BAbD> "B A*=A D F TA << D4'J__v-N c1c C s1vs1v &E I5$i0G a<< 2s C 9Q<Q (( ( 9Q<Q (( ( 9Q<Q (( (rcTttztzttfdy)Nc:tjtgS)Nrrrrsr rcz'test_symbol_not_found..sQ 7r)rrr LookupErrorrrs@r test_symbol_not_foundrts !aA ;78rc8tttfdy)Nc>tjtgdS)N bad_callbackr2rqrrsr rcz#test_bad_callback..sqsA^Tr)rr ValueErrorrrs@r test_bad_callbackrys A :TUr)!sympy.externalrsympy.testing.pytestrr7sympy.printing.llvmjitcodeprinting llvmjitcoderdisabledr" sympy.abcrrrrrrr$r(r+r.r?rDrJrNrTrYr`rgrnrtryrr rs(' **H * !!!!!! !&" ! !*"!; ' ):9 Vr