1j" d Z ddlZddlZddlZddlmZ ej dd d Zej
ed d
Z ej d ej
d d
Zej d ej
d d Zej dej ej ej g d Zd Zd Zd Zd Zej dej ej g ej dddg d Zd ZdS ) z
This file adds basic tests to test the NEP 50 style promotion compatibility
mode. Most of these test are likely to be simply deleted again once NEP 50
is adopted in the main test suite. A few may be moved elsewhere.
N)IS_WASMmoduleT)scopeautousec # K t j } t j d d V t j | d S )N
weak_and_warn)np_get_promotion_state_set_promotion_state)states z/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_nep50_promotions.py_weak_promotion_enabledr sE # % %EO,,, EEEE""""" z'wasm doesn't have support for fp errors)reasonc t j t d 5 t j d dz } d d d n # 1 swxY w Y | j t j k sJ t j t d 5 t j dgt j t j d z } d d d n # 1 swxY w Y | j t j k sJ t j t d 5 t j dgt j t j dt j z } d d d n # 1 swxY w Y | j t j k sJ t j t d 5 t j d 5 t j d d z } d d d n # 1 swxY w Y d d d n # 1 swxY w Y | j t j k sJ t j t 5 }t j
d d
z } d d d n # 1 swxY w Y t | t j
}| d sJ t | t j
}| d sJ t! | d k sJ t j | sJ | j t j
k sJ t j d
gt j
t j d
k } | d dk sJ t j t d 5 t j d
gt j
t j d
z } d d d n # 1 swxY w Y | j t j k sJ t j t d 5 t j dgt j
t j d z } d d d n # 1 swxY w Y | j t j k sJ d S )Nresult dtype changedmatch dtyperaiseoverd g;%_nToverflowr g?F ? )pytestwarns UserWarningr uint8r arrayint64errstateWarningfloat32strpopmessage
startswithRuntimeWarninglenisinffloat64)resrecwarnwarnings r
test_nep50_examplesr5 s k)? @ @ @ hqkkAo 9 k)? @ @ @ 4 4hsBH%% 34 4 4 4 4 4 4 4 4 4 4 4 4 4 4 9 k)? @ @ @ D DhsBH%%"((C(C(CCD D D D D D D D D D D D D D D 9 k)? @ @ @ & & [g
&
&
& & &(3--#%C & & & & & & & & & & & & & & && & & & & & & & & & & & & & &
9 g $'jmme#$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ '++k**233G 4 5 555 5'++n--566G j ) ))) ) w<<1
8C=== 9
" " " "
(C5"*
%
%C
8C q6U????
k)? @ @ @ < <hubj))BJsOO;< < < < < < < < < < < < < < < 9
" " " " k)? @ @ @ 7 7htRZ((28A;;67 7 7 7 7 7 7 7 7 7 7 7 7 7 7 9
" " " " " "s A AA=6B??CCH;)6N++N/2N/(6P**P.1P.r
AllIntegerc t j d t j | j }t t j | j }t j d 5 t j t 5 |d |z }d d d n # 1 swxY w Y d d d n # 1 swxY w Y |j | k sJ t j d| |z }|j | k sJ d S )Nweakwarnr r r ) r r r typeintiinfomaxr' r! r" r. r% )r scalar_typemaxintr2 s r
test_nep50_weak_integersr@ F sg F###(5//&K
% $
%
%F & ! ! ! , ,
\.
)
) , , +c""V+C , , , , , , , , , , , , , , ,, , , , , , , , , , , , , , , 9
(3e
$
$
$v
-C 9 s6 )B5B B5 B" "B5%B" &B55B9<B9AllFloatc t j d t j | j }t t j | j dz }| dv rt j t 5 |d |z d d d n # 1 swxY w Y t j t 5 t j
d| |z d d d d S # 1 swxY w Y d S | dv r5 t | n$# t $ r t j
d Y nw xY wt j t 5 |d |z }d d d n # 1 swxY w Y |j | k sJ |t j k sJ t j t 5 t j t j
d| || }d d d n # 1 swxY w Y |j | k sJ |t j k sJ d S )Nr8 r dDGr r gGz)`huge_int -> string -> longdouble` failed)r r r r: r; finfor= r! raises
OverflowErrorr% r*
ValueErrorskipr" r. infadd)r r> too_big_intr2 s r
%test_nep50_weak_integers_with_inexactrM Y s
F###(5//&Kbhuoo)**Q.K ~~ ]=
)
) ) ) KNN[ ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ]=
)
) 3 3 HQe $ $ ${ 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 D==
IK
I
I
I GHHHHH
I \.
)
) / / +a..;.C / / / / / / / / / / / / / / /yE!!!!bf}}}}
\.
)
) M M &!5111;eLLLC M M M M M M M M M M M M M M M
yE!!!!bf}}}}}}sZ 4BBB3CC !C ,C< < D D 9EEE,GGGopc , t j d | t j d d }| | dd k sJ |j t j k s|j t k sJ t j t 5 | t j d d d d d n # 1 swxY w Y | t j d d }| | dd k sJ |j t j k s|j t k sJ d S d S )Nr8 r i g @g @) r r r$ r boolr! rF rG r) )rN r2 s r
test_weak_promotion_scalar_pathrR sJ F###
"RXa[[!
C ""Q((???? 9 CI$5$5$5 5 } % %
28A;;
"RZ]]B
C ""R** 9
" "ci4&7&7&7 7 " "&7&7s B00B47B4c t j d t j t d 5 t j d t
ddz z } d d d n # 1 swxY w Y t | t j k sJ d S )Nr8 z
.*overflowr r r , )r r r! r" r. complex64complexr: r2 s r
test_nep50_complex_promotionrX s F### nL 9 9 9 0 0l1oo3/0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 99 $ $ $ $ $ $s (A$$A(+A(c t j d t j t d 5 t j dgt j dz d d d n # 1 swxY w Y t j t d 5 t j d dz d d d n # 1 swxY w Y t j t d 5 t j d dz d d d d S # 1 swxY w Y d S )Nr8 z.*uint8r r rT z)Python integer -1 out of bounds for uint8)r r r! rF rG r% r$ r r
$test_nep50_integer_conversion_errorsr\ s F### }I 6 6 6 & &
!bh #%%& & & & & & & & & & & & & & &
}I 6 6 6
c
}=
?
?
?
b s5 $A A$'A$B**B.1B.C55C9<C9c t j d t j d } | dz j t j k sJ | d dz j t j k sJ d S )Nlegacyr r[ )r r r% r r1 )arrs r
test_nep50_integer_regressionra s` H%%%
(1++C %K "* , , , , GeO "bj 0 0 0 0 0 0r c t j d t j t 5 t j t j dt j df d d d d S # 1 swxY w Y d S )Nr8 rP r )r r r! rF rG r_arangeint8r[ r r
test_nep50_with_axisconcatenatorrg s F### } % % 0 0
bi)))3.//0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s 3A..A25A2ufuncr r8 r c ` t j | t j t 5 | t j d d d d d n # 1 swxY w Y |dk rt j t d 5 t j t 5 | t j d d d d d n # 1 swxY w Y d d d n # 1 swxY w Y nOt j t 5 | t j d d d d d n # 1 swxY w Y |dk rRt j t d 5 | t j d d }d d d n # 1 swxY w Y n | t j d d }|j t j k sJ | | ddt k sJ t j t 5 | t j d d d d d n # 1 swxY w Y t j t 5 | t j d d d
z d d d n # 1 swxY w Y | d d d
z }t |t j sJ d S ) Nr r_ r z%result dtype changed.*float64.*uint64r l r r r r r )
r r r! rF rG r& r" r# uint64r object
isinstancer1 )rh r r2 s r
test_nep50_huge_integersrm s8 E""" } % % " "
bhqkk5!!!" " " " " " " " " " " " " " "
\+=? ? ? + + }--
+
+billE***
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ + + + + + + + + + + + + + + +
]=
)
) ' ' E")A,, & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' '
\+=? ? ? - -% ! e,,C - - - - - - - - - - - - - - - eBIaLL%(( 9 ! ! ! ! %%5/// / / / /
} % % " "
bhqkk5!!!" " " " " " " " " " " " " " "
} % % # #
bhqkk1c6"""# # # # # # # # # # # # # # # %QV
C c2: & &&& &&&s AA A C! C
> C!
C C!C C!!C%(C% D11D58D5 FF F ; H&&H*-H*
"I88I<?I<c t j d t j t d 5 t j t j d dgd } d d d n # 1 swxY w Y | j dk sJ t j t d 5 t j dt j d dg } d d d n # 1 swxY w Y | j dk sJ d S )Nr r r r r )axisr) ) r r r! r" r# concatenater) r chooserW s r
test_nep50_in_concat_and_chooserr sT O,,, k)? @ @ @ = =nbjmmR 0t<<<= = = = = = = = = = = = = = = 9 ! ! ! ! k)? @ @ @ 0 0iBJqMM2.//0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 9 ! ! ! ! ! !s# +A''A+.A+*CCC)__doc__operatornumpyr r!
numpy.testingr fixturer markskipifr5 parametrize typecodesr@ rM rK poweqrR rX r\ ra rg powerrm rr r[ r r
r s
! ! ! ! ! ! h---# # .-# G$MNN+# +# ON+#\ ",|"<== >= $ ",z":;;' ' <;' T hlHK HII8 8 JI8$% % %
1 1 10 0 0 2628"4556?";<<&' &' =< 65&'R " " " " "r