U
|
P ±d)K ã @ sð d Z ddlmZ ddlZddlm mZ ddl m
|
Z
|
ddl m Z m Z mZmZ e dg¡Ze ddg¡Ze dddg¡d Ze dd
|
dd g¡d Ze ddd dd g¡d Ze ddddddg¡d Ze dddddddg¡d Ze ddddddddg¡d Ze d dddd dd dd g ¡d Ze dddd dd!dd"dd#g
|
¡d Zeeeeeeeeeeg
|
Zd$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 d0d1 d1Z"G d2d3 d3Z#G d4d5 d5Z$G d6d7 d7Z%G d8d9 d9Z&dS ):z Tests for legendre module.
|
|
é )ÚreduceN©Úpolyval)Úassert_almost_equalÚ assert_raisesÚ assert_equalÚassert_é éÿÿÿÿé é éýÿÿÿé iâÿÿÿé# é é iºÿÿÿé? éûÿÿÿéi iÅþÿÿéç é iÝÿÿÿi; iKýÿÿi iûÿÿi iÑÿÿi# é iôíÿÿibF itÿÿi{/ c C s t j| ddS )Ngíµ ÷Æ°>)Ztol)ÚlegÚlegtrim©Úx© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_legendre.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 legdomain©Úselfr r r Útest_legdomain! s z TestConstants.test_legdomainc C s t tjdg d S )Nr )r r Zlegzeror r r r Ú test_legzero$ s zTestConstants.test_legzeroc C s t tjdg d S ©Nr )r r Zlegoner r r r Ú test_legone' s zTestConstants.test_legonec C s t tjddg d S )Nr r )r r Zlegxr r r r Ú test_legx* s zTestConstants.test_legxN)Ú__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 ©Nr úAt i=ú, j=r r ©Úerr_msg)ÚrangeÚnpÚzerosÚmaxr Úlegaddr r ©r! ÚiÚjÚmsgÚtgtÚresr r r Ú test_legadd1 s $zTestArithmetic.test_legaddc C s t dD ]}t dD ]}d| d| }t t||d ¡}|| d7 < || d8 < t dg| dg dg| dg ¡}tt|t||d qqd S r, )r1 r2 r3 r4 r Zlegsubr r r6 r r r Ú test_legsub; s $zTestArithmetic.test_legsubc C s t t dg¡dg t t dg¡ddg tddD ]T}d| d }dg| dg }dg|d || d|d | g }t t |¡| q4d S )Nr r r r )r r Zlegmulxr1 )r! r7 ÚtmpZserr: r r r Ú test_legmulxE s $zTestArithmetic.test_legmulxc
|
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/ )r1 r Úlegvalr Úlegmulr Úlenr )
|
r! r7 Zpol1Zval1r8 r9 Zpol2Zval2Zpol3Zval3r r r Ú test_legmulN s zTestArithmetic.test_legmulc
|
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/ )r1 r r5 ZlegdivrA r r )
|
r! r7 r8 r9 ÚciÚcjr: ZquoÚremr; r r r Ú test_legdiv\ s zTestArithmetic.test_legdivc 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/ )
|
r1 r2 Úaranger r rA ÚarrayZlegpowr r )r! r7 r8 r9 Úcr: r; r r r Ú test_legpowg s zTestArithmetic.test_legpowN) r' r( r) r2 Úlinspacer r< r= r? rC rG rK r r r r r* . s
|
|
r* c @ s e Zd Ze dddg¡Ze dee¡Ze deee¡Zej d¡d d Z
|
e e
|
dddgZ d d
|
Z d d Zd d Zdd Zdd ZdS )ÚTestEvaluationç @úi,j->ijú
|
i,j,k->ijk©r 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 ©Ú.0rJ r r r Ú
|
<listcomp> s z.TestEvaluation.test_legval.<locals>.<listcomp>é
|
r- r/ r r ) r r r@ Úsizer2 rL ÚLlistr1 r r3 Úshape)r! Úyr7 r9 r: r; Zdimsr r r Ú test_legval{ s
|
|
|
zTestEvaluation.test_legvalc
|
C s | j \}}}| j\}}}tttj||d d
| j || }t ||| j¡}t|| t d¡} t | | | j¡}t
|
|j dk d S ©Nr ©r r ) r rZ r Ú
|
ValueErrorr Úlegval2dÚc2dr r2 Úonesr rY ©
|
r! Úx1Úx2Úx3Úy1Úy2Zy3r: r; Úzr r r Ú test_legval2d s
|
|
z TestEvaluation.test_legval2dc
|
C s | j \}}}| j\}}}tttj|||d d
| j || | }t |||| j¡}t|| t d¡} t | | | | j¡}t
|
|j dk d S r\ ) r rZ r r^ r Úlegval3dÚc3dr r2 ra r rY rb r r r Ú test_legval3d¡ s
|
|
z TestEvaluation.test_legval3dc
|
C sl | j \}}}| j\}}}t d||¡}t ||| j¡}t|| t d¡} t | | | j¡}t |j
|
dk d S )NrO r] )r r r r ) r rZ r2 Úeinsumr Z leggrid2dr` r ra r rY rb r r r Útest_leggrid2d² s
|
|
z TestEvaluation.test_leggrid2dc
|
C sr | j \}}}| j\}}}t d|||¡}t |||| j¡}t|| t d¡} t | | | | j¡}t |j
|
dk d S )NrP r] )r r r r r r ) r rZ r2 rm r Z leggrid3drk r ra r rY rb r r r Útest_leggrid3dÀ s
|
|
z TestEvaluation.test_leggrid3dN)r' r( r) r2 rI Zc1drm r` rk Úrandomr r rZ r[ ri rl rn ro r r r r rM q s rM c @ s$ e Zd Zdd 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©Úaxisr r )ÚmÚk)rw rx rs )rw rx rt ©rw ) r Ú TypeErrorr Úlegintr^ r1 r Úpoly2legÚleg2polyr r@ Úlist)
|
r! r7 rx r; rt Úpolr: Zlegpolr{ r8 r r r Ú test_legintÑ s
|
|
|
|
|
zTestIntegral.test_legintc 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 r{ rS r r r rU ( s z1TestIntegral.test_legint_axis.<locals>.<listcomp>r ru c S s g | ]}t |¡qS r r rS r r r rU , s r c S s g | ]}t j|d dqS )r )rx r rS r r r rU 0 s r )rx rv )r2 rp ÚvstackÚTr r{ r ©r! r` r: r; r r r Útest_legint_axis$ s
|
|
z TestIntegral.test_legint_axisc C s t t dd¡d d S )N©r r r r )r r r{ r r r r Útest_legint_zerointord4 s z#TestIntegral.test_legint_zerointordN)r' r( r) r r r r r r r rq Ï s Srq c @ s$ e Zd Zdd 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 rr r
|
r r ry r )rw rt )
|
r rz r Úlegderr^ r1 r r r{ r )r! r7 r: r; r8 r r r Ú test_legder: s zTestDerivative.test_legderc C sl t j d¡}t dd |jD ¡j}tj|dd}t|| t dd |D ¡}tj|dd}t|| d S )Nr c S s g | ]}t |¡qS r ©r r rS r r r rU W s z3TestDerivative.test_legder_axis.<locals>.<listcomp>r ru c S s g | ]}t |¡qS r r rS r r r rU [ s r )r2 rp r r
r r r r r r r Útest_legder_axisS s
|
z TestDerivative.test_legder_axisc C s d}t t |d¡dg d S )N)r r r r r r )r r r )r! rJ r r r Ú test_legder_orderhigherthancoeff_ s z/TestDerivative.test_legder_orderhigherthancoeffN)r' r( r) r r r r r r r r 8 s r c @ s@ e Zd Zej d¡d d Zdd Zdd Zdd Zd
|
d Z d S ) Ú
|
TestVanderrQ r 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 r r r r .r r é )r r r )
|
r2 rH r Ú legvanderr rY r1 r r@ rI )r! r Úvr7 Úcoefr r r Útest_legvanderg s
|
zTestVander.test_legvanderc C sx | j \}}}tj d¡}t ||ddg¡}t |||¡}t ||j¡}t|| t |g|gddg¡}t |j
|
dk d S )Nr] r r )r r r ) r r2 rp r Z legvander2dr_ ÚdotÚflatr r rY ©r! rc rd re rJ Zvanr: r; r r r Útest_legvander2dx s
|
zTestVander.test_legvander2dc C s | j \}}}tj d¡}t |||dddg¡}t ||||¡}t ||j¡}t|| t |g|g|gdddg¡}t |j
|
dk d S )N)r r r r r r )r r é ) r r2 rp r Z legvander3drj r r r r rY r r r r Útest_legvander3d
s
|
zTestVander.test_legvander3dc C s t ttjdd d S )Nr r
|
)r r^ r r r r r r Útest_legvander_negdeg s z TestVander.test_legvander_negdegN)
|
r' r( r) r2 rp r r r r r r r r r r c 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 r r r r r r Úf s z"TestFitting.test_legfit.<locals>.fc S s | d | d d S )Nr r r r r r r r Úf2 s z#TestFitting.test_legfit.<locals>.f2r r
|
r r )Úwr r r r y ð?y ð¿)r r^ r Zlegfitrz r2 rL r rB r r@ rI r
Z
|
zeros_likeÚcopy)r! r r r rZ Zcoef3Zcoef4Zcoef2dr ZywZwcoef3Zwcoef2dZcoef1Zcoef2r r r Ú test_legfit sp "
|
|
|
& zTestFitting.test_legfitN)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 r^ r Ú legcompanionr r r r Ú test_raiseså s zTestCompanion.test_raisesc C s< t ddD ],}dg| dg }tt |¡j||fk q
|
d S )Nr r r )r1 r r r¤ rY )r! r7 r r r r Útest_dimensionsé s z TestCompanion.test_dimensionsc C s t t ddg¡d dk d S )Nr r )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¡ d}t|
|
¡ | d S )Nr+ éc r rN ) r Zleggaussr r2 r r
ÚsqrtZdiagonalr ZeyeÚsum)r! r r r ÚvvZvdr: r r r Útest_100ô s zTestGauss.test_100N)r' r( r) r® r r r r r© ò s r© c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd 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 r r
|
) r Ú legfromrootsr r r1 r2 ÚcosrL Úpir@ r rB r} )r! r; r7 Úrootsr r: r r r Útest_legfromroots s
|
*
|
zTestMisc.test_legfromrootsc C sl t t dg¡g t t ddg¡dg tddD ]4}t dd|¡}t t |¡¡}t t|t| q2d S )Nr r r§ r r
|
)r r Zlegrootsr1 r2 rL r° r )r! r7 r: r; r r r Ú test_legroots s zTestMisc.test_legrootsc C sf ddddg}t ttj|d tt |¡|d d
tt |d¡|d d
tt |d¡dg d S )Nr r
|
r r r )r r^ r r r )r! r r r r Ú test_legtrim s
|
zTestMisc.test_legtrimc C s t t dd¡ddg d S )Nr r ©r r Zlegliner r r r Ú test_legline& s zTestMisc.test_leglinec C s t t dd¡dg d S )Nr r r· r r r r Útest_legline_zeroscl) s z TestMisc.test_legline_zerosclc C s2 t dD ]$}tt dg| dg ¡t| qd S ©NrV r r )r1 r r r} rX ©r! r7 r r r Ú test_leg2poly, s zTestMisc.test_leg2polyc C s2 t dD ]$}tt t| ¡dg| dg qd S rº )r1 r r r| rX r» r r r Ú test_poly2leg0 s zTestMisc.test_poly2legc C s* t ddd¡}d}t |¡}t|| d S )Nr
|
r é rR )r2 rL r Z legweightr )r! r r: r; r r r Ú test_weight4 s
|
zTestMisc.test_weightN) r' r( r) r´ rµ r¶ r¸ r¹ r¼ r½ r¿ r r r r r¯ s r¯ )'Ú__doc__Ú functoolsr Únumpyr2 Znumpy.polynomial.legendreZ
|
polynomialZlegendrer Znumpy.polynomial.polynomialr Z numpy.testingr r r r rI ZL0ZL1ZL2ZL3ZL4ZL5ZL6ZL7ZL8ZL9rX r r r* rM rq r r r r£ r© r¯ r r r r Ú<module> s6 "C^i+3M
|
