U
|
P ±d L ã @ sÐ d Z ddlmZ ddlZddlm mZ ddl m
|
Z
|
ddl m Z m Z mZmZ e dg¡Ze ddg¡Ze dddg¡Ze ddddg¡Ze d dd
|
ddg¡Ze dd dd ddg¡Ze d dddd ddg¡Ze ddddddddg¡Ze dddddddddg ¡Ze ddddddddddg
|
¡Zeeeeeeeeeeg
|
Zdd Z G dd d Z G d d d Z G d d d Z G d!d" d"Z G d#d$ d$Z!G d%d& d&Z"G d'd( d(Z#G d)d* d*Z$G d+d, d,Z%G d-d. d.Z&dS )/z Tests for hermite_e module.
|
|
é )ÚreduceN©Úpolyval)Úassert_almost_equalÚ assert_raisesÚ assert_equalÚassert_é éÿÿÿÿéýÿÿÿé iúÿÿÿé iöÿÿÿiñÿÿÿé- iÿÿÿéi iëÿÿÿi\þÿÿéÒ iäÿÿÿi± iûÿÿiz iÜÿÿÿc C s t j| ddS )Ngíµ ÷Æ°>)Ztol)ÚhermeÚ hermetrim©Úx© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_hermite_e.pyÚtrim s r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestConstantsc C s t tjddg d S )Nr
|
r )r r Z hermedomain©Úselfr r r Útest_hermedomain! s z TestConstants.test_hermedomainc C s t tjdg d S )Nr )r r Z hermezeror r r r Útest_hermezero$ s z TestConstants.test_hermezeroc C s t tjdg d S ©Nr )r r Zhermeoner r r r Ú test_hermeone' s zTestConstants.test_hermeonec C s t tjddg d S )Nr r )r r Zhermexr r r r Ú test_hermex* s zTestConstants.test_hermexN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r r r s r c @ sJ e Zd Ze ddd¡Zdd Zdd Zdd Zd
|
d Z d d Z
|
dd Z dS )ÚTestArithmeticr r éd c C s t dD ]}t dD ]}d| d| }t t||d ¡}|| d7 < || d7 < t dg| dg dg| dg ¡}tt|t||d qqd S ©Né úAt i=ú, j=r r ©Úerr_msg)ÚrangeÚnpÚzerosÚmaxr Úhermeaddr r ©r ÚiÚjÚmsgÚtgtÚresr r r Ú test_hermeadd1 s $z TestArithmetic.test_hermeaddc C s t dD ]}t dD ]}d| d| }t t||d ¡}|| d7 < || d8 < t dg| dg dg| dg ¡}tt|t||d qqd S r% )r+ r, r- r. r Zhermesubr r r0 r r r Ú test_hermesub; s $z TestArithmetic.test_hermesubc C sv t t dg¡dg t t dg¡ddg tddD ]<}dg| dg }dg|d |ddg }t t |¡| q4d S )Nr r r& )r r Z hermemulxr+ )r r1 Zserr4 r r r Útest_hermemulxE s z TestArithmetic.test_hermemulxc
|
C s² t dD ]¤}dg| dg }t | j|¡}t dD ]x}d| d| }dg| dg }t | j|¡}t ||¡}t | j|¡} tt||| d k| t| || |d q2qd S )Nr& r r r' r( r) )r+ r Úhermevalr Úhermemulr Úlenr )
|
r r1 Zpol1Zval1r2 r3 Zpol2Zval2Zpol3Zval3r r r Ú test_hermemulM s z TestArithmetic.test_hermemulc
|
C s t dD ]}t dD ]z}d| d| }dg| dg }dg| dg }t ||¡}t ||¡\}}t t ||¡|¡} tt| t||d qqd S )Nr& r' r( r r r) )r+ r r/ Zhermedivr: r r )
|
r r1 r2 r3 ÚciÚcjr4 ZquoÚremr5 r r r Ú test_hermediv[ s z TestArithmetic.test_hermedivc C s| t dD ]n}t dD ]`}d| d| }t |d ¡}ttj|g| t dg¡}t ||¡}tt |t ||d qqd S )Nr& r' r( r r) )
|
r+ r, Úaranger r r: ÚarrayZhermepowr r )r r1 r2 r3 Úcr4 r5 r r r Ú test_hermepowf s z TestArithmetic.test_hermepowN) r r! r" r, Úlinspacer r6 r7 r8 r< r@ rD r r r r r# . s
|
|
r# c @ s e Zd Ze dddg¡Ze dee¡Ze deee¡Zej d¡d d Z
|
e e
|
d ddgZ d
|
d Z d d Zdd Zdd Zdd ZdS )ÚTestEvaluationg @g @g @úi,j->ijú
|
i,j,k->ijk©r r& é r g ð?c sè t t g dg¡jd t dd¡ fddtD }tdD ]<}d| }|| }t dg| dg ¡}t|||d q<td D ]`}d
|
g| }t |¡ t t dg¡j
|
| t t ddg¡j
|
| t t dddg¡j
|
| qd S ) Nr r r
|
c s g | ]}t |qS r r ©Ú.0rC r r r Ú
|
<listcomp> s z0TestEvaluation.test_hermeval.<locals>.<listcomp>é
|
r' r) r rJ ) r r r9 Úsizer, rE ÚHelistr+ r r- Úshape)r Úyr1 r3 r4 r5 Zdimsr r r Ú test_hermevalz s
|
|
|
z TestEvaluation.test_hermevalc
|
C s | j \}}}| j\}}}tttj||d d
| j || }t ||| j¡}t|| t d¡} t | | | j¡}t
|
|j dk d S ©NrJ ©rJ r ) r rR r Ú
|
ValueErrorr Ú
|
hermeval2dÚc2dr r, Úonesr rQ ©
|
r Úx1Úx2Úx3Úy1Úy2Zy3r4 r5 Úzr r r Útest_hermeval2d s
|
|
z TestEvaluation.test_hermeval2dc
|
C s | j \}}}| j\}}}tttj|||d d
| j || | }t |||| j¡}t|| t d¡} t | | | | j¡}t
|
|j dk d S rT ) r rR r rV r Ú
|
hermeval3dÚc3dr r, rY r rQ rZ r r r Útest_hermeval3d s
|
|
z TestEvaluation.test_hermeval3dc
|
C sl | j \}}}| j\}}}t d||¡}t ||| j¡}t|| t d¡} t | | | j¡}t |j
|
dk d S )NrG rU )rJ r rJ r ) r rR r, Úeinsumr Z hermegrid2drX r rY r rQ rZ r r r Útest_hermegrid2d± s
|
|
z TestEvaluation.test_hermegrid2dc
|
C sr | j \}}}| j\}}}t d|||¡}t |||| j¡}t|| t d¡} t | | | | j¡}t |j
|
dk d S )NrH rU )rJ r rJ r rJ r ) r rR r, re r Z hermegrid3drc r rY r rQ rZ r r r Útest_hermegrid3d¿ s
|
|
z TestEvaluation.test_hermegrid3dN)r r! r" r, rB Zc1dre rX rc Úrandomr r rR rS ra rd rf rg r r r r rF p s rF c @ s e Zd Zdd Zdd ZdS )Ú TestIntegralc
|
C s2 t ttjdgd t ttjdgd t ttjdgdddg t ttjdgdgd t ttjdgdgd t ttjdgdd tdd D ]8}dg|d dg }tjdg||d
|
}t|ddg qtd 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 }|d d
}t| D ]}tj|dd }qJtj|| d }tt |t | q"qtd D ]}tdd D ]n} dg| dg }|d d
}t| D ]}tj|d|gd
|
}qÆtj|| t t| d
|
}tt |t | qqtd D ]}tdd D ]r} dg| dg }|d d
}t| D ]}tj|d|gdd }qPtj|| t t| dd }tt |t | q(qtd D ]}tdd D ]r} dg| dg }|d d
}t| D ]}tj|d|gdd }qÞtj|| t t| dd }tt |t | q¶q¨d S )Nr ç à?r
|
r )Úlbnd)Úscl©ÚaxisrJ r& )ÚmÚk)ro rp rk )ro rp rl ©ro ) r Ú TypeErrorr ÚhermeintrV r+ r Ú
|
poly2hermeÚ
|
herme2polyr r9 Úlist)
|
r r1 rp r5 rl Úpolr4 Zhermepolrs r2 r r r Ú test_hermeintÐ s
|
|
|
|
|
zTestIntegral.test_hermeintc C s t j d¡}t dd |jD ¡j}tj|dd}t|| t dd |D ¡}tj|dd}t|| t dd |D ¡}tj|d dd
|
}t|| d S ) N©r é c S s g | ]}t |¡qS r ©r rs rK r r r rM ' s z3TestIntegral.test_hermeint_axis.<locals>.<listcomp>r rm c S s g | ]}t |¡qS r r{ rK r r r rM + s r c S s g | ]}t j|d dqS )r )rp r{ rK r r r rM / s r )rp rn )r, rh ÚvstackÚTr rs r ©r rX r4 r5 r r r Útest_hermeint_axis# s
|
|
z TestIntegral.test_hermeint_axisN)r r! r" rx r r r r r ri Î s Sri c @ s e Zd Zdd Zdd ZdS )ÚTestDerivativec C s t ttjdgd t ttjdgd tdD ]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Âd S ) Nr rj r
|
r& r rq rJ )ro rl )
|
r rr r ÚhermederrV r+ r r rs r )r r1 r4 r5 r2 r r r Ú test_hermeder6 s( ÿz TestDerivative.test_hermederc C sl t j d¡}t dd |jD ¡j}tj|dd}t|| t dd |D ¡}tj|dd}t|| d S )Nry c S s g | ]}t |¡qS r ©r r rK r r r rM T s z5TestDerivative.test_hermeder_axis.<locals>.<listcomp>r rm c S s g | ]}t |¡qS r r rK r r r rM X s r )r, rh r| r} r r r r~ r r r Útest_hermeder_axisP s
|
z!TestDerivative.test_hermeder_axisN)r r! r" r r r r r r r 4 s r c @ s8 e Zd Zej d¡d d Zdd Zdd Zdd Zd
|
S ) Ú
|
TestVanderrI rJ r c C sÎ t d¡}t |d¡}t|jdk tdD ].}dg| dg }t|d|f t ||¡ q,t ddgddgdd gg¡}t |d¡}t|jd
|
k tdD ].}dg| dg }t|d|f t ||¡ qd S ) Nr ry rz r r .rJ r& é )r rJ rz )
|
r, rA r Ú hermevanderr rQ r+ r r9 rB )r r Úvr1 Úcoefr r r Útest_hermevandera s
|
zTestVander.test_hermevanderc C sx | j \}}}tj d¡}t ||ddg¡}t |||¡}t ||j¡}t|| t |g|gddg¡}t |j
|
dk d S )NrU r rJ )r r& r ) r r, rh r Z hermevander2drW ÚdotÚflatr r rQ ©r r[ r\ r] rC Zvanr4 r5 r r r Útest_hermevander2dr s
|
z TestVander.test_hermevander2dc C s | j \}}}tj d¡}t |||dddg¡}t ||||¡}t ||j¡}t|| t |g|g|gdddg¡}t |j
|
dk d S )N)rJ r rz r rJ r )r r& é ) r r, rh r Z hermevander3drb r r r r rQ r r r r Útest_hermevander3d s
|
z TestVander.test_hermevander3dN) r r! r" r, rh r r r r r r r r r
] s r
c @ s e Zd Zdd ZdS )Ú TestFittingc C s& dd }dd }t ttjdgdgd t ttjdggdgd t ttjg dgd t ttjdgdgggd t ttjddgdgd t ttjdgddgd t ttjdgdgddggd t ttjdgdgdddgd t ttjdgdgdg t ttjdgdgddd
|
g t ttjdgdgg t dd¡}||}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 ||g¡j d ¡}t |t ||g¡j t |t ||g¡j dddd g¡}t |t ||g¡j t |¡}| ¡ } d|dd d
< d|dd d
< tj|| d |d }
|
t |
|
| tj|| dddd g|d }
|
t |
|
| tj|t | | g¡j d |d } t | t ||g¡j tj|t | | g¡j dddd g|d } t | t ||g¡j ddddg}t t ||d¡ddg t t ||ddg¡ddg t dd¡}||}t ||d ¡} t t
|
|| ¡| t ||ddd g¡} t t
|
|| ¡| t | | d S )Nc S s | | d | d S )Nr rJ r r r r r Úf s z$TestFitting.test_hermefit.<locals>.fc S s | d | d d S )Nrz rJ r r r r r r Úf2 s z%TestFitting.test_hermefit.<locals>.f2r r
|
r rJ )Úwr r rz r& y ð?y ð¿)r rV r Zhermefitrr r, rE r r; r r9 rB r} Z
|
zeros_likeÚcopy)r r r r rR Zcoef3Zcoef4Zcoef2dr ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r Ú test_hermefit sp "
|
|
|
& zTestFitting.test_hermefitN)r r! r" r r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestCompanionc C s" t ttjg t ttjdg d S r )r rV r Úhermecompanionr r r r Ú test_raisesÜ s zTestCompanion.test_raisesc C s< t ddD ],}dg| dg }tt |¡j||fk q
|
d S )Nr r& r )r+ r r r rQ )r r1 r r r r Útest_dimensionsà s z TestCompanion.test_dimensionsc C s t t ddg¡d dk d S )Nr rJ )r r ç à¿)r r r r r r r Útest_linear_rootå s z TestCompanion.test_linear_rootN)r r! r" r r r r r r r r Ú s r c @ s e Zd Zdd ZdS )Ú TestGaussc C s t d¡\}}t |d¡}t |j| |¡}dt | ¡ ¡ }|d d
d f | | }t|t d¡ t dtj
|
¡}t| ¡ | d S )Nr$ éc r rJ ) r Z
|
hermegaussr r, r r} ÚsqrtZdiagonalr ZeyeÚpiÚsum)r r r r ÚvvZvdr4 r r r Útest_100ë s zTestGauss.test_100N)r r! r" r£ r r r r r é s r c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestMiscc C s¤ t g ¡}tt|dg tddD ]z}t t tj dd| d ¡dd d
¡}t |¡}t ||¡}d}t
|
t ||d k tt |¡d d t|| q$d S )Nr r& r rJ r
|
) r Úhermefromrootsr r r+ r, ÚcosrE r r9 r r; ru )r r5 r1 Úrootsrw r4 r r r Útest_hermefromrootsþ s
|
*
|
z TestMisc.test_hermefromrootsc C sl t t dg¡g t t ddg¡dg tddD ]4}t dd|¡}t t |¡¡}t t|t| q2d S )Nr r
|
rJ r& )r r Z
|
hermerootsr+ r, rE r¥ r )r r1 r4 r5 r r r Útest_hermeroots
|
s zTestMisc.test_hermerootsc C sf ddddg}t ttj|d tt |¡|d d
tt |d¡|d d
tt |d¡dg d S )NrJ r
|
r r r )r rV r r r )r r r r r Útest_hermetrim s
|
zTestMisc.test_hermetrimc C s t t dd¡ddg d S )Nr rz )r r Z hermeliner r r r Útest_hermeline s zTestMisc.test_hermelinec C s2 t dD ]$}tt dg| dg ¡t| qd S ©NrN r r )r+ r r ru rP ©r r1 r r r Útest_herme2poly s zTestMisc.test_herme2polyc C s2 t dD ]$}tt t| ¡dg| dg qd S r¬ )r+ r r rt rP r r r r Útest_poly2herme$ s zTestMisc.test_poly2hermec C s8 t ddd¡}t d|d ¡}t |¡}t|| d S )Néûÿÿÿr& é r rJ )r, rE Úexpr Z hermeweightr )r r r4 r5 r r r Ú test_weight( s
|
zTestMisc.test_weightN)
|
r r! r" r¨ r© rª r« r® r¯ r³ r r r r r¤ ü s r¤ )'Ú__doc__Ú functoolsr Únumpyr, Znumpy.polynomial.hermite_eZ
|
polynomialZ hermite_er Znumpy.polynomial.polynomialr Z numpy.testingr r r r rB ZHe0ZHe1ZHe2ZHe3ZHe4ZHe5ZHe6ZHe7ZHe8ZHe9rP r r r# rF ri r r
r r r r¤ r r r r Ú<module> s6 B^f)0M
|
