U
|
O ±dÕ` ã @ sb d dl Z d dlZd dlZd dlmZmZmZmZ d dlm Z m
|
Z
|
m Z m Z dd Z G dd dZdS )é N)Úuint16Úfloat16Úfloat32Úfloat64)Úassert_Ú assert_equalÚ_OLD_PROMOTIONÚIS_WASMc
|
O s` z||| W n> t k
|
rL } z tt| | ¡dkd| W 5 d }~X Y nX tdd| d S )Nr z%Did not raise floating point %s errorF)ÚFloatingPointErrorr ÚstrÚfind)ZstrmatchÚcallableÚargsÚkwargsÚexc© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_half.pyÚassert_raises_fpe s ÿÿr c @ sZ e Zd Zdd Zdd Zej dddg¡dd Zej dddg¡d
|
d Z ej d d ddg¡ej dd ddg¡ej de
|
j e
|
j g¡e
|
¡ dd Zej dddge
|
j e
|
jdfe
|
j e
|
jdfg¡dd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Ze
|
¡ d*d+ Zejje ¡ d,kd-d.ejje d/d.d0d1 Z d2d3 Z d S )4ÚTestHalfc C sÈ t jdtd| _t| j_t j| jtd| _t j| jt d| _
|
t t jdddtdt jdddtdf¡| _ t| j _t j| j td| _ t j| j t d| _| j dd
| _| j dd
| _| jdd
| _d S ) Ni ©Údtypeé ü iÿ éÿÿÿÿr i| é )ÚnpÚaranger Úall_f16r r Úarrayr Úall_f32r Úall_f64Z concatenateÚ nonan_f16Z nonan_f32Z nonan_f64Ú
|
finite_f16Ú
|
finite_f32Ú
|
finite_f64)Úselfr r r Ú setup_method s ÿÿzTestHalf.setup_methodc C sÊ t j| jtd}t| jjtd|jtd t j| jtd}t| jjtd|jtd t j| j t j
|
d}t j|td}t| j jtd|jtd t dd¡}t j|td}t j|t d}t|| dS )z[Checks that all 16-bit values survive conversion
|
to/from 32-bit and 64-bit floatr i øÿÿi N) r r r r r r Úviewr r r Z
|
longdoubler Úint)r$ ÚbZa_ldZi_intZi_f16Újr r r Útest_half_conversions( s"
|
ÿ
|
ÿ
|
ÿ z TestHalf.test_half_conversionsÚ string_dtÚSÚUc C sd t | d¡}t t j|¡|ks&tt |t j¡|ks<tt jdt jd |¡}|j|ks`td S )NÚ32é r )r r Z promote_typesr ÚAssertionErrorÚonesÚastype)r$ r+ Ú expected_dtZarrr r r Ú test_half_conversion_to_stringF s
|
z'TestHalf.test_half_conversion_to_stringc C s2 t jd|d}| t j¡t jdt jdks.td S )Nz3.1416r g§èH.ÿ! @)r r r2 r r0 )r$ r+ Ústringr r r Ú test_half_conversion_from_stringP s z)TestHalf.test_half_conversion_from_stringÚoffsetNÚupÚdownÚshiftÚfloat_tc
|
C s t t t j¡j¡ t j¡}t jd|d t jd}| t j¡ |¡}|dkrrd|d d
|dd
dd
}n:|dkr d|d d
|dd
d d
}n |dd
}|dkrÈt ||t j ¡}n |dkrät ||t j ¡}| t j¡ t j¡}|dd
|
¡ }|dkr"|dkr"d} n |dkr<|dkr<d} nd} |d krn|dd d
t j ¡d | 7 < n| t j ¡d | 7 < t || d S )
|
Nr r r r8 ç à?r r9 é .) r r ZfinfoÚmaxr& r r r2 Ú nextafterÚinfÚcopyÚint16r )
|
r$ r; r: r7 Z max_patternZ f16s_patternsZ
|
f16s_floatZ res_patternsZ cmp_patternsZ shift_patternr r r Ú test_half_conversion_roundingU s. &&
|
$z&TestHalf.test_half_conversion_roundingÚuint_tÚbitsé é4 c C s t d¡ t j¡ |¡}|dks$t||d }| t j¡dksDtt|D ]8}| |¡|d|> B }| |¡}| t j¡|ksLtqLd S )Nr ç p>r= r )r r r& r r2 r0 Úrange) r$ r; rD rE Zsmallest_valueZrounded_to_zeroÚiZlarger_patternZ larger_valuer r r Ú(test_half_conversion_denormal_round_even s
|
z1TestHalf.test_half_conversion_denormal_round_evenc C s t jddz tt | j¡t | j¡ tt | j¡t | j¡ tt | j¡t | j¡ tt | j¡t | j¡ tt t
|
d¡t j t
|
t j }t | j|k ¡ t || jk ¡ t | j|k ¡ t || jk ¡ t | j|k ¡ t || jk ¡ t | j|k ¡ t || jk ¡ t | j|k ¡ t || jk ¡ t | j|k ¡ t || jk ¡ W 5 Q R X d S )NÚignore©Úalléàÿ )r Úerrstater Úisnanr r ÚisinfÚisfiniteÚsignbitÚspacingr r@ Únanr ÚanyrN )r$ rV r r r Útest_nans_infs¥ s&
|
zTestHalf.test_nans_infsc C sr t ddddddddd d
|
d d d dt jt j g¡}t jddddddddddddd dd d gtd }t|_t|| d S ) z,Confirms a small number of known half valuesç ð?ç ð¿ç @g Àg ¹?ç TÕ?rO é ÿÿç ?ç ¿rH g p¾r g i < i ¼ i @ i À if. iU5 iÿ{ iÿû é i r i é é | r r N)r r r@ r r r r ©r$ Úar( r r r Útest_half_valuesà sH ù ùùzTestHalf.test_half_valuesc C s¦ t jddddddddgtd }d
|
d d d d d dt jg}t jdd t j|td }W 5 Q R X t|| t j|td }t jdd t j|td }W 5 Q R X t|| dS )z7Checks that rounding when converting to half is correctg `>g `>g P>g ð?g ð?g ð?iïÿ iðÿ r rH g g ð?rY rO rL )ZoverN)r r r r@ rP r r r )r$ rd Zroundedr( r r r Útest_half_roundingØ s6 ùø ù
|
|
zTestHalf.test_half_roundingc C s | j jtd}d|d@ d? }tj|d@ d? tjdd }|d@ d }||d k d
|
7 < d ||d k< || d | }t | j|k¡d }t|d krÆ|d }t | j|d| j | | j| || f t | j
|
|k¡d }t|d kr|d }t | j
|
|d| j | | j
|
| || f dS )z_Take every finite float16, and check the casting functions with
|
a manual conversion.r rZ ra é rb é
|
iÿ g P?iñÿÿÿr iòÿÿÿr[ r z,First non-equal is half value %x -> %g != %gN) r! r& r r r Úint32Únonzeror" Úlenr r# ) r$ Za_bitsZa_sgnZa_expZa_manZa_manualZa32_failZ bad_indexZa64_failr r r Útest_half_correctness÷ s: þÿÿþÿÿz TestHalf.test_half_correctnessc C s$ | j ddd
¡ }tj|td}| ¡ | ¡ t|| t|dd
|dd
k ¡ t|dd
|dd
k ¡ t|dd
|dd
k ¡ t|dd
|dd
k ¡ tt
|
|dd
|dd
k ¡d j |j d tt
|
|dd
|dd
k¡d j |j d dS )z'Make sure comparisons are working rightNr r r r r= ) r rA r r r Úsortr r rN rW rj Úsizerc r r r Útest_half_ordering s
|
" "0zTestHalf.test_half_orderingc C sj t tjdtdtjdtd tjdtd}| d¡ t |tjdtd tjddddddd gtd}t | ¡ d d
|
d d g |
|
¡ ¡ }t | ¡ d d
|
d d g tjddd td}tjdtd}t t ||¡d tjdtj dd ddddgtd}t | ¡ d tjdtj dtj dtjddgtd}t | ¡ d tjdtd}tdD ]}t | |¡| qNdS )zTest the various ArrFuncsrh r )é r r r g#B ¡Ç»rH gÖTÿß>r= rp é r< )é é_ éþÿÿÿg)@g333333 @gÍÌÌÌÌÌ @gÍÌÌÌÌÌ(@é N)r r r r r ÚzerosÚfillr1 r rj ZbyteswapZ newbyteorderÚdotr@ ZargmaxrV rI Úitem)r$ rd r( rJ r r r Útest_half_funcs2 s< ÿ
|
ÿ ÿ ÿ"ÿ&ÿ zTestHalf.test_half_funcsc C sÐ t jdtd}t jt jftd}t jt jftd}|jtd}tt |dd
¡|dd
|dd
tt
|
|dd
|¡|dd
tt
|
|d | ¡|d tt
|
|dd
| ¡|dd
tt
|
||¡|d tt
|
| |¡|d tt
|
||¡| tt
|
|| ¡|d tt
|
| |¡|d tt
|
| | ¡| tt
|
||¡|d tt
|
||¡|d tt
|
||¡| tt
|
||¡| tt
|
||¡| |dO }tt |d ¡t |d ¡ tt |dd
¡|dd
|dd
tt
|
|d |¡|d tt
|
|dd
|¡|dd
tt
|
|dd
| ¡|dd
tt
|
||¡|d tt
|
| |¡|d tt
|
||¡|d tt
|
||¡|d dS )z Test np.spacing and np.nextafterrb r Nr r r ra ) r r r r r@ r rV r& r rU r? )r$ rd ZhinfZhnanZa_f16r r r Útest_spacing_nextafterY s: ," $ , "$z TestHalf.test_spacing_nextafterc C s
|
t jdddddgtd}t jdddddgtd}t jdd t j t jd
|
gtd}tt ||¡dd
|
dd dg tt ||¡dd ddd g tt ||¡dddd d
|
g tt ||¡dddddg tt
|
||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddg tt |¡dddddg tt |¡dddddg tt |¡dddddg tt |¡dddddg tt |¡dddddg tt ||¡dddddg tt ||¡dddddg t ||¡}tt |d ¡ d|d< t|ddddd
|
g tt ||¡dddddg t ||¡}tt |d ¡ d|d< t|dd t j ddg tt ||¡dddddg tt ||¡ddddd
|
g tt ||¡dddddg tt ||¡dd t j ddg tt ||¡dddddg tt ||¡dddddg tt ||¡dddddgdddddgf tt !|¡dddd dg tt "|¡dddddg tt #|¡dddddg tt $|¡| tt %|¡dddddg tt &|¡ddd d dg tt '|¡| tt (|¡d ddddg tt )|¡dddddg|f tt *|¡dddddgdddddgf tt +|dddddg¡dd dd d g d S ) zTest the various ufuncsr r r= ru r rt rp r/ r rq é éüÿÿÿé g É?g Tå?FTé é g à¿g Ð?r\ éûÿÿÿéýÿÿÿg ä?r< g è?rh é@ é N),r r r r@ rV r ÚaddÚsubtractÚmultiplyÚdivideÚequalÚ not_equalZlessZ
|
less_equalZgreaterZ greater_equalÚ logical_andÚ
|
logical_orÚ logical_xorZ logical_notrQ rR rS rT ÚcopysignÚmaximumr ZminimumZfmaxZfminZ floor_divideÚ remainderÚdivmodZsquareZ
|
reciprocalZ ones_likeÚ conjugateÚabsoluteÚnegativeZpositiveÚsignÚmodfÚfrexpÚldexp)r$ rd r( ÚcÚxr r r Útest_half_ufuncs sd * (zTestHalf.test_half_ufuncsc C sÆ t jdtd}t jdtd}td}td}t |d¡jtksBtt |d¡jtksXtt ||¡jtksnt|rvtnt}t ||¡j|kstt ||¡jtks¦tt ||¡jtks¼t|rÄtnt}t |d¡j|ksÞtt |d¡j|ksôtt ||¡js ttt ||¡js$ttt ||¡js<ttt ||¡jsTttt ||¡jtksltt ||¡jtkst|rtnt}t ||¡j|ksªtt ||¡jtksÂtdS )z9Test that half gets coerced properly with the other types©r r r r= r[ N)r r r r Úpowerr r0 r )r$ Zweak_promotionZa16Za32Zb16Zb32r3 r r r Útest_half_coercionÆ s. zTestHalf.test_half_coercionZarmv5telz See gh-413.)Úreasonz!fp exceptions don't work in wasm.c C s² t jdd t jdtd}t jdtd}td}td}tdd d
|
|| tdd d
|
|| tdd d
|
|| tdd d
|
|| tddd
|
|| tddd
|
|| tddd
|
|| tddd
|
|| tddd
|
tdtd tddd
|
tdtd tddd
|
tdtd tddd
|
tdtd tdd d
|
td td td d d
|
|| td d!d
|
|| td d"d
|
|| td d#d
|
|| td d$d
|
|| td d%d
|
|| td d&d
|
|| td d'd
|
|| td d(d
|
td)td* td d+d
|
td,td* td t jtd)tt j td t jtd,tt j td t jtd) td-t jtt jtt j td-t jtt j td-t jtt j td.td/ td0td tdtd1 t td,¡ t td)tt j ¡ t td,tt j¡ t tt jtd2¡ t tt j td2¡ t td2tt j ¡ t tt j td2¡ tdtd1 tdtd1 td td td3td W 5 Q R X d S )4NÚraiserM )ç-C ëâ6?r )ç Ã@r¡ r¢ Z underflowc S s | | S ©Nr ©rd r( r r r Ú<lambda>ð ó z(TestHalf.test_half_fpe.<locals>.<lambda>c S s | | S r£ r r¤ r r r r¥ ñ r¦ c S s | | S r£ r r¤ r r r r¥ ò r¦ c S s | | S r£ r r¤ r r r r¥ ó r¦ c S s | | S r£ r r¤ r r r r¥ ô r¦ c S s | | S r£ r r¤ r r r r¥ õ r¦ c S s | | S r£ r r¤ r r r r¥ ö r¦ c S s | | S r£ r r¤ r r r r¥ ÷ r¦ c S s | | S r£ r r¤ r r r r¥ ø r¦ r^ i c S s | | S r£ r r¤ r r r r¥ ú r¦ r_ c S s | | S r£ r r¤ r r r r¥ ü r¦ g ?r= c S s | | S r£ r r¤ r r r r¥ þ r¦ g ¿c S s | | S r£ r r¤ r r r r¥ r¦ g ?ru Zoverflowc S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥
|
r¦ c S s | | S r£ r r¤ r r r r¥ r¦ c S s | | S r£ r r¤ r r r r¥ r¦ rO é c S s | | S r£ r r¤ r r r r¥ r¦ r] ÚinvalidiÀÿ é g ?r` r g ¿)
|
r rP r r r r? r@ rU r rV )r$ Zsx16Zbx16Zsy16Zby16r r r Ú test_half_fpeä s
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ zTestHalf.test_half_fpec C sP G dd d}t jdtd}| }|j|_t |¡}t|jtk t|| dS )z5Test that half is compatible with __array_interface__c @ s e Zd ZdS )z1TestHalf.test_half_array_interface.<locals>.DummyN)Ú__name__Ú
|
__module__Ú __qualname__r r r r ÚDummy+ s r® r r N)r r1 r Z__array_interface__r r r r )r$ r® rd r( r r r r Útest_half_array_interface) s
|
z"TestHalf.test_half_array_interface)!r« r¬ r r% r* ÚpytestÚmarkZ parametrizer4 r6 r r r Z_no_nep50_warningrC Zuint32Zuint64rK rX re rf rl ro rz r{ r r ZskipifÚplatformÚmachiner rª r¯ r r r r r sF
|
|
5 ÿÿ
|
$'-@
|
ÿÿ Ar )r² r° Únumpyr r r r r Z numpy.testingr r r r r r r r r r Ú<module> s
|
