K i3(ddlmZddlmZedZer&ej ej ejcZZZndZddl m Z m Z m Z ddl mZmZddlmZddlmZmZmZmZdd lmZdd lZdd lZdd lmZmZmZdd lm Z dd l!m"Z"ddl#m$Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4e dZ5d Z6d!Z7d"Z8d#Z9d$Z:d%Z;y )&) version_tuple) import_modulenumpyT)FloatIntegerRational)Symbolsymbols)sin)Matrix list2numpy matrix2numpysymarray)lambdifyN)xyz)conserve_mpmath_dps)ignore_warnings)raisesc d}td}td}tddtd||t|||zt dtdg}t d gt gd t ||dztdgt ||z t |zdtdgg}|D]}|D] }||| y) Nc X||z||z||z ||z ||z||z||z ||z ||z||zg }y)N) sympy_object numpy_array_s e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/external/tests/test_numpy.pysz test_systematic_basic..s%sY K 'l"{"l"{"l"{"l" #|# %rrz1.3g@)r!)r rrpowrarrayr )rrr sympy_objs numpy_objss rtest_systematic_basicr*$s % s As AA e    Aq !   c J qc  j q!Q$ $% qs3q6z1hqk*+ J  A aG rctd}td}tdt|k(sJt|gt|gk(sJttgttgk(sJttttdk(sJt|tztdtzk(sJt|||g}|t|||gk(j sJ|t|ddgk(j sJy)Nr#rr)rr'rr all)onezeroXs r test_basicsr0Is 1+C A;D 8uSz !! ! #<5#< '' ' !:s ## # 8uVC[) )) ) q>U1q5\ )) ) sD$ A T4() ) . . 00 0 Q{# # ( ( ** *rcZtd}td}t|||g}||z}ttdtddzg}||z}|tddtdzzgk(sJ|dz}|tddtdzzgk(sJ||z }|tdgk(sJy)Nr#rar )rr'r )r-r.r/Ys r test_arraysr4Ws 1+C A;D sD$ A AA vc{Xa^+,-A AA q1VC[=()* ** * AA q1VC[=()* ** * AA qc ??rcttdztg}t|tsJ|dtdzk(sJ|dtk(sJt |dk(sJy)Nr rr#)r r isinstancendarraylenr2s rtest_conversion1r:esXAqD!9A a !! ! Q41a4<< Q4199 q6Q;;rcdttdztgz}tdtdzzdtzg}||k(jsJtd}td}t|||g}||z}tt dtddzg}||z}|t ddt dzzgk(sJ|dz}|t ddt dzzgk(sJ||z }|t dgk(sJy)Nr r#rr2)r rr,rr r')r2br-r.r/r3s rtest_conversion2r=os *adAY AAadFAaC=!A F<<>> 1+C A;DCt$%A AAF3K(1a.012A AA q1VC[=()* ** * AA q1VC[=()* ** * AA qc ??rcttdztgttdztgk(jsJy)Nr )r'rr r,rrrtest_list2numpyr?s3 1a4)  AqD!9 5 5 : : << #? ? D D FF FrcLtttdzgddtz gg}tt5t |j tdt ddgddggk(j sJ dddttttdzgddtz gg}tt5t |j tdt tddgddggk(j sJ dddy#1swYxYw#1swYyxYwrA)r rrPendingDeprecationWarningmatrixrCr,r rDs r test_Matrix2rJsAqD Aqs8$%A 2 3Hqvva|$AA/?(@@EEGGGHQAAaC)*A 2 3Mqvva|$Q aV/D(EEJJLLLMMHHMMsAD9A DDD#cvtddgddgg}t|tddgddggk(sJt|tddgddggk7sJttddgddgg}t|ttddgddggk(sJt|ttddgddggk7sJyNr rBr"r#r)r'r r r9s r test_Matrix3rMs 1v1vA !9AA/0 00 0 !9AA/0 00 0 A{QF#$A !9Q aV45 55 5 !9Q aV45 55 5rctt5tddgddgg}dddttddgddggk(sJt|tddgddggk7sJtt5tt ddgddgg}dddt|tt ddgddggk(sJt|tt ddgddggk7sJy#1swYxYw#1swYfxYwrL)rrHrIr r r9s r test_Matrix4rOs 2 3% QFQF# $% !9AA/0 00 0 !9AA/0 00 0 2 3* SVQK!Q( )* !9Q aV45 55 5 !9Q aV45 55 5%%**sC+7C7+C47Dc tgdtttgdtzdttzgg}t t 5t gdtddgtttdzgg}ddd|ztgddtztdztdzgdtztztdz ttztdzzggk(sJ||ztgddtztdztdzgdtztztdz ttztdzzggk(sJ||z|j|k(sJy#1swYxYw) Nr#r r!r i)r r!rBr")r!r"2)r rrrrrHrIadd)MrEs rtest_Matrix_sumrWs( Aq!9qsC1o67A 2 39 I1ay1aA,7 89 q5FI!QUAE':QqS1Wa"faPQcTUWXTXj.matarrayNcN| |s tdddlm}|gdgdgdgS)Nz=Cannot implement copy=False when converting Matrix to ndarrayr)r'rQrBr"rRrSr$rZ) TypeErrorrr')selfdtypecopyr's r __array__z-test_Matrix_array..matarray.__array__s+ _`` #)Y :; ;r)__name__ __module__ __qualname__objectrfrrrmatarrayr^s "(t VY 9$EF FF Frc *ttdtdzgdttzdgg}t |t sJ|j dk(sJ|ddk(sJ|dtdzk(sJ|ddttzk(sJ|d dk(sJy) Nr#r r!r)r r rr)rr#)r#r)r#r#)rr rr r6r7shaper9s rtest_matrix2numpyrqsVaAY3q61 678A a !! ! 77f   T7a<< T7ad?? T7aAh   T7a<AVaVb"X./yAA 77ekk&) )) ) 77ekk), ,, ,rctddtdtzdgzttdgk(jsJtddtdtzdgztdtztddztddgk(jsJt dtdtzdgztt dtzdgk(jsJt dtdtzdgztdtzt dzt dgk(jsJy)Nr#r rz0.5z1.0)rr'rr,rrrrtest_issue_3728r|s1 QN5!A#q? *eQFm ; @ @ BB B QNU 1a AaC(1a.0(1a.AB CDGCE JJ J %L!Qx (E5<>12E,F F K K MM M %L5 1a AaC%,.e => ?@C FF Frcdtj_tjd}t t t t d}d}| |d|z cxkr|ksJJttjtdk\r%tt5|t dddytt5|t dddy#1swYyxYw#1swYyxYw)NrYz0.198669330795061215459412627rgV瞯 (D (( (( (U&&'=+@@ I   aD  N #  aD      s C> C C C)cttttdtzgddggdtjidg}|dt ddgddggk(j sJy)Nr r#ImmutableMatrixr)rrr rr'r,)rs rtest_lambdify_matrixrscFQ!Hq!f-.2CU[[1QSZ0[\A aDEAq6Aq6*+ + 0 0 22 2rctjtdzttzttzgttztdzttzgttzttztdzgg}t tttf|dt jidg}d\}}}t|dz||z||zg||z|dz||zg||z||z|dzgg}||||}t j||sJy)Nr rrg?g@g@) sympyr rrrrrr'allclose)rVrxhyhzhexpectedactuals r test_lambdify_matrix_multi_inputrs q!tQqS!A#&sAqD!A#&sAaCA&( )A !QA!2EKK @'JKAJBBr1ubeRU+"ub!eRU+"ubeRU+-.Hr2r]F >>&( ++ +rcBtjd}t|ddz|d|dz|d|dzg|d|dz|ddz|d|dzg|d|dz|d|dz|ddzgg}t||dtj idg}t gd}t |ddz|d|dz|d|dzg|d|dz|ddz|d|dzg|d|dz|d|dz|ddzgg}||}tj ||sJy)Nr/rr r#rrr)rDeferredVectorr rrr'r)r/rVrXhrrs rtest_lambdify_matrix_vec_inputrss S!A 1q!A$qt)QqT!A$Y' 1adAaD!GQqT!A$Y' 1adAaD1IqtQw') *A A*EKK8'BCA  Br!uaxAr!ubeBqEk:!uRU{BqE1HbeBqEk:!uRU{BqE"Q%KA:<=HrUF >>&( ++ +rcddlm}|jD]-\}}|tjvsJ|t jvr-Jy)Nr)NUMPY_TRANSLATIONS)sympy.utilities.lambdifyritemsr__dict__r)rsymmats rtest_lambdify_translr$sE;&,,.%Senn$$$enn$$$%rcddl}ddlm}td}t dd}t dd}|j ||j |t|d|dk(sJt dd}t dd}|d|dk(rJtd }|j ||j |tt dd }|jd k(sJtd \} } |d | k(sJ|d | k(sJt dd} | jdk(sJtd\} } }| d| k(sJ| d| k(sJ| d|k(sJy)z/Test creation of numpy arrays of SymPy symbols.rNz_0,_1,_2r!rvr2r<z a_0,a_1,a_2)r r!z a_0_0,a_1_2ro)r#r )r r!r za_0_0_0,a_1_2_0,a_1_2_1)rrr)r#r r)r#r r#) r numpy.testingtestingr rassert_array_equalr'rjrp)npnptsymss1s2r2r<asymsa2da00a12a3da000a120a121s r test_symarrayr+sg : D "aB "aB2rxxFx;< a5BqE>>aAaAtqt|  M "E1bhhuFh;< 3 C 99  }%HC t9   t9   3 "C 99 !! !89D$ w<4   w<4   w<4  rctjtgdtjtdtdtdgk(j sJy)NrQr#r r!)r vectorizer r'r,rrrtest_vectorizerMsW OO    ;;AAA'?@ ABE#% HH Hr)<sympy.external.importtoolsrsympy.externalrrr'rIr7disabledsympy.core.numbersrrrsympy.core.symbolr r (sympy.functions.elementary.trigonometricr sympy.matrices.denser r rrrrrr~ sympy.abcrrrsympy.utilities.decoratorrsympy.utilities.exceptionsrsympy.testing.pytestrr*r0r4r:r=r?rFrJrMrOrWr[rmrqrzr|rrrrrrrrrrrs5(g"[[%,, E67H:9/8MM- 96'J + $=GM66&G -F"3 , , % DHr