fume-manage-python.git

U  
P±d¼Jã@sÐdZddlmZddlZddlmmZddl    m
Z
ddlmZm Z mZmZe dg¡Ze ddg¡Ze ddd    g¡Ze dd
ddg¡Ze ddd ddg¡Ze ddddddg¡Ze dddddddg¡Ze ddddddddg¡Ze dddddddddg    ¡Ze dddd dd!dd"dd#g
¡Zeeeeeeeeeeg
Zd$d%ZGd&d'd'ZGd(d)d)ZGd*d+d+ZGd,d-d-Z Gd.d/d/Z!Gd0d1d1Z"Gd2d3d3Z#Gd4d5d5Z$Gd6d7d7Z%Gd8d9d9Z&dS):zTests for hermite module.
 
é)ÚreduceN©Úpolyval)Úassert_almost_equalÚ assert_raisesÚassert_equalÚassert_éééþÿÿÿéiôÿÿÿééiÐÿÿÿééxi`ÿÿÿé iÿÿÿiÐi þÿÿé@ipùÿÿi  iÀúÿÿéiiËÿÿi4iòÿÿéi viÅþÿi½iÜÿÿicCstj|ddS)Ngíµ ÷Æ°>)Ztol)ÚhermÚhermtrim©Úx©rúZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_hermite.pyÚtrimsrc@s,eZdZddZddZddZddZd    S)
Ú TestConstantscCsttjddgdS)Néÿÿÿÿr    )rrZ
hermdomain©ÚselfrrrÚtest_hermdomain!szTestConstants.test_hermdomaincCsttjdgdS)Nr)rrZhermzerorrrrÚ test_hermzero$szTestConstants.test_hermzerocCsttjdgdS©Nr    )rrZhermonerrrrÚtest_hermone'szTestConstants.test_hermonecCsttjddgdS)Nrçà?)rrZhermxrrrrÚ
test_hermx*szTestConstants.test_hermxN)Ú__name__Ú
__module__Ú__qualname__r r!r#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Úhermaddrr©rÚiÚjÚmsgÚtgtÚresrrrÚtest_hermadd1s$zTestArithmetic.test_hermaddcCstdD]}tdD]}d|d|}t t||d¡}||d7<||d8<t dg|dgdg|dg¡}tt|t||dqqdSr-)r3r4r5r6rZhermsubrrr8rrrÚtest_hermsub;s$zTestArithmetic.test_hermsubcCsvtt dg¡dgtt dg¡ddgtddD]<}dg|dg}dg|d|ddg}tt |¡|q4dS)Nrr    r$r.)rrZhermmulxr3)rr9Zserr<rrrÚ test_hermmulxEszTestArithmetic.test_hermmulxc
Cs²tdD]¤}dg|dg}t |j|¡}tdD]x}d|d|}dg|dg}t |j|¡}t ||¡}t |j|¡}    tt|||dk|t|    |||dq2qdS)Nr.rr    r/r0r1)r3rÚhermvalrÚhermmulrÚlenr)
rr9Zpol1Zval1r:r;Zpol2Zval2Zpol3Zval3rrrÚtest_hermmulMszTestArithmetic.test_hermmulc
CstdD]}tdD]z}d|d|}dg|dg}dg|dg}t ||¡}t ||¡\}}t t ||¡|¡}    tt|    t||dqqdS)Nr.r/r0rr    r1)r3rr7ZhermdivrBrr)
rr9r:r;ÚciÚcjr<ZquoÚremr=rrrÚtest_hermdiv[szTestArithmetic.test_hermdivcCs|tdD]n}tdD]`}d|d|}t |d¡}ttj|g|t dg¡}t ||¡}tt    |t    ||dqqdS)Nr.r/r0r    r1)
r3r4ÚarangerrrBÚarrayZhermpowrr)rr9r:r;Úcr<r=rrrÚtest_hermpowfszTestArithmetic.test_hermpowN)r&r'r(r4Úlinspacerr>r?r@rDrHrLrrrrr).s
 
r)c@seZdZe dddg¡Ze dee¡Ze deee¡Zej         d¡ddZ
ee
dd    d
gZddZ d dZddZddZddZdS)ÚTestEvaluationg@gð?gè?úi,j->ijú
i,j,k->ijk©r+r.r
r    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    rrcsg|]}t|qSrr©Ú.0rKrrrÚ
<listcomp>sz/TestEvaluation.test_hermval.<locals>.<listcomp>é
r/r1r+r
)rrrAÚsizer4rMÚHlistr3rr5Úshape)rÚyr9r;r<r=ZdimsrrrÚtest_hermvalzs
 
 
zTestEvaluation.test_hermvalc
Cs|j\}}}|j\}}}tttj||dd|j||}t |||j¡}t||t     d¡}    t |    |    |j¡}t
|jdkdS©Nr
©r
r+)rrYrÚ
ValueErrorrÚ    hermval2dÚc2drr4ÚonesrrX©
rÚx1Úx2Úx3Úy1Úy2Zy3r<r=ÚzrrrÚtest_hermval2ds
 
zTestEvaluation.test_hermval2dc
Cs|j\}}}|j\}}}tttj|||dd|j|||}t ||||j¡}t||t     d¡}    t |    |    |    |j¡}t
|jdkdSr[)rrYrr]rÚ    hermval3dÚc3drr4r`rrXrarrrÚtest_hermval3d s
 
zTestEvaluation.test_hermval3dc
Csl|j\}}}|j\}}}t d||¡}t |||j¡}t||t d¡}    t |    |    |j¡}t    |j
dkdS)NrOr\)r
r+r
r+)rrYr4ÚeinsumrZ
hermgrid2dr_rr`rrXrarrrÚtest_hermgrid2d±s
 
zTestEvaluation.test_hermgrid2dc
Csr|j\}}}|j\}}}t d|||¡}t ||||j¡}t||t d¡}    t |    |    |    |j¡}t    |j
dkdS)NrPr\)r
r+r
r+r
r+)rrYr4rlrZ
hermgrid3drjrr`rrXrarrrÚtest_hermgrid3d¿s
 
zTestEvaluation.test_hermgrid3dN)r&r'r(r4rJZc1drlr_rjÚrandomrrrYrZrhrkrmrnrrrrrNpsrNc@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)Nrr$rr    )Úlbnd)Úscl©Úaxisr
r.)ÚmÚk)rurvrq)rurvrr©ru)rÚ    TypeErrorrÚhermintr]r3rÚ    poly2hermÚ    herm2polyrrAÚlist)
rr9rvr=rrÚpolr<Zhermpolryr:rrrÚtest_hermintÐs
 
 
 
 
zTestIntegral.test_hermintcCstj d¡}t dd|jD¡j}tj|dd}t||t dd|D¡}tj|dd}t||t dd|D¡}tj|d    dd
}t||dS)N©r+rcSsg|]}t |¡qSr©rryrRrrrrT'sz2TestIntegral.test_hermint_axis.<locals>.<listcomp>rrscSsg|]}t |¡qSrrrRrrrrT+sr    cSsg|]}tj|ddqS)r+)rvrrRrrrrT/sr+)rvrt)r4roÚvstackÚTrryr©rr_r<r=rrrÚtest_hermint_axis#s
 
zTestIntegral.test_hermint_axisN)r&r'r(r~rrrrrrpÎsSrpc@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)    Nrr$rr.r    rwr
)rurr)
rrxrÚhermderr]r3rrryr)rr9r<r=r:rrrÚtest_hermder6s zTestDerivative.test_hermdercCsltj d¡}t dd|jD¡j}tj|dd}t||t dd|D¡}tj|dd}t||dS)NrcSsg|]}t |¡qSr©rrrRrrrrTSsz4TestDerivative.test_hermder_axis.<locals>.<listcomp>rrscSsg|]}t |¡qSrrrRrrrrTWsr    )r4rorrrrrrrrrÚtest_hermder_axisOs
z TestDerivative.test_hermder_axisN)r&r'r(rrrrrrr4src@s8eZdZej d¡ddZddZddZdd    Zd
S)Ú
TestVanderrQr
r    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+rrrr    .r
r.é)r+r
r)
r4rIrÚ
hermvanderrrXr3rrArJ)rrÚvr9ÚcoefrrrÚtest_hermvander`s
zTestVander.test_hermvandercCsx|j\}}}tj d¡}t ||ddg¡}t |||¡}t ||j¡}t||t |g|gddg¡}t    |j
dkdS)Nr\r    r
)r    r.r)rr4rorZhermvander2dr^ÚdotÚflatrrrX©rrbrcrdrKZvanr<r=rrrÚtest_hermvander2dqs
zTestVander.test_hermvander2dcCs|j\}}}tj d¡}t |||dddg¡}t ||||¡}t ||j¡}t||t |g|g|gdddg¡}t    |j
dkdS)N)r
r+rr    r
r+)r    r.é)rr4rorZhermvander3drirrrrrXrrrrÚtest_hermvander3d~s
zTestVander.test_hermvander3dN)    r&r'r(r4rorrrrrrrrr\s rc@seZdZddZdS)ÚTestFittingc    Cs&dd}dd}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggd    tttjdgdgdddgd    tttjdgdgdgtttjdgdgddd
gtttjdgdggt dd¡}||}t ||d¡}tt|dt    t 
||¡|t ||ddddg¡}tt|dt    t 
||¡|t ||d¡}tt|d t    t 
||¡|t ||dddddg¡}tt|d t    t 
||¡|t ||dddddg¡}tt|d t    t 
||¡|t |t ||g¡jd¡}t    |t ||g¡jt |t ||g¡jddddg¡}t    |t ||g¡jt  |¡}| ¡}    d|ddd<d|ddd<tj||    d|d    }
t    |
|tj||    ddddg|d    }
t    |
|tj|t |    |    g¡jd|d    }t    |t ||g¡jtj|t |    |    g¡jddddg|d    }t    |t ||g¡jddddg}t    t ||d¡ddgt    t ||ddg¡ddgt dd¡}||}t ||d¡}t    t 
||¡|t ||dddg¡} t    t 
|| ¡|t    || dS)NcSs||d|dS)Nr    r
rrrrrÚfsz#TestFitting.test_hermfit.<locals>.fcSs|d|ddS)Nrr
r    rrrrrÚf2sz$TestFitting.test_hermfit.<locals>.f2r    rrr
)Úwrr+rr.yð?yð¿r$)rr]rZhermfitrxr4rMrrCrrArJrZ
zeros_likeÚcopy)rrrrrYZcoef3Zcoef4Zcoef2drZywZwcoef3Zwcoef2dZcoef1Zcoef2rrrÚtest_hermfitsp"
 
 
&zTestFitting.test_hermfitN)r&r'r(rrrrrrsrc@s$eZdZddZddZddZdS)Ú TestCompanioncCs"tttjgtttjdgdSr")rr]rÚ hermcompanionrrrrÚtest_raisesÛszTestCompanion.test_raisescCs<tddD],}dg|dg}tt |¡j||fkq
dS)Nr    r.r)r3rrrrX)rr9rrrrÚtest_dimensionsßszTestCompanion.test_dimensionscCstt ddg¡ddkdS)Nr    r
)rrgÐ¿)rrrrrrrÚtest_linear_rootäszTestCompanion.test_linear_rootN)r&r'r(rrr rrrrrÙsrc@seZdZddZdS)Ú    TestGausscCst d¡\}}t |d¡}t |j||¡}dt | ¡¡}|dddf||}t|t     d¡t tj
¡}t| ¡|dS)Nr,écr    )rZ    hermgaussrr4rrÚsqrtZdiagonalrZeyeÚpiÚsum)rrrrÚvvZvdr<rrrÚtest_100êszTestGauss.test_100N)r&r'r(r§rrrrr¡èsr¡c@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.rr
r) rÚ hermfromrootsrrr3r4ÚcosrMr¤rArrCr{)rr=r9Úrootsr}r<rrrÚtest_hermfromrootsýs
*
zTestMisc.test_hermfromrootscCsltt dg¡gtt ddg¡dgtddD]4}t dd|¡}t t |¡¡}tt|t|q2dS)Nr    gà¿r
r.r)rrZ    hermrootsr3r4rMr©r)rr9r<r=rrrÚtest_hermroots    szTestMisc.test_hermrootscCsfddddg}tttj|dtt |¡|ddtt |d¡|ddtt |d¡dgdS)Nr
rr    rr*)rr]rrr)rrrrrÚ test_hermtrims
zTestMisc.test_hermtrimcCstt dd¡ddgdS)Nr+rr
)rrZhermlinerrrrÚ test_hermlineszTestMisc.test_hermlinecCs2tdD]$}tt dg|dg¡t|qdS©NrUrr    )r3rrr{rW©rr9rrrÚtest_herm2polyszTestMisc.test_herm2polycCs2tdD]$}tt t|¡dg|dgqdSr°)r3rrrzrWr±rrrÚtest_poly2herm#szTestMisc.test_poly2hermcCs6t ddd¡}t |d¡}t |¡}t||dS)Néûÿÿÿr.ér
)r4rMÚexprZ
hermweightr)rrr<r=rrrÚtest_weight's
zTestMisc.test_weightN)
r&r'r(r¬rr®r¯r²r³r·rrrrr¨ûsr¨)'Ú__doc__Ú    functoolsrÚnumpyr4Znumpy.polynomial.hermiteZ
polynomialZhermiterZnumpy.polynomial.polynomialrZ numpy.testingrrrrrJZH0ZH1ZH2ZH3ZH4ZH5ZH6ZH7ZH8ZH9rWrrr)rNrprrrrr¡r¨rrrrÚ<module>s6B^f(0M