U
|
P ±dÓ ã @ sä d Z ddlZddlmZ ddlmZ ddlmZm Z m
|
Z
|
G dd dZ G dd dZ ej d ejd
|
d d gd ded
|
d d gf¡G dd dZG dd dZG dd dZG dd dZdd Zdd Zdd Zd d Zd d ZdS )!zE
|
Tests related to the ``symbol`` attribute of the ABCPolyBase class.
|
é N)Úarray)Ú assert_equalÚ assert_raisesÚassert_c @ sv e Zd ZdZdddgZdd Zej dde fd e fd
|
e
|
fde
|
ff¡d d Z ej d d¡dd Z dd Z dd Zd
|
S )ÚTestInitz5
|
Test polynomial creation with symbol kwarg.
|
é é é c C s t | j¡}t|jd d S )NÚx©ÚpolyÚ
|
PolynomialÚcr Úsymbol©ÚselfÚp© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/polynomial/tests/test_symbol.pyÚtest_default_symbol s z TestInit.test_default_symbol)Ú bad_inputÚ exceptionÚ Ú3Nc C s* t |¡ tj| j|d}W 5 Q R X d S )N©r )ÚpytestÚraisesr r r )r r r r r r r Útest_symbol_bad_input s z TestInit.test_symbol_bad_inputr )r
|
Zx_1ÚAZxyzu βc C s t j| j|d}t|j| dS )zF
|
Values for symbol that should pass input validation.
|
r Nr )r r r r r r Útest_valid_symbols s zTestInit.test_valid_symbolsc C s0 t j| jdd}t t¡ d|_W 5 Q R X dS )z2
|
'symbol' attribute is read only.
|
r
|
r ÚzN)r r r r r ÚAttributeErrorr r r r r Ú test_property- s zTestInit.test_propertyc C s0 t j| jdd}t j|jdd}t|jd d S )NÚyr Út)r r r Zcoefr r )r r Úptr r r Útest_change_symbol5 s zTestInit.test_change_symbol)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r ÚmarkÚ parametrizeÚ
|
ValueErrorÚ TypeErrorr r r" r& r r r r r s
|
ü
|
|
r c @ s@ e Zd ZejdddgddZdd Zdd Zd
|
d Zd d Z dS )ÚTestUnaryOperatorsr r r r r c C s | j }t|jd d S ©Nr ©r r r )r Únr r r Útest_neg? s zTestUnaryOperators.test_negc C s | j d }t|jd d S ©Né
|
r r1 ©r Úoutr r r Útest_scalarmulC s
|
z!TestUnaryOperators.test_scalarmulc C s d| j }t|jd d S r4 r1 r6 r r r Útest_rscalarmulG s
|
z"TestUnaryOperators.test_rscalarmulc C s | j d }t|jd d S )Nr r r1 r6 r r r Útest_powK s
|
zTestUnaryOperators.test_powN)
|
r' r( r) r r r r3 r8 r9 r: r r r r r/ < s
|
r/ Úrhsé é é r r c @ sd e Zd ZdZejdddgddZdd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dd Z dd ZdS )Ú TestBinaryOperatorsSameSymbolzc
|
Ensure symbol is preserved for numeric operations on polynomials with
|
the same symbol
|
r r r r r c C s | j | }t|jd d S r0 r1 ©r r; r7 r r r Útest_add^ s
|
z&TestBinaryOperatorsSameSymbol.test_addc C s | j | }t|jd d S r0 r1 r@ r r r Útest_subb s
|
z&TestBinaryOperatorsSameSymbol.test_subc C s | j | }t|jd d S r0 r1 r@ r r r Ú test_polymulf s
|
z*TestBinaryOperatorsSameSymbol.test_polymulc C s" t | j|D ]}t|jd q d S r0 ©Údivmodr r r r@ r r r Ú test_divmodj s z)TestBinaryOperatorsSameSymbol.test_divmodc C s || j }t|jd d S r0 r1 r@ r r r Ú test_raddn s
|
z'TestBinaryOperatorsSameSymbol.test_raddc C s || j }t|jd d S r0 r1 r@ r r r Ú test_rsubr s
|
z'TestBinaryOperatorsSameSymbol.test_rsubc C s || j }t|jd d S r0 r1 r@ r r r Ú test_rmulv s
|
z'TestBinaryOperatorsSameSymbol.test_rmulc C s" t || jD ]}t|jd q d S r0 rD r@ r r r Ú test_rdivmodz s z*TestBinaryOperatorsSameSymbol.test_rdivmodN)r' r( r) r* r r r rA rB rC rF rG rH rI rJ r r r r r? P s r? c @ sb e Zd ZejdddgddZejdddgd dZejejej ej
|
ej fZ e j d
|
e ¡d d Zd S )Ú"TestBinaryOperatorsDifferentSymbolr r r r
|
r r< r= r> r# Úfc C s t t|| j d S )N)r r- Úother)r rL r r r Útest_binops_fails s z4TestBinaryOperatorsDifferentSymbol.test_binops_failsN)r' r( r) r r r rM Ú__add__Ú__sub__Ú__mul__Ú __floordiv__Ú__mod__Úopsr r+ r, rN r r r r rK s
|
rK c @ s0 e Zd ZejdddgddZdd Zdd Zd
|
S ) Ú TestEqualityr r r r
|
r c C s& t jdddgdd}t| j|k d S )Nr r r r
|
r ©r r r r ©r rM r r r Útest_eq s zTestEquality.test_eqc C s( t jdddgdd}t| j|k d S )Nr r r r# r rV rW r r r Útest_neq s zTestEquality.test_neqN)r' r( r) r r r rX rY r r r r rU s rU c @ s e Zd ZdZejddddgddZdd Zd
|
d Zd d Z e
|
j ddddgidddgidej if¡dd Zdd Zdd ZdS )ÚTestExtraMethodszJ
|
Test other methods for manipulating/creating polynomial objects.
|
r r r r r r c C s | j ¡ }t|jd d S r0 )r Úcopyr r rW r r r Ú test_copy s
|
zTestExtraMethods.test_copyc C s | j ¡ }t|jd d S r0 )r Ztrimr r rW r r r Ú test_trim s
|
zTestExtraMethods.test_trimc C s | j d¡}t|jd d S )Nr r )r Útruncater r rW r r r Ú test_truncate£ s z TestExtraMethods.test_truncateÚkwargÚdomainiöÿÿÿr5 ÚwindowÚkindc C s | j jf |}t|jd d S r0 )r Úconvertr r )r r` rM r r r Ú test_convert§ s z TestExtraMethods.test_convertc C s | j ¡ }t|jd d S r0 )r Zintegr r rW r r r Ú
|
test_integ° s
|
zTestExtraMethods.test_integc C s | j ¡ }t|jd d S r0 )r Zderivr r rW r r r Ú
|
test_deriv´ s
|
zTestExtraMethods.test_derivN)r' r( r) r* r r r r\ r] r_ r r+ r, Z Chebyshevre rf rg r r r r rZ s
|
|
ý
|
rZ c C sD t jdddgdd} t jddddgd d}| |}|jd ks@td S )
|
Nr r r r$ r r= r éÿÿÿÿu λ_1)r r r ÚAssertionError)r ÚqÚrr r r Útest_composition¹ s rl c C s6 t dfd \} }tjj| |ddd}t|jd d S )Nr5 r r r )Údegr )Úranger r Zfitr r )r
|
r# r r r r Útest_fitÅ s ro c C s( ddg} t jj| dd}t|jd d S )Néþÿÿÿr r r )r r Z fromrootsr r )Úrootsr r r r Útest_froomrootsË s rr c C s* t jjddgddgdd} t| jd d S )Nrh r r= é r )ra rb r )r r Úidentityr r ©r r r r Ú test_identityÑ s rv c C s t jjddd} t| jd d S )Nr r r )r r Zbasisr r ru r r r Ú
|
test_basisÖ s rw )r* r Znumpy.polynomialZ
|
polynomialr Z
|
numpy.corer Z numpy.testingr r r r r/ r+ r, r r? rK rU rZ rl ro rr rv rw r r r r Ú<module> s* 1 þþ(
|
$
|
