B )Ztol)chebchebtrim)xrX/opt/alt/python37/lib64/python3.7/site-packages/numpy/polynomial/tests/test_chebyshev.pytrimsriii i8i@ii iiic@seZdZddZddZdS) TestPrivatecCshxbtdD]V}tdgdg|tj}tdg|dgdg|tj}t|}t||q WdS)Nrrrg?)rangenparraydoubler Z_cseries_to_zseriesr )selfiinptgtresrrrtest__cseries_to_zseries!s $ z$TestPrivate.test__cseries_to_zseriescCshxbtdD]V}tdg|dgdg|tj}tdgdg|tj}t|}t||q WdS)Nrg?rr)r#r$r%r&r Z_zseries_to_cseriesr )r'r(r)r*r+rrrtest__zseries_to_cseries(s $ z$TestPrivate.test__zseries_to_cseriesN)__name__ __module__ __qualname__r,r-rrrrr"sr"c@s,eZdZddZddZddZddZd S) TestConstantscCsttjddgdS)Nrr)r r Z chebdomain)r'rrrtest_chebdomain2szTestConstants.test_chebdomaincCsttjdgdS)Nr)r r Zchebzero)r'rrr test_chebzero5szTestConstants.test_chebzerocCsttjdgdS)Nr)r r Zchebone)r'rrr test_chebone8szTestConstants.test_chebonecCsttjddgdS)Nrr)r r Zchebx)r'rrr test_chebx;szTestConstants.test_chebxN)r.r/r0r2r3r4r5rrrrr10sr1c@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)r#r$zerosmaxr chebaddr r)r'r(jmsgr*r+rrr test_chebaddAs $zTestArithmetic.test_chebaddcCsxtdD]}xtdD]~}d||f}tt||d}||d7<||d8<tdg|dgdg|dg}tt|t||dqWq WdS)Nrz At i=%d, j=%drr)r7)r#r$r8r9r Zchebsubr r)r'r(r;r<r*r+rrr test_chebsubKs $zTestArithmetic.test_chebsubcCszttdgdgttdgddgxJtddD]<}dg|dg}dg|ddddg}tt||q6WdS)Nrrrg?)r r Zchebmulxr#)r'r(Zserr*rrr test_chebmulxUs zTestArithmetic.test_chebmulxcCsxtdD]}xtdD]}d||f}t||d}|||d7<|t||d7<tdg|dgdg|dg}tt|t||dqWq WdS)Nrz At i=%d, j=%drg?r)r7)r#r$r8absr chebmulr r)r'r(r;r<r*r+rrr test_chebmul]s $zTestArithmetic.test_chebmulc CsxtdD]}xtdD]v}d||f}dg|dg}dg|dg}t||}t||\}}tt|||} tt| t||dqWq WdS)Nrz At i=%d, j=%drr)r7)r#r r:ZchebdivrAr r) r'r(r;r<ZciZcjr*ZquoZremr+rrr test_chebdivgs  zTestArithmetic.test_chebdivN)r.r/r0r=r>r?rBrCrrrrr6?s    r6c@seZdZedddgZedeeZedeeeZej dddZ e e d dd gZ d d Z d dZddZddZddZdS)TestEvaluationg@g@g?zi,j->ijz i,j,k->ijk)rrrg?g@csttgdgjdtddfddtD}xFtdD]:}d|}||}tdg|dg}t|||dq>Wxltd D]`}d g|}t |ttdgj |ttddgj |ttdddgj |qWdS) Nrrrcsg|]}t|qSr)r).0c)rrr sz/TestEvaluation.test_chebval.. zAt i=%d)r7rEr) r r chebvalsizer$linspaceTlistr#rr8shape)r'yr(r<r*r+Zdimsr)rr test_chebval}s   zTestEvaluation.test_chebvalc Cs|j\}}}|j\}}}tttj||dd|j||}t|||j}t||t d} t| | |j}t |j dkdS)Nr)rrE) rrOr ValueErrorr chebval2dc2drr$onesr rN) r'x1x2x3y1y2y3r*r+zrrrtest_chebval2ds    zTestEvaluation.test_chebval2dc Cs|j\}}}|j\}}}tttj|||dd|j|||}t||||j}t||t d} t| | | |j}t |j dkdS)Nr)rrE) rrOrrQr chebval3dc3drr$rTr rN) r'rUrVrWrXrYrZr*r+r[rrrtest_chebval3ds     zTestEvaluation.test_chebval3dc Csl|j\}}}|j\}}}td||}t|||j}t||td} t| | |j}t |j dkdS)Nzi,j->ij)rrE)rrErrE) rrOr$einsumr Z chebgrid2drSrrTr rN) r'rUrVrWrXrYrZr*r+r[rrrtest_chebgrid2ds    zTestEvaluation.test_chebgrid2dc Csr|j\}}}|j\}}}td|||}t||||j}t||td} t| | | |j}t |j dkdS)Nz i,j,k->ijk)rrE)rrErrErrE) rrOr$r`r Z chebgrid3dr^rrTr rN) r'rUrVrWrXrYrZr*r+r[rrrtest_chebgrid3ds    zTestEvaluation.test_chebgrid3dN)r.r/r0r$r%Zc1dr`rSr^randomrrrOrPr\r_rarbrrrrrDssrDc@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?rrrr)mk)rerfZlbnd)rerfscl)re) rrQr chebintr#r poly2cheb cheb2polyrrJlist) r'r(rfr+rgZpolr*Zchebpolrhr;rrr test_chebintsz         zTestIntegral.test_chebintcCstjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||tdd|D}tj|d dd }t||dS) N)rErcSsg|]}t|qSr)r rh)rFrGrrrrH'sz2TestIntegral.test_chebint_axis..r)axiscSsg|]}t|qSr)r rh)rFrGrrrrH+srcSsg|]}tj|ddqS)rE)rf)r rh)rFrGrrrrH/srE)rfrm)r$rcvstackTr rhr)r'rSr*r+rrrtest_chebint_axis#s   zTestIntegral.test_chebint_axisN)r.r/r0rlrprrrrrdsPrdc@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)rer)rerg) rrQr chebderr#r rrhr)r'r(r*r+r;rrr test_chebder6s zTestDerivative.test_chebdercCsltjd}tdd|jDj}tj|dd}t||tdd|D}tj|dd}t||dS)N)rErcSsg|]}t|qSr)r rr)rFrGrrrrHSsz4TestDerivative.test_chebder_axis..r)rmcSsg|]}t|qSr)r rr)rFrGrrrrHWsr)r$rcrnror rrr)r'rSr*r+rrrtest_chebder_axisOs  z TestDerivative.test_chebder_axisN)r.r/r0rsrtrrrrrq4srqc@s8eZdZejdddZddZddZdd Zd S) TestVander)rErrrcCstd}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) NrE)rErrrr.rr)rErr) r$Zaranger chebvanderr rNr#rrJr%)r'rvr(coefrrrtest_chebvander`s   zTestVander.test_chebvandercCsx|j\}}}tjd}t||ddg}t|||}t||j}t||t|g|gddg}t |j dkdS)N)rrErr)rrrv) rr$rcr Z chebvander2drRdotflatrr rN)r'rUrVrWrGvanr*r+rrrtest_chebvander2dqs   zTestVander.test_chebvander2dcCs|j\}}}tjd}t|||dddg}t||||}t||j}t||t|g|g|gdddg}t |j dkdS)N)rrErrrrE)rr) rr$rcr Z chebvander3dr]r{r|rr rN)r'rUrVrWrGr}r*r+rrrtest_chebvander3d~s   zTestVander.test_chebvander3dN) r.r/r0r$rcrrzr~rrrrrru\s ruc@seZdZddZdS) TestFittingc Cs&dd}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||dd d ddg}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)rrrrfsz#TestFitting.test_chebfit..fcSs|d|ddS)Nrrrr)rrrrf2sz$TestFitting.test_chebfit..f2rrrr)wrvrErry?y)rrQr Zchebfit TypeErrorr$rLr lenrrJr%roZ zeros_likecopy)r'rrrrOZcoef3Zcoef4Zcoef2drZywZwcoef3Zwcoef2dZcoef1Zcoef2rrr test_chebfitsp "   &  zTestFitting.test_chebfitN)r.r/r0rrrrrrsrc@s$eZdZddZddZddZdS) TestCompanioncCs"tttjgtttjdgdS)Nr)rrQr chebcompanion)r'rrr test_raisesszTestCompanion.test_raisescCs@x:tddD],}dg|dg}tt|j||fkq WdS)Nrrr)r#r r rrN)r'r(ryrrrtest_dimensionsszTestCompanion.test_dimensionscCsttddgddkdS)Nrr)rrg)r r r)r'rrrtest_linear_rootszTestCompanion.test_linear_rootN)r.r/r0rrrrrrrrsrc@seZdZddZdS) TestGausscCs~td\}}t|d}t|j||}dt|}|dddf||}t|t dtj }t| |dS)Ndcr) r Z chebgaussrwr$r{rosqrtZdiagonalrZeyepisum)r'rrrxZvvZvdr*rrrtest_100s zTestGauss.test_100N)r.r/r0rrrrrrsrc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestMiscc Cstg}tt|dgxttddD]f}tttj dd|dddd}dg|dg}t|d|d}tt|t|q&WdS)Nrrrr) r chebfromrootsrrr#r$ZcosrLr)r'r+r(rootsr*rrrtest_chebfromrootss *zTestMisc.test_chebfromrootscCspttdggttddgdgxBtddD]4}tdd|}tt|}tt|t|q4WdS)Nrrgrr)rr Z chebrootsr#r$rLrr)r'r(r*r+rrrtest_chebrootss zTestMisc.test_chebrootscCsfddddg}tttj|dtt||ddtt|d|ddtt|ddgdS)Nrrrrr)rrQr r r )r'ryrrr test_chebtrims  zTestMisc.test_chebtrimcCsttddddgdS)NrEr)r r Zchebline)r'rrr test_cheblineszTestMisc.test_cheblinecCs6x0tdD]$}ttdg|dgt|q WdS)NrIrr)r#rr rjrM)r'r(rrrtest_cheb2polyszTestMisc.test_cheb2polycCs6x0tdD]$}ttt|dg|dgq WdS)NrIrr)r#rr rirM)r'r(rrrtest_poly2cheb szTestMisc.test_poly2chebcCsNtddddd}dtd|td|}t|}t||dS)Nrr g?)r$rLrr Z chebweightr)r'rr*r+rrr test_weight$s  zTestMisc.test_weightcCstttjdtttjddg}ttd|ddg}ttd|dddg}ttd |d d d d g}ttd|dS)Ng?rrg;fg;f?rgLXzgLXz?rEg(1kgŜ}ؿgŜ}?g(1k?r)rrQr Zchebpts1r)r'r*rrr test_chebpts1*s  zTestMisc.test_chebpts1cCstttjdtttjdddg}ttd|dddg}ttd|ddddg}ttd |d d dd d g}ttd|dS)Ng?rrrrrEgg?rggKfgKf?g?r)rrQr Zchebpts2r)r'r*rrr test_chebpts29s  zTestMisc.test_chebpts2N) r.r/r0rrrrrrrrrrrrrrs  r__main__),__doc__Z __future__rrrZnumpyr$Znumpy.polynomial.chebyshevZ polynomialZ chebyshevr Znumpy.polynomial.polynomialrZ numpy.testingrrrr r r rZT0ZT1ZT2ZT3ZT4ZT5ZT6ZT7ZT8ZT9rMr"r1r6rDrdrqrurrrrr.rrrrs<    4^c(0MM