y> j''"&//BBJJ
v
j''26'2B2BCCJ "((4499")DD %QrT*//11 F??v~~ MM
d]]v / / MM M > A # #BH - -c 2 2 2m C 2 2 2 2 bh ' ' , , ,
= , , ,\<00000r- )rK uint_tbits 4 c t j d t j | }|dk sJ | |d z }| t j dk sJ t | D ]`}| | |d|z z }| | }| t j |k sJ ad S )Nr p>rN r )r r r/ r r@ range) r+ rK r\ r] smallest_valuerounded_to_zeroilarger_pattern larger_values r (test_half_conversion_denormal_round_evenz1TestHalf.test_half_conversion_denormal_round_even s 1 **2:66==gFF '''' )771::5 %%bj11Q6666 t E EA , 1 1& 9 9FF16NN JN)..w77L &&rz22nDDDDD E Er- c t j d 5 t t j | j t j | j t t j | j t j | j t t j | j t j | j t t j | j t j | j t t j t d t j t t j }t | j |k t || j k t | j |k t || j k t | j |k t || j k t | j |k t || j k t | j |k t || j k t | j |k t || j k d d d d S # 1 swxY w Y d S )Nignoreall )r errstater isnanr r" isinfisfinitesignbitspacingr rR nanr anyrl )r+ rt s r test_nans_infszTestHalf.test_nans_infs s
[X
&
&
& 5 5 $,//$,1G1G H H H $,//$,1G1G H H H T\22BK 4M4M N N N DL112:dl3K3K L L L GENN33RV < < < "&//C ,11333 4 4 4 ,11333 4 4 4 T\S(--// 0 0 0 SDL(--// 0 0 0 +00222 3 3 3 t|+00222 3 3 3 ,11333 4 4 4 ,11333 4 4 4 +00222 3 3 3 t|+00222 3 3 3 ,11333 4 4 4 ,11333 4 4 47 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s M
M00M47M4c t j ddddddddd d
d d d
dt j t j g }t j g dt }t |_ t
|| dS )z,Confirms a small number of known half values ? @g g ? T?rm ? ra g pr g )i < i i @ i if. iU5 i{ i i r i r | r r N)r r! rR r r r r r+ ar2 s r test_half_valueszTestHalf.test_half_values s Hc44%~V + +frvg'
(
(
H & & & .4
5
5
5 Qr- c t j g dt }dddddddt j g}t j d 5 t j |t
}d
d
d
n # 1 swxY w Y t
|| t j |t }t j d 5 t j |t
}d
d
d
n # 1 swxY w Y t
|| d
S ) z7Checks that rounding when converting to half is correct)g `>g `>g P>g ?g ?g ?i i r ra g g ?rx rm rj )overN)r r! r rR rn r r r )r+ r roundedr2 s r test_half_roundingzTestHalf.test_half_rounding s H $
%
%
% !6 [h
'
'
' + +'***A + + + + + + + + + + + + + + +Q
HQg & & &
[h
'
'
' + +'***A + + + + + + + + + + + + + + +Q s$ A//A36A3: C""C&)C&c | j t }d|dz dz z }t j |dz dz t j dz
}|dz dz }||d k xx d
z
cc< d ||d k < ||z d |z z }t j | j |k d
}t | d
k r@|d
}t | j |d| j | | j | || fz t j | j
|k d
}t | d
k rB|d
}t | j
|d| j | | j
| || fz dS dS )z_Take every finite float16, and check the casting functions with
a manual conversion.r ry r r
i g P?ir irz r z,First non-equal is half value %x -> %g != %gN) r( r/ r r r! int32nonzeror) lenr r* ) r+ a_bitsa_sgna_expa_mana_manuala32_fail bad_indexa64_fails r test_half_correctnessz TestHalf.test_half_correctness s
%%F%33 &6/b01&6/b0AAABF& J.
eslq !esl5=3:-:do 9::1= x==A I (? !_Y 7 !_Y 7 %i 0 22
3
3
3 :do 9::1= x==A I (? !_Y 7 !_Y 7 %i 0 22
3
3
3
3
3 r- c | j ddd }t j |t }| | t
|| t |dd |dd k t |dd |dd k t |dd |dd k t |dd |dd k t
t j
|dd |dd k d j |j dz
t
t j
|dd |dd k d j |j dz
dS )z'Make sure comparisons are working rightNr r r r rN ) r% rS r r! r sortr r rl ru r sizer s r test_half_orderingzTestHalf.test_half_ordering s
N44R4 % % ' '
HQg & & &
Q 3B31QRR5 %%''(((QssVae^((***+++122!CRC& %%''(((QqrrUQssV^((***+++RZ#2#122//27BBBRZ!""#2#//27BBBBBr- c
( t t j dt t j dt t j dt }| d t |t j dt t j g dt }t | d g d |
}t | d g d t j dddt }t j d t }t t j || d
t j dt j
d dd d
ddgt }t | d t j dt j
d t j
d t j ddgt }t | d t j dt }t! d D ]%}t | | | &dS )zTest the various ArrFuncsr r ) r )r r r g#B ǻr ra gT>r )rN r rM ) _ g)@g333333 @g @g(@ r N)r r r r r zerosfillr? r! r byteswap newbyteorderdotrR argmaxrt rb item)r+ r r2 re s r test_half_funcszTestHalf.test_half_funcs2 s8 RYr111Yr111 3 3 3
HT ) ) ) q QG444555
H???w O O OQYY[[^ YY
JJLL % % ' 'QYY[[^ YY
IaS 0 0 0 GE ) ) )RVAq\\
Ha"&"c5#sDA Q Q QQXXZZ Ha"&"bfeRVS$Gw W W WQXXZZ
Ib ( ( (r ' 'A A & & & & ' 'r- c t j dt }t j t j ft
}t j t j ft
}| t
}t t j |dd |dd |dd z
t t j
|dd | |dd t t j
|d | |d t t j
|dd | |dd t t j
|| |d t t j
| | |d t t j
|| | t t j
|| |d t t j
| | |d t t j
| | | t t j
|| |d t t j
|| |d t t j
|| | t t j
|| | t t j
|| | |dz }t t j |d t j |d t t j |dd |dd |dd z
t t j
|d | |d t t j
|dd | |dd t t j
|dd | |dd t t j
|| |d t t j
| | |d t t j
|| |d t t j
|| |d dS )z Test np.spacing and np.nextafterr r Nr r r r ) r r r r! rR r rt r/ r rs rQ )r+ r hinfhnana_f16s r test_spacing_nextafterz TestHalf.test_spacing_nextafterY s
IfF + + +x 111x 111W%%RZcrc
++U122YuSbSz-ABBBR\%*d33U122Y???R\%(TE22U1XI>>>R\%)dU33U3B3Z@@@R\$..b :::R\4%//%)<<<R\$--t444R\$..b :::R\4%..r
;;;R\4%$//$777R\%..Q888R\$..Q888R\$--t444R\$--t444R\$--t444
V RZa))2:eAh+?+?@@@RZabb **E#2#JuQRRy,@AAAR\%(D11E!H9===R\%)T22E#2#J???R\%*te44eABBi@@@R\$..r
;;;R\4%//r;;;R\%..Q888R\$..Q88888r- c
t j g dt }t j g dt }t j ddt j t j dgt }t t j || g d t t j || g d t t j || g d t t j || g d
t t j
|| g d t t j || g d t t j || g d
t t j
|| g d t t j || g d t t j || g d t t j || g d t t j || g d t t j || g d t t j | g d t t j | g d t t j | g d t t j | g d t t j | g d t t j || g d t t j || g d t j || }t5 t j |d d|d< t |g d t t j || g d t j || }t5 t j |d d|d< t |ddt j ddg t t j || g d t t j || g d t t j || g d t t j || ddt j d dg t t j || g d t t j || g d t t j || g d g d f t t j! | g d t t j" | g d! t t j# | g d" t t j$ | | t t j% | g d t t j& | g d# t t j' | | t t j( | g d$ t t j) | g d%|f t t j* | g d&g d'f t t j+ |g d g d( d)S )*zTest the various ufuncs)r r rN r rN r )r r r r r= r r r )r r r= r )rN r r r )r r rN r )r ?rN r g T?)FFFTF)TTTFT)FTFFT)FTFTT)TFTFF)TFTTF)FTTTT)TTTTT)TFFFF)FFTFF)TTFFT)rN r r r r= )r r rN r r= r= )r r r r r )r r r r rN r )r r r r r r )r r rN r r )r r r r rN )r r r ) r r g ?r{ )r r r r r )rN r r )r r r r r )r r r r r )r g ?rM rM g ?)rN r= r r= rN )r r r @ N),r r! r rR rt r addsubtractmultiplydivideequal not_equalless
less_equalgreater
greater_equal logical_and
logical_or logical_xor logical_notro rp rq rr copysignmaximumr minimumfmaxfmin floor_divide remainderdivmodsquare
reciprocal ones_like conjugateabsolutenegativepositivesignmodffrexpldexp)r+ r r2 cxs r test_half_ufuncszTestHalf.test_half_ufuncs s
H___G 4 4 4 H%%%W 5 5 5 HabfWbfa0 @ @ @RVAq\\#3#3#3444R[A&&(9(9(9:::R[A&&(8(8(8999RYq!__&N&N&NOOORXa^^%G%G%GHHHR\!Q'')H)H)HIIIRWQ]]$E$E$EFFFR]1a((*J*J*JKKKRZ1%%'H'H'HIIIR%a++-M-M-MNNNR^Aq))+J+J+JKKKR]1a((*H*H*HIIIR^Aq))+M+M+MNNNR^A&&(J(J(JKKKRXa[["D"D"DEEERXa[["D"D"DEEER[^^%E%E%EFFFRZ]]$F$F$FGGGR[A&&888RZ1%%777 Jq! 1 !Q(((RZ1%%'7'7'7888 Jq! 1 !QR"&!Q/000RWQ]]OOO444RWQ]]OOO444RWQ]]$4$4$4555RWQ]]RbfWa$;<<<R_Q**OOO<<<R\!Q'' 999RYq!__&HIIIRYq\\#4#4#4555R]1%%'V'V'VWWWR\!__ooo666R\!__a(((R[^^___555R[^^%8%8%8999R[^^Q'''RWQZZ!1!1!1222RWQZZ///1!5666RXa[[#@#@#@///"RSSSRXa 113F3F3FGGGGGr- c t j dt }t j dt }t d }t d }t j |d j t k sJ t j |d j t k sJ t j || j t k sJ |rt nt }t j || j |k sJ t j || j t k sJ t j || j t k sJ |rt nt }t j |d j |k sJ t j |d j |k sJ t j || j sJ t t j || j sJ t t j || j sJ t t j || j sJ t t j || j t k sJ t j || j t k sJ |rt nt }t j || j |k sJ t j || j t k sJ dS )z9Test that half gets coerced properly with the other typesr r r rN rz N)r r! r r powerr r )r+ weak_promotiona16a32b16b32rA s r test_half_coercionzTestHalf.test_half_coercion s~ ht7+++ht7+++ajjajjxQ %0000xS!!'72222xS!!'72222!/z(TestHalf.test_half_fpe..
qs r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- r} i c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- r~ c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- g ?rN c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- g c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- g ?r overflowc | |z S r r r s r r z(TestHalf.test_half_fpe..
ac r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe..
r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- c | |z S r r r s r r z(TestHalf.test_half_fpe.. r r- rm c | |z
S r r r s r r z(TestHalf.test_half_fpe.. r r- r| invalidi g ?r r g )
r rn r! r r rQ rR rs r rt )r+ sx16bx16sy16by16s r
test_half_fpezTestHalf.test_half_fpe s
[W
%
%
% > / > /8G7333D8F'222D4==D3<->
P
P
P k??-4^-D-Dgajj
R
R
R k??-4_-E-Ewqzz
S
S
S k??-4^-D-Dgajj
R
R
R
j//4 F F F j//4 F F F j//4 F F F j//4 F F F j//4 F F F j//4 F F F j//4 F F F j//4 F F F j//-4U^^WR[[
J
J
J j//-4V__gbkk
K
K
K j",PRPV X X X j",RTRXQXIYIY Z Z Z j"*genn E E E
iGBFOOWRV__ U U U iWRV__ E E E iWRV__ E E E
ENN72;; & & FOOGAJJ & & FOOGENN * * Jwv ' ' ' L "&)9)9 : : : L '"&// : : : L '!** 5 5 5 L"&))71:: 6 6 6 LWRV__ 5 5 5 L '!** 5 5 5 FOOGENN * * G WU^^ + + M " "71:: - - N # #GAJJ . .}> / > / > / > / > / > / > / > / > / > / > / > / > / > / > / > / > / > /s Y)Z ZZc G d d }t j dt } | }|j |_ t j | }t |j t k t || dS )z5Test that half is compatible with __array_interface__c e Zd ZdS )1TestHalf.test_half_array_interface..DummyN)__name__
__module__ __qualname__r r- r Dummyr + s Dr- r r r N)r r? r __array_interface__r! r r r )r+ r r r2 r s r test_half_array_interfacez"TestHalf.test_half_array_interface) s
GD ( ( ( EGG ! 5 HQKK7"###Qr- N)!r r r r, r7 pytestmark parametrizerC rF r r r _no_nep50_warningr[ uint32uint64rh rv r r r r r r r r skipifplatformmachiner
r r r r- r r r s / / /( < [ [3*55( ( 65( [ [3*55O O 65O [ X';';';<< [ W&:&:&:;; [ YRZ(@AAR31 31 BA <; =<31j [ : : : !z29b 9 !z29b 9 ;< <E E< <E( 5 5 5< * ! ! !>"3 "3 "3HC C C.%' %' %'N+9 +9 +9Z>H >H >H@ R3 3 3: [((**j8 , . . [ B D D?/ ?/D D. .?/B
r- r )r& r numpyr r r r r
numpy.testingr r r r
r r r r- r r* s
3 3 3 3 3 3 3 3 3 3 3 3 H H H H H H H H H H H HD D D` ` ` ` ` ` ` ` ` `r-