B )Ztol)polypolytrim)xrY/opt/alt/python37/lib64/python3.7/site-packages/numpy/polynomial/tests/test_polynomial.pytrim sriii i8i@ii iiic@s,eZdZddZddZddZddZd S) TestConstantscCsttjddgdS)Nrr)rr Z polydomain)selfrrrtest_polydomain szTestConstants.test_polydomaincCsttjdgdS)Nr)rr Zpolyzero)r"rrr test_polyzero#szTestConstants.test_polyzerocCsttjdgdS)Nr)rr Zpolyone)r"rrr test_polyone&szTestConstants.test_polyonecCsttjddgdS)Nrr)rr Zpolyx)r"rrr test_polyx)szTestConstants.test_polyxN)__name__ __module__ __qualname__r#r$r%r&rrrrr!sr!c@s4eZdZddZddZddZddZd d Zd S) TestArithmeticcCsxtdD]}xtdD]~}d||f}tt||d}||d7<||d7<tdg|dgdg|dg}tt|t||dqWq WdS)Nrz At i=%d, j=%drr)err_msg)rangenpzerosmaxr polyaddrr)r"ijmsgtgtresrrr test_polyadd/s $zTestArithmetic.test_polyaddcCsxtdD]}xtdD]~}d||f}tt||d}||d7<||d8<tdg|dgdg|dg}tt|t||dqWq WdS)Nrz At i=%d, j=%drr)r+)r,r-r.r/r Zpolysubrr)r"r1r2r3r4r5rrr test_polysub9s $zTestArithmetic.test_polysubcCsvttdgdgttdgddgxFtddD]8}dg|dg}dg|ddg}tt||q6WdS)Nrrr)rr Zpolymulxr,)r"r1Zserr4rrr test_polymulxCs zTestArithmetic.test_polymulxcCsxtdD]}x|tdD]p}d||f}t||d}|||d7<tdg|dgdg|dg}tt|t||dqWq WdS)Nrz At i=%d, j=%drr)r+)r,r-r.r polymulrr)r"r1r2r3r4r5rrr test_polymulKs $zTestArithmetic.test_polymulc Cstttjdgdgtdgdg\}}t||fdtddgdg\}}t||fdxtdD]}x~tdD]r}d||f}dg|ddg}dg|ddg}t||}t||\}}tt|||} t| ||dqrWqdWdS) Nrrr)rr))rrrrz At i=%d, j=%d)r+)rZeroDivisionErrorr Zpolydivrr,r0r9) r"ZquoZremr1r2r3ZciZcjr4r5rrr test_polydivTs  zTestArithmetic.test_polydivN)r'r(r)r6r7r8r:r<rrrrr*-s    r*c@seZdZedddgZedeeZedeeeZej dddZ e e dddgZ d d Zd d Zd dZddZddZddZdS)TestEvaluationg?g@g@zi,j->ijz i,j,k->ijk)rrrcsttgdgjdtddfddtdD}x:tdD].}||}tdg|dg}t||qBWdd}tddddg}t||xltdD]`}dg|}t|ttdgj |ttddgj |ttdddgj |qWdS) Nrrrcsg|] }|qSrr).0r1)r rr zsz/TestEvaluation.test_polyval..rrr>) rr polyvalsizer-linspacer,rr.shape)r"yr1r4r5dimsr)r r test_polyvalts     zTestEvaluation.test_polyvalc stttjdgdgddttgdgjdttgdgjdkttgdgdgjdttgdgdgjdkttdddttdt djd kt d dfd d t dD}x6t ddD](}||}tdg|}t ||qWdd}td ddg}t ||xnt d D]b}dg|}t |ttdgj|ttddgj|ttdddgj|qBWdddddg}t|}t d dt t|t|d}tddtjjdd|d}tj|dd}t|jdd}x4t |jD]&} t| |dd| f|| <q8Wt||tdgtj|dd}t|jddj}x\t |jdD]J} xBt jdD]0} t| |dd| f|| | ddf<qWqWt||dS)NrF)Ztensorr)rr)rr)r>r>)r>rcsg|] }|qSrr)r?r1)r rrr@sz8TestEvaluation.test_polyvalfromroots..r>ri)r>rr)rBT)r ValueErrorr ZpolyvalfromrootsrrBr rDr-onesrCr,rr. polyrootsrAarangerandomZrandintemptyvstack) r"rEr1r4r5rFZptestrZrshapeZiiZjjr)r rtest_polyvalfromrootssV         & 6z$TestEvaluation.test_polyvalfromrootsc Cs|j\}}}|j\}}}tttj||dd|j||}t|||j}t||t d} t| | |j}t |j dkdS)Nr)rr>) r rErrKr polyval2dc2drr-rLr rD) r"x1x2x3y1y2y3r4r5zrrrtest_polyval2ds    zTestEvaluation.test_polyval2dc Cs|j\}}}|j\}}}tttj|||dd|j|||}t||||j}t||t d} t| | | |j}t |j dkdS)Nr)rr>) r rErrKr polyval3dc3drr-rLr rD) r"rVrWrXrYrZr[r4r5r\rrrtest_polyval3ds     zTestEvaluation.test_polyval3dc Csl|j\}}}|j\}}}td||}t|||j}t||td} t| | |j}t |j dkdS)Nzi,j->ij)rr>)rr>rr>) r rEr-einsumr Z polygrid2drUrrLr rD) r"rVrWrXrYrZr[r4r5r\rrrtest_polygrid2ds    zTestEvaluation.test_polygrid2dc Csr|j\}}}|j\}}}td|||}t||||j}t||td} t| | | |j}t |j dkdS)Nz i,j,k->ijk)rr>)rr>rr>rr>) r rEr-rar Z polygrid3dr_rrLr rD) r"rVrWrXrYrZr[r4r5r\rrrtest_polygrid3ds    zTestEvaluation.test_polygrid3dN)r'r(r)r-arrayZc1drarUr_rOr r rArErGrSr]r`rbrcrrrrr=js@r=c@seZdZddZddZdS) TestIntegralc CstttjdgdtttjdgdtttjdgdddgxFtddD]8}dg|ddg}tjdg||d}t|ddgqHWxftdD]Z}|d}dg|dg}|gdg|d|g}tj|d|gd}tt|t|qWxNtdD]B}|d}dg|dg}tj|d|gdd}ttd||qWxjtdD]^}|d}dg|dg}|gdg|d|g}tj|d|gdd }tt|t|qFWxtdD]z}xrtddD]d}dg|dg}|dd}x t|D]}tj|dd }qWtj||d }tt|t|qWqWxtdD]}xtddD]r}dg|dg}|dd}x$t|D]}tj|d|gd}qtWtj||tt|d}tt|t|qJWq:WxtdD]}xtddD]v}dg|dg}|dd}x&t|D]}tj|d|gdd}q Wtj||tt|dd}tt|t|qWqWxtdD]}xtddD]v}dg|dg}|dd}x&t|D]}tj|d|gdd }qWtj||tt|dd }tt|t|qzWqjWdS) Nrg?rrrr)mk)rfrgZlbnd)rfrgscl)rf) rrKr polyintr,rrrAlist)r"r1rgr5rhZpolr4r2rrr test_polyint sp    zTestIntegral.test_polyintcCstjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||tdd|D}tj|d dd }t||dS) N)r>rcSsg|]}t|qSr)r ri)r?crrrr@[sz2TestIntegral.test_polyint_axis..r)axiscSsg|]}t|qSr)r ri)r?rlrrrr@_srcSsg|]}tj|ddqS)r>)rg)r ri)r?rlrrrr@csr>)rgrm)r-rOrQTr rir)r"rUr4r5rrrtest_polyint_axisWs   zTestIntegral.test_polyint_axisN)r'r(r)rkrorrrrre sKrec@seZdZddZddZdS)TestDerivativecCs.tttjdgdtttjdgdx@tdD]4}dg|dg}tj|dd}tt|t|q.Wx^tdD]R}xLtddD]>}dg|dg}tjtj||d|d}tt|t|qWqpWxbtdD]V}xPtddD]B}dg|dg}tjtj||dd|dd}tt|t|qWqWdS) Nrg?rrr)rfr)rfrh) rrKr polyderr,rrrir)r"r1r4r5r2rrr test_polyderjs zTestDerivative.test_polydercCsltjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||dS)N)r>rcSsg|]}t|qSr)r rq)r?rlrrrr@sz4TestDerivative.test_polyder_axis..r)rmcSsg|]}t|qSr)r rq)r?rlrrrr@sr)r-rOrQrnr rqr)r"rUr4r5rrrtest_polyder_axiss  z TestDerivative.test_polyder_axisN)r'r(r)rrrsrrrrrphsrpc@s8eZdZejdddZddZddZdd Zd S) TestVander)r>rrrcCstd}t|d}t|jdkx:tdD].}dg|dg}t|d|ft||q.Wt ddgddgdd gg}t|d}t|jd kx:tdD].}dg|dg}t|d|ft||qWdS) Nr>)r>rrrr.rr)r>rr) r-rNr Z polyvanderr rDr,rrArd)r"r vr1coefrrrtest_polyvanders   zTestVander.test_polyvandercCsx|j\}}}tjd}t||ddg}t|||}t||j}t||t|g|gddg}t |j dkdS)N)rr>rr)rrru) r r-rOr Z polyvander2drTdotflatrr rD)r"rVrWrXrlvanr4r5rrrtest_polyvander2ds   zTestVander.test_polyvander2dcCs|j\}}}tjd}t|||dddg}t||||}t||j}t||t|g|g|gdddg}t |j dkdS)N)rr>rrrr>)rr) r r-rOr Z polyvander3dr^ryrzrr rD)r"rVrWrXrlr{r4r5rrrtest_polyvander3ds   zTestVander.test_polyvander3dN) r'r(r)r-rOr rxr|r~rrrrrts rtc@s$eZdZddZddZddZdS) TestCompanioncCs"tttjgtttjdgdS)Nr)rrKr polycompanion)r"rrr test_raisesszTestCompanion.test_raisescCs@x:tddD],}dg|dg}tt|j||fkq WdS)Nrrr)r,r r rrD)r"r1rwrrrtest_dimensionsszTestCompanion.test_dimensionscCsttddgddkdS)Nrr)rrg)r r r)r"rrrtest_linear_rootszTestCompanion.test_linear_rootN)r'r(r)rrrrrrrrsrc@s4eZdZddZddZddZddZd d Zd S) TestMiscc Cstg}tt|dgxltddD]^}tttj dd|dddd}t |}t|d|d}tt|t|q&WdS)Nrrrr) r polyfromrootsrrr,r-ZcosrCZpiTlist)r"r5r1rootsr4rrrtest_polyfromrootss *zTestMisc.test_polyfromrootscCspttdggttddgdgxBtddD]4}tdd|}tt|}tt|t|q4WdS)Nrrgrr)rr rMr,r-rCrr)r"r1r4r5rrrtest_polyrootss zTestMisc.test_polyrootsc Csdd}dd}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggd tttjdgdgdddgd tttjdgdgdgtttjdgdgddd gtttjdgdggtdd}||}t||d }tt|d t t |||t||dddd g}tt|d t t |||t||d }tt|d t t |||t||dddd d g}tt|d t t |||t|t ||gj d }t |t ||gj t|t ||gj dddd g}t |t ||gj t |}|} d|ddd<d| ddd<tj|| d |d } t | |tj|| dddd g|d } t | |tj|t | | gj d |d } t | t ||gj tj|t | | gj dddd g|d } t | t ||gj ddddg}t t||dddgt t||ddgddgtdd}||}t||d } t t || |t||ddd g} t t || |t | | dS)NcSs||d|dS)Nrrr)r rrrfsz TestMisc.test_polyfit..fcSs|d|ddS)Nrrrr)r rrrf2sz!TestMisc.test_polyfit..f2rrrr)wrur>rry?y)rrKr Zpolyfit TypeErrorr-rCrlenrrArdrnZ zeros_likecopy)r"rrr rEZcoef3Zcoef4Zcoef2drZywZwcoef3Zwcoef2dZcoef1Zcoef2rrr test_polyfitsj "   &  zTestMisc.test_polyfitcCsfddddg}tttj|dtt||ddtt|d|ddtt|ddgdS)Nrrrrr)rrKr r r)r"rwrrr test_polytrim(s  zTestMisc.test_polytrimcCsttddddgdS)Nr>r)rr Zpolyline)r"rrr test_polyline3szTestMisc.test_polylineN)r'r(r)rrrrrrrrrrs  F r__main__)&__doc__Z __future__rrrZnumpyr-Znumpy.polynomial.polynomialZ polynomialr Z numpy.testingrrrrr r rZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9rr!r*r=rerprtrrr'rrrrs6   =!^(0h