fume-manage-python.git

U  
P±dFã    @sdZddlmZddlZddlmmZddl    m
Z
ddlmZm Z mZmZe dg¡dZe ddg¡dZe dd    dg¡dZe d
dddg¡d
Ze d ddddg¡d Ze ddddddg¡dZe dddddddg¡dZeeeeeeegZddZGdddZGd d!d!ZGd"d#d#ZGd$d%d%ZGd&d'd'ZGd(d)d)ZGd*d+d+Z Gd,d-d-Z!Gd.d/d/Z"Gd0d1d1Z#dS)2zTests for laguerre module.
 
é)ÚreduceN©Úpolyval)Úassert_almost_equalÚ assert_raisesÚassert_equalÚassert_ééÿÿÿÿééüÿÿÿéiîÿÿÿé    éi ÿÿÿéHiðÿÿÿéxi¨ýÿÿiXi8ÿÿÿéiÐi ïÿÿii öÿÿiÂiÜÿÿÿcCstj|ddS)Ngíµ ÷Æ°>)Ztol)ÚlagÚlagtrim©Úx©rú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_laguerre.pyÚtrimsrc@s,eZdZddZddZddZddZd    S)
Ú TestConstantscCsttjddgdS)Nrr    )rrZ    lagdomain©ÚselfrrrÚtest_lagdomainszTestConstants.test_lagdomaincCsttjdgdS)Nr)rrZlagzerorrrrÚtest_lagzero!szTestConstants.test_lagzerocCsttjdgdS©Nr    )rrZlagonerrrrÚtest_lagone$szTestConstants.test_lagonecCsttjddgdS)Nr    r
)rrZlagxrrrrÚ    test_lagx'szTestConstants.test_lagxN)Ú__name__Ú
__module__Ú__qualname__rrr r!rrrrrsrc@sJeZdZe ddd¡ZddZddZdd    Zd
dZ    dd Z
ddZdS)ÚTestArithmeticéýÿÿÿéédcCstdD]}tdD]}d|d|}t t||d¡}||d7<||d7<t dg|dgdg|dg¡}tt|t||dqqdS©NéúAt i=ú, j=r    r©Úerr_msg)ÚrangeÚnpÚzerosÚmaxrÚlagaddrr©rÚiÚjÚmsgÚtgtÚresrrrÚtest_lagadd.s$zTestArithmetic.test_lagaddcCstdD]}tdD]}d|d|}t t||d¡}||d7<||d8<t dg|dgdg|dg¡}tt|t||dqqdSr))r/r0r1r2rZlagsubrrr4rrrÚtest_lagsub8s$zTestArithmetic.test_lagsubcCstt dg¡dgtt dg¡ddgtddD]L}dg|dg}dg|d|d|d|dg}tt |¡|q4dS)Nrr    r
r*r)rrZlagmulxr/r)rr5Zserr8rrrÚtest_lagmulxBs(zTestArithmetic.test_lagmulxc
Cs²tdD]¤}dg|dg}t |j|¡}tdD]x}d|d|}dg|dg}t |j|¡}t ||¡}t |j|¡}    tt|||dk|t|    |||dq2qdS)Nr*rr    r+r,r-)r/rÚlagvalrÚlagmulrÚlenr)
rr5Zpol1Zval1r6r7Zpol2Zval2Zpol3Zval3rrrÚtest_lagmulJszTestArithmetic.test_lagmulc
CstdD]}tdD]z}d|d|}dg|dg}dg|dg}t ||¡}t ||¡\}}t t ||¡|¡}    tt|    t||dqqdS)Nr*r+r,rr    r-)r/rr3Zlagdivr>rr)
rr5r6r7ÚciÚcjr8ZquoÚremr9rrrÚtest_lagdivXszTestArithmetic.test_lagdivcCs|tdD]n}tdD]`}d|d|}t |d¡}ttj|g|t dg¡}t ||¡}tt    |t    ||dqqdS)Nr*r+r,r    r-)
r/r0Úarangerrr>ÚarrayZlagpowrr)rr5r6r7Úcr8r9rrrÚtest_lagpowcszTestArithmetic.test_lagpowN)r"r#r$r0Úlinspacerr:r;r<r@rDrHrrrrr%+s
 
r%c@seZdZe dddg¡Ze dee¡Ze deee¡Zej         d¡ddZ
ee
d    d
dgZdd Z ddZddZddZddZdS)ÚTestEvaluationg"@g,Àg@úi,j->ijú
i,j,k->ijk©r'r*rr    çð?g@g@csètt gdg¡jdt dd¡fddtD}tdD]<}d|}||}t dg|dg¡}t|||dq<td    D]`}d
g|}t     |¡tt dg¡j
|tt ddg¡j
|tt dddg¡j
|qdS)Nr    rr
csg|]}t|qSrr©Ú.0rGrrrÚ
<listcomp>}sz.TestEvaluation.test_lagval.<locals>.<listcomp>ér+r-r'r)rrr=Úsizer0rIÚLlistr/rr1Úshape)rÚyr5r7r8r9ZdimsrrrÚtest_lagvalws
 
 
zTestEvaluation.test_lagvalc
Cs|j\}}}|j\}}}tttj||dd|j||}t |||j¡}t||t     d¡}    t |    |    |j¡}t
|jdkdS©Nr©rr')rrVrÚ
ValueErrorrÚlagval2dÚc2drr0ÚonesrrU©
rÚx1Úx2Úx3Úy1Úy2Zy3r8r9ÚzrrrÚ test_lagval2ds
 
zTestEvaluation.test_lagval2dc
Cs|j\}}}|j\}}}tttj|||dd|j|||}t ||||j¡}t||t     d¡}    t |    |    |    |j¡}t
|jdkdSrX)rrVrrZrÚlagval3dÚc3drr0r]rrUr^rrrÚ test_lagval3ds
 
zTestEvaluation.test_lagval3dc
Csl|j\}}}|j\}}}t d||¡}t |||j¡}t||t d¡}    t |    |    |j¡}t    |j
dkdS)NrKrY)rr'rr')rrVr0ÚeinsumrZ    laggrid2dr\rr]rrUr^rrrÚtest_laggrid2d®s
 
zTestEvaluation.test_laggrid2dc
Csr|j\}}}|j\}}}t d|||¡}t ||||j¡}t||t d¡}    t |    |    |    |j¡}t    |j
dkdS)NrLrY)rr'rr'rr')rrVr0rirZ    laggrid3drgrr]rrUr^rrrÚtest_laggrid3d¼s
 
zTestEvaluation.test_laggrid3dN)r"r#r$r0rFZc1drir\rgÚrandomrrrVrWrerhrjrkrrrrrJmsrJc@seZdZddZddZdS)ÚTestIntegralc
    Cs2tttjdgdtttjdgdtttjdgdddgtttjdgdgdtttjdgdgdtttjdgddtdd    D]8}dg|ddg}tjdg||d
}t|ddgqtd    D]n}|d}dg|dg}|gdg|d|g}t |¡}tj|d|gd
}t |¡}tt    |t    |qÈtd    D]N}|d}dg|dg}t |¡}tj|d|gdd}tt 
d|¡|q@td    D]r}|d}dg|dg}|gdg|d|g}t |¡}tj|d|gdd}t |¡}tt    |t    |qtd    D]r}tdd    D]`}    dg|dg}|dd}t|    D]}tj|dd }qJtj||    d }tt    |t    |q"qtd    D]}tdd    D]n}    dg|dg}|dd}t|    D]}tj|d|gd
}qÆtj||    tt|    d
}tt    |t    |qqtd    D]}tdd    D]r}    dg|dg}|dd}t|    D]}tj|d|gdd}qPtj||    tt|    dd}tt    |t    |q(qtd    D]}tdd    D]r}    dg|dg}|dd}t|    D]}tj|d|gdd}qÞtj||    tt|    dd}tt    |t    |q¶q¨dS)Nrçà?r
r    )Úlbnd)Úscl©Úaxisrr*)ÚmÚk)rsrtro)rsrtrp©rs)rÚ    TypeErrorrÚlagintrZr/rÚpoly2lagÚlag2polyrr=Úlist)
rr5rtr9rpÚpolr8Zlagpolrwr6rrrÚtest_lagintÍs
 
 
 
 
zTestIntegral.test_lagintcCstj d¡}t dd|jD¡j}tj|dd}t||t dd|D¡}tj|dd}t||t dd|D¡}tj|d    dd
}t||dS)N©r'écSsg|]}t |¡qSr©rrwrOrrrrQ$sz1TestIntegral.test_lagint_axis.<locals>.<listcomp>rrqcSsg|]}t |¡qSrrrOrrrrQ(sr    cSsg|]}tj|ddqS)r')rtrrOrrrrQ,sr')rtrr)r0rlÚvstackÚTrrwr©rr\r8r9rrrÚtest_lagint_axis s
 
zTestIntegral.test_lagint_axisN)r"r#r$r|rrrrrrmËsSrmc@seZdZddZddZdS)ÚTestDerivativecCstttjdgdtttjdgdtdD]4}dg|dg}tj|dd}tt|t|q,tdD]N}tddD]>}dg|dg}tjtj||d|d}t    t|t|qxqjtdD]R}tddD]B}dg|dg}tjtj||dd|dd}t    t|t|qÐqÂdS)    Nrrnr
r*r    rur)rsrp)
rrvrÚlagderrZr/rrrwr)rr5r8r9r6rrrÚtest_lagder3s zTestDerivative.test_lagdercCsltj d¡}t dd|jD¡j}tj|dd}t||t dd|D¡}tj|dd}t||dS)Nr}cSsg|]}t |¡qSr©rrrOrrrrQPsz3TestDerivative.test_lagder_axis.<locals>.<listcomp>rrqcSsg|]}t |¡qSrrrOrrrrQTsr    )r0rlrrrrrrrrrÚtest_lagder_axisLs
zTestDerivative.test_lagder_axisN)r"r#r$rrrrrrr1src@s8eZdZej d¡ddZddZddZdd    Zd
S)Ú
TestVanderrMrr    cCsÎt d¡}t |d¡}t|jdktdD].}dg|dg}t|d|ft ||¡q,t     ddgddgdd    gg¡}t |d¡}t|jd
ktdD].}dg|dg}t|d|ft ||¡qdS)Nr'r}r~rr    .rr*r )r'rr~)
r0rErÚ    lagvanderrrUr/rr=rF)rrÚvr5ÚcoefrrrÚtest_lagvander]s
zTestVander.test_lagvandercCsx|j\}}}tj d¡}t ||ddg¡}t |||¡}t ||j¡}t||t |g|gddg¡}t    |j
dkdS)NrYr    r)r    r*r )rr0rlrZlagvander2dr[ÚdotÚflatrrrU©rr_r`rarGZvanr8r9rrrÚtest_lagvander2dns
zTestVander.test_lagvander2dcCs|j\}}}tj d¡}t |||dddg¡}t ||||¡}t ||j¡}t||t |g|g|gdddg¡}t    |j
dkdS)N)rr'r~r    rr')r    r*r)rr0rlrZlagvander3drfrrrrrUrrrrÚtest_lagvander3d{s
zTestVander.test_lagvander3dN)    r"r#r$r0rlrrrrrrrrrYs rc@seZdZddZdS)ÚTestFittingc    Csdd}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggdtttjdgdgdddgdtttjdgdgdgtttjdgdgdddgtttjdgdggt dd¡}||}t ||d    ¡}tt|d
t    t 
||¡|t ||dddd    g¡}tt|d
t    t 
||¡|t ||d
¡}tt|dt    t 
||¡|t ||dddd    d
g¡}tt|dt    t 
||¡|t |t ||g¡jd    ¡}t    |t ||g¡jt |t ||g¡jdddd    g¡}t    |t ||g¡jt  |¡}| ¡}d|ddd<d|ddd<tj||d    |d}    t    |    |tj||dddd    g|d}    t    |    |tj|t ||g¡jd    |d}
t    |
t ||g¡jtj|t ||g¡jdddd    g|d}
t    |
t ||g¡jdddd g}t    t ||d¡ddgt    t ||ddg¡ddgdS)NcSs||d|dS)Nr    rrrrrrÚfsz"TestFitting.test_lagfit.<locals>.fr    r
rr)Úwr r'r~r*yð?yð¿)rrZrZlagfitrvr0rIrr?rr=rFrZ
zeros_likeÚcopy)rrrrVZcoef3Zcoef4Zcoef2drZywZwcoef3Zwcoef2drrrÚtest_lagfitsZ"
 
 
&zTestFitting.test_lagfitN)r"r#r$rrrrrrsrc@s$eZdZddZddZddZdS)Ú TestCompanioncCs"tttjgtttjdgdSr)rrZrÚlagcompanionrrrrÚtest_raisesÉszTestCompanion.test_raisescCs<tddD],}dg|dg}tt |¡j||fkq
dS)Nr    r*r)r/rrrrU)rr5rrrrÚtest_dimensionsÍszTestCompanion.test_dimensionscCstt ddg¡ddkdS)Nr    r)rrgø?)rrrrrrrÚtest_linear_rootÒszTestCompanion.test_linear_rootN)r"r#r$rrrrrrrrÇsrc@seZdZddZdS)Ú    TestGausscCs|t d¡\}}t |d¡}t |j||¡}dt | ¡¡}|dddf||}t|t     d¡d}t| 
¡|dS)Nr(écr    rN)rZlaggaussrr0rrÚsqrtZdiagonalrZeyeÚsum)rrrrÚvvZvdr8rrrÚtest_100ØszTestGauss.test_100N)r"r#r$r¢rrrrrÖsrc@sDeZdZddZddZddZddZd    d
ZddZd dZ    dS)ÚTestMiscc    Cs¤t g¡}tt|dgtddD]z}t t tjdd|d¡ddd¡}t |¡}t     ||¡}d}t
t||dktt |¡ddt||q$dS)Nr    r*rrr
) rÚlagfromrootsrrr/r0ÚcosrIÚpir=rr?ry)rr9r5Úrootsr{r8rrrÚtest_lagfromrootsës
*
zTestMisc.test_lagfromrootscCsltt dg¡gtt ddg¡dgtddD]4}t dd|¡}t t |¡¡}tt|t|q2dS)Nr    rrr*r')rrZlagrootsr/r0rIr¤r)rr5r8r9rrrÚ test_lagroots÷szTestMisc.test_lagrootscCsfddddg}tttj|dtt |¡|ddtt |d¡|ddtt |d¡dgdS)Nrr
r    rr&)rrZrrr)rrrrrÚtest_lagtrimÿs
zTestMisc.test_lagtrimcCstt dd¡ddgdS)Nr'r~rRr)rrZlaglinerrrrÚtest_lagline
szTestMisc.test_laglinecCs2tdD]$}tt dg|dg¡t|qdS©NrRrr    )r/rrryrT©rr5rrrÚ test_lag2poly szTestMisc.test_lag2polycCs2tdD]$}tt t|¡dg|dgqdSr¬)r/rrrxrTrrrrÚ test_poly2lagszTestMisc.test_poly2lagcCs2t ddd¡}t |¡}t |¡}t||dS)Nré
é)r0rIÚexprZ    lagweightr)rrr8r9rrrÚtest_weights
zTestMisc.test_weightN)
r"r#r$r¨r©rªr«r®r¯r³rrrrr£ésr£)$Ú__doc__Ú    functoolsrÚnumpyr0Znumpy.polynomial.laguerreZ
polynomialZlaguerrerZnumpy.polynomial.polynomialrZ numpy.testingrrrrrFZL0ZL1ZL2ZL3ZL4ZL5ZL6rTrrr%rJrmrrrrrr£rrrrÚ<module>s0B^f(0>