B )Ztol)laglagtrim)xrW/opt/alt/python37/lib64/python3.7/site-packages/numpy/polynomial/tests/test_laguerre.pytrimsrc@s,eZdZddZddZddZddZd S) TestConstantscCsttjddgdS)Nrr )r rZ lagdomain)selfrrrtest_lagdomainszTestConstants.test_lagdomaincCsttjdgdS)Nr)r rZlagzero)rrrr test_lagzero!szTestConstants.test_lagzerocCsttjdgdS)Nr )r rZlagone)rrrr test_lagone$szTestConstants.test_lagonecCsttjddgdS)Nr r )r rZlagx)rrrr test_lagx'szTestConstants.test_lagxN)__name__ __module__ __qualname__rrr r!rrrrrsrc@sBeZdZedddZddZddZdd Zd d Z d d Z dS)TestArithmeticdcCsxtdD]}xtdD]~}d||f}tt||d}||d7<||d7<tdg|dgdg|dg}tt|t||dqWq WdS)Nz At i=%d, j=%dr r)err_msg)rangenpzerosmaxrlagaddr r)rijmsgtgtresrrr test_lagadd.s $zTestArithmetic.test_lagaddcCsxtdD]}xtdD]~}d||f}tt||d}||d7<||d8<tdg|dgdg|dg}tt|t||dqWq WdS)Nr)z At i=%d, j=%dr r)r*)r+r,r-r.rZlagsubr r)rr0r1r2r3r4rrr test_lagsub8s $zTestArithmetic.test_lagsubcCsttdgdgttdgddgxZtddD]L}dg|dg}dg|d| d|d|d g}tt||q6WdS)Nrr r r)r)r rZlagmulxr+r)rr0Zserr3rrr test_lagmulxBs (zTestArithmetic.test_lagmulxc CsxtdD]}dg|dg}t|j|}xtdD]t}d||f}dg|dg}t|j|}t||}t|j|} tt|||dk|t| |||dq6Wq WdS)Nr)rr z At i=%d, j=%d)r*)r+rlagvalrlagmulr lenr) rr0Zpol1Zval1r1r2Zpol2Zval2Zpol3Zval3rrr test_lagmulJs  zTestArithmetic.test_lagmulc CsxtdD]}xtdD]v}d||f}dg|dg}dg|dg}t||}t||\}}tt|||} tt| t||dqWq WdS)Nr)z At i=%d, j=%drr )r*)r+rr/Zlagdivr9rr) rr0r1r2ZciZcjr3ZquoZremr4rrr test_lagdivXs  zTestArithmetic.test_lagdivN) r"r#r$r,linspacerr5r6r7r;r<rrrrr%+s   r%c@seZdZedddgZedeeZedeeeZej dddZ e e d d d gZ d d Z ddZddZddZddZdS)TestEvaluationg"@g,g@zi,j->ijz i,j,k->ijk)r'r)rr g?g@g@csttgdgjdtddfddtD}xFtdD]:}d|}||}tdg|dg}t|||dq>Wxltd D]`}d g|}t |ttdgj |ttddgj |ttdddgj |qWdS) Nr rr csg|]}t|qSr)r).0c)rrr tsz.TestEvaluation.test_lagval..zAt i=%d)r*r'r) r rr8sizer,r=Llistr+rr-shape)ryr0r2r3r4Zdimsr)rr test_lagvalns   zTestEvaluation.test_lagvalc Cs|j\}}}|j\}}}tttj||dd|j||}t|||j}t||t d} t| | |j}t |j dkdS)Nr)rr') rrFr ValueErrorrlagval2dc2drr,onesr rE) rx1x2x3y1y2y3r3r4zrrr test_lagval2ds    zTestEvaluation.test_lagval2dc Cs|j\}}}|j\}}}tttj|||dd|j|||}t||||j}t||t d} t| | | |j}t |j dkdS)Nr)rr') rrFrrHrlagval3dc3drr,rKr rE) rrLrMrNrOrPrQr3r4rRrrr test_lagval3ds     zTestEvaluation.test_lagval3dc Csl|j\}}}|j\}}}td||}t|||j}t||td} t| | |j}t |j dkdS)Nzi,j->ij)rr')rr'rr') rrFr,einsumrZ laggrid2drJrrKr rE) rrLrMrNrOrPrQr3r4rRrrrtest_laggrid2ds    zTestEvaluation.test_laggrid2dc Csr|j\}}}|j\}}}td|||}t||||j}t||td} t| | | |j}t |j dkdS)Nz i,j,k->ijk)rr')rr'rr'rr') rrFr,rWrZ laggrid3drUrrKr rE) rrLrMrNrOrPrQr3r4rRrrrtest_laggrid3ds    zTestEvaluation.test_laggrid3dN)r"r#r$r,arrayZc1drWrJrUrandomrrrFrGrSrVrXrYrrrrr>dsr>c@seZdZddZddZdS) TestIntegralc Cs2tttjdgdtttjdgdtttjdgdddgxFtddD]8}dg|ddg}tjdg||d}t|ddgqHWxztdD]n}|d}dg|dg}|gdg|d|g}t|}tj|d|gd}t|}tt|t|qWxZtdD]N}|d}dg|dg}t|}tj|d|gdd}tt d||q Wx~tdD]r}|d}dg|dg}|gdg|d|g}t|}tj|d|gdd }t|}tt|t|qfWxtdD]z}xrtddD]d} dg|dg}|dd}x t| D]}tj|dd }q Wtj|| d }tt|t|qWqWxtdD]}xtddD]r} dg|dg}|dd}x$t| D]}tj|d|gd}qWtj|| t t| d}tt|t|q~WqnWxtdD]}xtddD]v} dg|dg}|dd}x&t| D]}tj|d|gdd}q>Wtj|| t t| dd}tt|t|qWqWxtdD]}xtddD]v} dg|dg}|dd}x&t| D]}tj|d|gdd }qWtj|| t t| dd }tt|t|qWqWdS) Nrg?r r rr))mk)r]r^Zlbnd)r]r^scl)r]) rrHrlagintr+rpoly2laglag2polyrr8list) rr0r^r4r_polr3Zlagpolr`r1rrr test_lagintsz         zTestIntegral.test_lagintcCstjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||tdd|D}tj|d dd }t||dS) N)r'cSsg|]}t|qSr)rr`)r?r@rrrrAsz1TestIntegral.test_lagint_axis..r)axiscSsg|]}t|qSr)rr`)r?r@rrrrAsr cSsg|]}tj|ddqS)r')r^)rr`)r?r@rrrrA sr')r^rg)r,r[vstackTrr`r)rrJr3r4rrrtest_lagint_axiss   zTestIntegral.test_lagint_axisN)r"r#r$rerjrrrrr\sPr\c@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?r r)r )r]r)r]r_) rrHrlagderr+r rr`r)rr0r3r4r1rrr test_lagder's zTestDerivative.test_lagdercCsltjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||dS)N)r'rfcSsg|]}t|qSr)rrl)r?r@rrrrADsz3TestDerivative.test_lagder_axis..r)rgcSsg|]}t|qSr)rrl)r?r@rrrrAHsr )r,r[rhrirrlr)rrJr3r4rrrtest_lagder_axis@s  zTestDerivative.test_lagder_axisN)r"r#r$rmrnrrrrrk%srkc@s8eZdZejdddZddZddZdd Zd S) TestVander)r'r)rr cCstd}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'rfrfrr .rr)r)r'rrf) r,Zaranger lagvanderr rEr+rr8rZ)rrvr0coefrrrtest_lagvanderQs   zTestVander.test_lagvandercCsx|j\}}}tjd}t||ddg}t|||}t||j}t||t|g|gddg}t |j dkdS)N)rr'r r)r r)r) rr,r[rZ lagvander2drIdotflatrr rE)rrLrMrNr@vanr3r4rrrtest_lagvander2dbs   zTestVander.test_lagvander2dcCs|j\}}}tjd}t|||dddg}t||||}t||j}t||t|g|g|gdddg}t |j dkdS)N)rr'rfr rr')r r)r) rr,r[rZ lagvander3drTrtrurr rE)rrLrMrNr@rvr3r4rrrtest_lagvander3dos   zTestVander.test_lagvander3dN) r"r#r$r,r[rrsrwrxrrrrroMs roc@seZdZddZdS) TestFittingc Csdd}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggdtttjdgdgdddgdtttjdgdgdgtttjdgdgdddgtttjdgdggtdd}||}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 dd dd g}t t||dddgt t||ddgddgdS)NcSs||d|dS)Nr rr)rrrrfsz"TestFitting.test_lagfit..fr r rr)wrr'rfr)y?y)rrHrZlagfit TypeErrorr,r=r r:rr8rZriZ zeros_likecopy) rrzrrFZcoef3Zcoef4Zcoef2dr{ZywZwcoef3Zwcoef2drrr test_lagfitsZ "   & zTestFitting.test_lagfitN)r"r#r$r~rrrrry}sryc@s$eZdZddZddZddZdS) TestCompanioncCs"tttjgtttjdgdS)Nr )rrHr lagcompanion)rrrr test_raisesszTestCompanion.test_raisescCs@x:tddD],}dg|dg}tt|j||fkq WdS)Nr r)r)r+r rrrE)rr0rrrrrtest_dimensionsszTestCompanion.test_dimensionscCsttddgddkdS)Nr r)rrg?)r rr)rrrrtest_linear_rootszTestCompanion.test_linear_rootN)r"r#r$rrrrrrrrsrc@seZdZddZdS) TestGausscCs|td\}}t|d}t|j||}dt|}|dddf||}t|t dd}t| |dS)Nr(cr g?) rZlaggaussrpr,rtriZsqrtZdiagonalrZeyesum)rrr{rqZvvZvdr3rrrtest_100s zTestGauss.test_100N)r"r#r$rrrrrrsrc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestMiscc Cstg}tt|dgxtddD]z}tttj dd|dddd}t|}t ||}d}t t ||dktt |ddt||q&WdS)Nr r)rrr ) r lagfromrootsrrr+r,Zcosr=Zpir8r r:rb)rr4r0rootsrdr3rrrtest_lagfromrootss *  zTestMisc.test_lagfromrootscCspttdggttddgdgxBtddD]4}tdd|}tt|}tt|t|q4WdS)Nr rrr)r')rrZlagrootsr+r,r=rr)rr0r3r4rrr test_lagrootss zTestMisc.test_lagrootscCsfddddg}tttj|dtt||ddtt|d|ddtt|ddgdS)Nrr r rr&)rrHrrr )rrrrrr test_lagtrims  zTestMisc.test_lagtrimcCsttddddgdS)Nr'rfrBr)r rZlagline)rrrr test_laglineszTestMisc.test_laglinecCs6x0tdD]$}ttdg|dgt|q WdS)NrBrr )r+rrrbrD)rr0rrr test_lag2polyszTestMisc.test_lag2polycCs6x0tdD]$}ttt|dg|dgq WdS)NrBrr )r+rrrarD)rr0rrr test_poly2lagszTestMisc.test_poly2lagcCs2tddd}t| }t|}t||dS)Nr )r,r=ZexprZ lagweightr)rrr3r4rrr test_weight s  zTestMisc.test_weightN) r"r#r$rrrrrrrrrrrrs  r__main__))__doc__Z __future__rrrZnumpyr,Znumpy.polynomial.laguerreZ polynomialZlaguerrerZnumpy.polynomial.polynomialrZ numpy.testingrrrr r r rZZL0ZL1ZL2ZL3ZL4ZL5ZL6rDrrr%r>r\rkroryrrrr"rrrrs4  9^c(0>3