1j x d Z ddlZddlmZ ddlmZ ddlmZm Z m
Z
G d d Z G d d Z ej
d ej g d
d eg d
f G d
d Z G d d Z G d d Z G d d Zd Zd Zd Zd Zd ZdS )zE
Tests related to the ``symbol`` attribute of the ABCPolyBase class.
N)array) assert_equal
assert_raisesassert_c e Zd ZdZg dZd Zej dde fde fde
fde
ff d Z ej d
d d Z d
Z
d ZdS )TestInitz5
Test polynomial creation with symbol kwarg.
c b t j | j }t |j d d S )Nxpoly
Polynomialcr symbolselfps v/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_symbol.pytest_default_symbolz TestInit.test_default_symbol s+ ODF # #QXs##### ) bad_input exception 3Nr
c t j | 5 t j | j | }d d d d S # 1 swxY w Y d S )Nr )pytestraisesr r r )r r r r s r test_symbol_bad_inputz TestInit.test_symbol_bad_input s ]9
%
% : :y999A : : : : : : : : : : : : : : : : : :s >AAr )r x_1Axyzu βc f t j | j | }t |j | dS )zF
Values for symbol that should pass input validation.
r Nr )r r r s r test_valid_symbolszTestInit.test_valid_symbols s2
ODF6 2 2 2QXv&&&&&r c t j | j d }t j t
5 d|_ ddd dS # 1 swxY w Y dS )z2
'symbol' attribute is read only.
r r zN)r r r r r! AttributeErrorr r s r
test_propertyzTestInit.test_property- s
ODF3 / / /
]>
*
* AH s A
AAc t j | j d }t j |j d }t |j d d S )Nyr t)r r r coefr r )r r pts r test_change_symbolzTestInit.test_change_symbol5 sF ODF3 / / /
_QVC
0
0
0RY$$$$$r )__name__
__module__ __qualname____doc__r r r mark parametrize
ValueError TypeErrorr" r' r+ r1 r r r r s A$ $ $ [ 7 Z j
y
I : : :
: [ X ( ' ' ' % % % % %r r c N e Zd Z ej g dd Zd Zd Zd Zd Z dS ) TestUnaryOperatorsr r) r c @ | j }t |j d d S Nr) r r r )r ns r test_negzTestUnaryOperators.test_neg? s#
VGQXs#####r c D | j dz }t |j d d S N
r) r? r outs r test_scalarmulz!TestUnaryOperators.test_scalarmulC s% frkSZ%%%%%r c D d| j z }t |j d d S rC r? rE s r test_rscalarmulz"TestUnaryOperators.test_rscalarmulG s% 46kSZ%%%%%r c D | j dz }t |j d d S )Nr r) r? rE s r test_powzTestUnaryOperators.test_powK s% fkSZ%%%%%r N)
r2 r3 r4 r r r rA rG rI rK r: r r r< r< < sj #...A$ $ $& & && & && & & & &r r< rhs r) r c j e Zd ZdZ ej g dd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
S ) TestBinaryOperatorsSameSymbolzc
Ensure symbol is preserved for numeric operations on polynomials with
the same symbol
r r) r c D | j |z }t |j d d S r> r? r rL rF s r test_addz&TestBinaryOperatorsSameSymbol.test_add^ % fslSZ%%%%%r c D | j |z
}t |j d d S r> r? rT s r test_subz&TestBinaryOperatorsSameSymbol.test_subb rV r c D | j |z }t |j d d S r> r? rT s r test_polymulz*TestBinaryOperatorsSameSymbol.test_polymulf rV r c ` t | j | D ]}t |j d d S r> divmodr r r rT s r test_divmodz)TestBinaryOperatorsSameSymbol.test_divmodj s< $&#&& * *C S ) ) ) ) * *r c D || j z }t |j d d S r> r? rT s r test_raddz'TestBinaryOperatorsSameSymbol.test_raddn % DFlSZ%%%%%r c D || j z
}t |j d d S r> r? rT s r test_rsubz'TestBinaryOperatorsSameSymbol.test_rsubr ra r c D || j z }t |j d d S r> r? rT s r test_rmulz'TestBinaryOperatorsSameSymbol.test_rmulv ra r c ` t || j D ]}t |j d d S r> r\ rT s r test_rdivmodz*TestBinaryOperatorsSameSymbol.test_rdivmodz s< #tv&& * *C S ) ) ) ) * *r N)r2 r3 r4 r5 r r r rU rX rZ r^ r` rc re rg r: r r rR rR P s #...A& & && & && & &* * *& & && & && & &* * * * *r rR c e Zd Z ej g dd Z ej g dd Zej ej ej ej
ej fZ e
j de d ZdS ) "TestBinaryOperatorsDifferentSymbolr r r rM r- fc < t t || j d S )N)r r8 other)r rj s r test_binops_failsz4TestBinaryOperatorsDifferentSymbol.test_binops_fails s j!TZ00000r N)r2 r3 r4 r r r rl __add____sub____mul__ __floordiv____mod__opsr r6 r7 rm r: r r ri ri s #...A DOIIIc 2 2 2E 9aiANAI
FC [ S#&&1 1 '&1 1 1r ri c B e Zd Z ej g dd Zd Zd ZdS ) TestEqualityr r r c f t j g dd }t | j |k d S )Nr r r r r r r r rl s r test_eqzTestEquality.test_eq s4 #666% r c h t j g dd }t | j |k d S )Nr r- r rw rx s r test_neqzTestEquality.test_neq s7 #666DFeO#$$$$$r N)r2 r3 r4 r r r ry r{ r: r r ru ru sL #...A! ! !% % % % %r ru c e Zd ZdZ ej g dd Zd Zd Zd Z e
j dd d
d gid d
d gid
ej
if d Zd Zd ZdS )TestExtraMethodszJ
Test other methods for manipulating/creating polynomial objects.
)r
r r r r) r c b | j }t |j d d S r> )r copyr r rx s r test_copyzTestExtraMethods.test_copy )
U\3'''''r c b | j }t |j d d S r> )r trimr r rx s r test_trimzTestExtraMethods.test_trim r r c d | j d }t |j d d S )Nr r) )r truncater r rx s r
test_truncatez TestExtraMethods.test_truncate s- ""U\3'''''r kwargdomainirD windowkindc T | j j di |}t |j d d S )Nr) r: )r convertr r )r r rl s r test_convertz TestExtraMethods.test_convert s5 ''''U\3'''''r c b | j }t |j d d S r> )r integr r rx s r
test_integzTestExtraMethods.test_integ ) U\3'''''r c b | j }t |j d d S r> )r derivr r rx s r
test_derivzTestExtraMethods.test_deriv r r N)r2 r3 r4 r5 r r r r r r r r6 r7 Chebyshevr r r r: r r r} r} s S111A( ( (( ( (( ( ( [ W C9 C9 '
( (
(( ( (( ( ( ( (r r} c t j g dd } t j g dd } | | }|j dk sJ d S )N)r r r
r. r )rO r
r u λ_1)r r r )r qrs r test_compositionr sX #...A
f555A !A 8v r c t d fdz \ } }t j | |dd }t |j d d S )NrD r r
r) )degr )ranger r fitr r )r r- r s r test_fitr sM "II<>DAq Aqa44A3 r c z ddg} t j | d }t |j d d S )Nr r) r )r r fromrootsr r )rootsr s r test_froomrootsr s=
GE !!%!44A3 r c | t j ddgddgd } t | j d d S )Nr r
rO r) )r r r )r r identityr r r s r
test_identityr s> QB LLA3 r c r t j dd } t | j d d S )Nr r) r )r r basisr r r s r
test_basisr s4 a ,,A3 r )r5 r numpy.polynomial
polynomialr
numpy.corer
numpy.testingr r r r r< r6 r7 r rR ri ru r} r r r r r r: r r r s
> > > > > > > > > >.% .% .% .% .% .% .% .%b& & & & & & & &( #...
iii %* %* %* %* %* %* %* %*P1 1 1 1 1 1 1 1 % % % % % % % %!( !( !( !( !( !( !( !(H
r