K ixHdZddlZddlmZddlZddlZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZmZeee ee e fZedeDZej2eed Zej6j6Zd+d ZeZeZd Zd Z d Z!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3d Z4d!Z5d"Z6d#Z7d$Z8d%Z9d&Z:d'Z;d(Z<Gd)d*Z=y),zTest inter-conversion of different polynomial classes. This tests the convert and cast methods of all the polynomial classes. N)Number) RankWarning) ChebyshevHermiteHermiteELaguerreLegendre Polynomial)assert_assert_almost_equal assert_equal assert_raisesc#4K|]}|jywN)__name__).0clss i/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/polynomial/tests/test_classes.py r$s1#1s)paramsidsc|jSr)param)requests rPolyr&s ==cZ ttj|j|jk(ttj|j|jk(t |j |j y#t$rd|d|}t|wxYw)NzResult: z Target: )r npalldomainwindowr coefAssertionError)p1p2msgs rassert_poly_almost_equalr'1s"ryyBII-./ryyBII-./BGGRWW- "Jrd+S!!"s B B B*ctjddd}td}|jtddzz}|jtddzz}||||}|jtddzz}|jtddzz}|j |||} t | j|t | j|t | |||y) Nr ?r r!)kindr r!)rlinspacerandomr r!convertr Poly1Poly2xr"d1w1r$d2w2r%s rtest_conversionr=Cs Aq"A $.fAE{a!e$$rrr,r-r/r0)rr)r.r,r)r.)rE) rr2r[r r r rPr3r! zeros_likeshape) rrer8r^rFrDrEzr$r%p3s rtest_fitrks0% AqA !A AqA1a&)!a Q fTlS((A fTlS((A Aq1-A!a !$!$ A|Aa8A!a !$!$ Aq"A4;;'4;;' A|R(A4;;'4;;' aA F177Oc !!A AccF !CaC&!CaC&! $B !QQ B !Q  *B1r!u%1r!u%rc|gdddgddg}|gdddgddg}|gdddgddg}|gdddgddg}t||k(t||k( t||k( t||k( yNr)r.r,rr)r.r,r0)r)r)r)r rr$r%rjp4s r test_equalrrs iA1v 6B iA1v 6B iA1v 6B iA1v 6B B"H bL bL bLrc|gdddgddg}|gdddgddg}|gdddgddg}|gdddgddg}t||k7 t||k7t||k7t||k7yrmrorps rtest_not_equalrts iA1v 6B iA1v 6B iA1v 6B iA1v 6B bL B"H B"H B"Hrc ^ttddz}ttddz}||}||}||z}t||z|t||z|t||z|t|t|z|tt||z|t|t j |z|tt j ||z|t ttj||dg|jdzt ttj||dg|jdz|tur+t ttj|tdgyt ttj|tdgyN?r+rr)r r!)listr3r'tuplerarrayr TypeErroropaddr r!r rrc1c2r$r%rjs rtest_addrs= fTlR B fTlR B bB bB bBR"Wb)R"Wb)R"Wb)R%)^R0U2Y^R0R"((2,.3RXXb\B.3)RVVRqc$++/)JK)RVVRqc$++/)JK ziYs^<iZ_=rc fttddz}ttddz}||}||}||z }t||z | t||z |t||z | t|t|z |tt||z | t|t j |z |tt j ||z | t ttj||dg|jdzt ttj||dg|jdz|tur+t ttj|tdgyt ttj|tdgyrv)r|r3r'r}rr~rrrsubr r!r rrs rtest_subrsE fTlR B fTlR B bB bB bBR"Wrc*R"Wb)R"Wrc*R%)^R0U2Y^bS1R"((2,.3RXXb\B.4)RVVRqc$++/)JK)RVVRqc$++/)JK ziYs^<iZ_=rc ttddz}ttddz}||}||}||z}t||z|t||z|t||z|t|t|z|tt||z|t|t j |z|tt j ||z|t|dz||dgztd|z||dgzt ttj||dg|jdzt ttj||dg|jdz|tur+t ttj|tdgyt ttj|tdgy) Nrwryr+r.rr)rzr{)r|r3r'r}rr~rrrmulr r!r rrs rtest_mulr sm fTlR B fTlR B bB bB bBR"Wb)R"Wb)R"Wb)R%)^R0U2Y^R0R"((2,.3RXXb\B.3R!VR$s)^4QVR$s)^4)RVVRqc$++/)JK)RVVRqc$++/)JK ziYs^<iZ_=rc ttddz}ttddz}ttddz}||}||}||}||z|z}t|j}t||z|t||z|t||z|t|t |z|tt ||z|t|t j |z|tt j ||z|td|z|dgt|dzd|ztttj||dg|jdztttj||dg|jdz |tur+tttj|tdgytttj|tdgy Nrwryr+r-r.rr)rzr{)r|r3r"r'r}rr~rrrfloordivr r!r r rrrc3r$r%rjrqc4s r test_floordivr s fTlR B fTlR B fTlR B bB bB bB b2B bggBR2Xr*R2Xr*R2Xr*R59_b1U2Y"_b1R288B</4RXXb\R/4Q"WdA3i0R1WcBh/2;;D!T[[1_$EG2;;D!T[[1_$EG zib)QC.Aib*aS/Brc |gd}|dz}tjD]k}t|trt|tr$|d}t t j|||ttt j||mttfD]J}|d}t t j|||ttt j||LtfD]K}|dd}t t j|||ttt j||MdgidtjdgfD]B}ttt j||ttt j||DtD](}ttt j||d*y)NrnrIrrdFr))r ScalarType issubclassrboolr'rtruedivrrintfloatcomplexr~classes)rr$r%stypesptypes r test_truedivr<sj iB aB4%(Jud,C  !H B!2B7iQ3 4 u4 !H B!2B7iQ344 !QK B!2B7iQ34"b%1# /4iR3iQ34;iRq:;rc $ttddz}ttddz}ttddz}||}||}||}||z|z}t|j}t||z|t||z|t||z|t|t |z|tt ||z|t|t j |z|tt j ||z|td|z|dgt|dz|dgtttj||dg|jdztttj||dg|jdz |tur+tttj|tdgytttj|tdgyr)r|r3r"r'r}rr~rrrmodr r!r rrs rtest_modrWs fTlR B fTlR B fTlR B bB bB bB b2B bggBR"Wb)R"Wb)R"Wb)R%)^R0U2Y^R0R"((2,.3RXXb\B.3QVT1#Y/R!VT1#Y/)RVVRqc$++/)JK)RVVRqc$++/)JK ziYs^<iZ_=rc ttddz}ttddz}ttddz}||}||}||}||z|z}t|j}t||\} } t | |t | |t||\} } t | |t | |t||\} } t | |t | |t|t |\} } t | |t | |tt ||\} } t | |t | |t|t j|\} } t | |t | |tt j||\} } t | |t | |t|d\} } t | d|zt | |dgtd|\} } t | |dgt | |dgttt||dg|jdzttt||dg|jdz |tur!ttt|tdgyttt|tdgyr)r|r3r"divmodr'r}rr~rrr r!r r) rrrrr$r%rjrqrquorems r test_divmodrrsK fTlR B fTlR B fTlR B bB bB bB b2B bggBb"~HCS"%S"%b"~HCS"%S"%b"~HCS"%S"%b%)$HCS"%S"%eBi$HCS"%S"%b"((2,'HCS"%S"%bhhrlB'HCS"%S"%b!}HCS#(+S$s),a}HCS$s),S$s),)VRqc$++/)JK)VRqc$++/)JK ziYs^<iZ_=rcx|jdzdz}|j}tj|d|dd}tj|j |||j }t||tj|j |j }t||y)Ng?r/rr)rIr0)r r!rr2sortrOrootsr )rrDrEtgtress r test_rootsrs dS A A ++adAaD! $C ''$..Qq.9??A BCS! ''$..%++- .CS!rcZ|jd}t|jdyNrI)rJr rPrrFs r test_degreers 1 AQrcB|jd}|j}t||k(t||ut|j|jut|j|jut|j |j uyr)rJcopyr r"r r!)rr$r%s r test_copyrsq AB B B"H BbL BGG277 "# BIIRYY &' BIIRYY &'rcFt}|j|gd}|j|j}|j|jd}t||gdt||gd|j|gd}|j|jd}|j|jdddg}t||gdt||gd|j|gd}|j|jd }|j|jdd }t||gd t||gd d|jz}|j|gd| }|j|j}|j|jd}t||gdt||gdy) N)r. r.)rr.r,rx)rrr)r)r)r)k)r)r.r,rx)r)r)r)r)r))lbnd)r.r,rx)rrr)r)r)rz)r r?integr'r )rPp0r$r%rDs r test_integrsA 1*+ ,B  B  BR<1R?!34 1*+ ,B 1 B q!f% &BR<1R?!34 1*+ ,B a !B # $BR=!12R#3!45 DKKA 1*+A 6B  B  BR<1R?!34rc|jtddzz}|jtddzz}|gd||}|jdddg}|jddg}t |j dj |j t |j dj |j |gd}|jdddg}|jddg}t |j dj |j t |j dj |j y)Nr-r/rnr0r.r)r)r r3r!rr derivr")rrDrEr$r%rjs r test_derivrs  fTlS((A fTlS((A i! ,B !1v B !s B (("''2 (("''2 iB !1v B !s B (("''2 (("''2rc|jtddzz}|jtddzz}|gd||}tj|d|dd}||}|j d\}}t ||t ||tjddd}||}|j dddg \}}t ||t ||y) Nr-r/rnr0rr)r.rz)r r3r!rr2r )rrDrErFxtgtytgtxresyress r test_linspacers fTlS((A fTlS((A Yq+A ;;qtQqT2 &D T7DBJD$d#d# ;;q!R D T7DB1v.JD$d#d#rc|jtddzz}|jtddzz}|dg||}|gd||}tdD]}t ||z|||z}|dg}|gd}tdD]}t ||z|||z}t t tj|dt t tj|dy) Nr-r/r)r0rnrIg?rN) r r3r!ranger'r ValueErrorrpow)rrDrErtstis rtest_powrs fTlS((A fTlS((A s1Q 'C y1 -C 1X a-Ci s)C y/C 1X a-Ci*bffc3/*bffc2.rct}|j}tj|d|dd}|j |gd}d|dd|zzzz}||}t ||y)Nrr)rBrnr.r,)r r rr2r?r )rrrDr8rFrrs r test_callr sgA A AaD!A$#A !I,A a1q1u9o C A$CS!rcz|gd}gd}||}t||tj|y)Nrn)rNrr.)r rr~)rrFr8rs rtest_call_with_listrs/ YAA A$Ca n%rc|gd}tt|jdtt|jdtt |jddtt |jddtt |jddtt |jddy)NrnryrNr,r.r)r)rrcutdegr rQrs r test_cutdegr s YA*ahh+*ahh+QXXa[!1%QXXa[!1%QXXa[!1%QXXa[!1%rc|gd}tt|jdtt|jdtt |jddtt |jddtt |jddtt |jddy)Nrnryrrxr,r.r))rrtruncater rQrs r test_truncater*s YA*ajj"-*ajj!,QZZ]#Q'QZZ]#Q'QZZ]#Q'QZZ]#Q'rc gd}||}t|jj|ddt|jdj|ddt|jdj|ddy)N)r)gư>g-q=rr,g|=r.gh㈵>r))r trimr")rcrFs r test_trimr4sfA QA"1&##QrU+""AbqE*rc|j}|j}|dg||}tddg|jd|zdz}|dg||}tddg|jy)Nr)r0rr.)r r!r mapparmsrKs r test_mapparmsr<si A A aS1%AA - A A aS1%AA -rc|gd}tjd}tttj||tttj||y)Nrnr,)ronesrrr)rrFr8s rtest_ufunc_overriderHs< YA  A)RVVQ*)RVVQ*rc$eZdZdZdZdZdZy)TestInterpolatec||dz z|dz zSrcrd)selfr8s rrezTestInterpolate.fVrfrctttj|jdtt tj|jdy)NrNg$@)rrr interpolaterer)rs r test_raiseszTestInterpolate.test_raisesYs2j)"7"7Di!6!6DrctddD]<}ttj|j|j |k(>y)Nr)rI)rr rrrerP)rdegs rtest_dimensionszTestInterpolate.test_dimensions]s@A; HC I))$&&#6==?3F G Hrc d}tjddd}tdD]L}t|dzD]9}tj||ddg|f}t |||||d;Ny) Nc ||zSrrd)r8rFs rpowxz0TestInterpolate.test_approximation..powxcs a4Krrr.r*r))r r]rB)decimal)rr2rrrr )rrr8rtrFs rtest_approximationz"TestInterpolate.test_approximationas{  KK1b !9 BC37^ B))$QF!N#AaD$q!*bA B BrN)r __module__ __qualname__rerrrrdrrrrTs%EH Brr))>__doc__operatorrnumbersrrYnumpyrnumpy.exceptionsrnumpy.polynomialrrrrr r numpy.testingr r r rrr}classidsfixturerr3r'r6r7r=r@rGrLrUr`rkrrrtrrrrrrrrrrrrrrrrrrrrrrrdrrrs?  ()X X  11 1wH-.   "  &"&,!((&L)&X>,>,>.C8;6>6*>Z" (5:3$$$/& "&&(+ .+BBr