U
|
O ±d? ã @ sH d dl Z d dlmZ d dlZd dlZddlmZmZm Z m
|
Z
|
m Z m Z ddl mZ ddlmZmZmZmZmZmZmZmZmZmZmZmZ dd Z d d
|
Z d d Z d d Z dd Z!dd Z"dd Z#dd Z$dd Z%ej& 'dddd g¡ej& 'd d d d g¡d!d" Z(d#d$ Z)ej& 'd%e*d&e*d&d ed ed d'gg¡d(d) Z+d*d+ Z,dS ),é N)Ú assert_raisesé )ÚonesÚasarrayÚreshapeÚ result_typeÚallÚequal)ÚArray) Ú _all_dtypesÚ_boolean_dtypesÚ_floating_dtypesÚ_integer_dtypesÚ_integer_or_boolean_dtypesÚ_numeric_dtypesÚint8Úint16Úint32Úint64Úuint64Úboolc sL t d tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tt fd d tt fd
|
d tt fd d tt fd d tt fd d tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tt fdd tddggtt fdd tt fdd tt fd d tt fd d tt fd d tt fd d tt fd d tt fd!d tt fd"d tt fd#d d S )$N©é é c s d d
S ©Nr © r ©Úar ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/tests/test_array_object.pyÚ<lambda>) ó z%test_validate_index.<locals>.<lambda>c s d d
S ©Néüÿÿÿr r r r r r * r c s d dd
S )Nr éÿÿÿÿr r r r r r + r c s d dd
S )Néûÿÿÿr# r r r r r r , r c s dd
S r r r r r r r - r c s dd
S r! r r r r r r . r c s dd d
S )Nr r# r r r r r r / r c s dd d
S )Nr" r# r r r r r r 0 r c s dd d
f S ©N.é r r r r r r 2 r c s dd d
f S ©N.r$ r r r r r r 3 r c s dd dd
f S ©N.r& r# r r r r r r 4 r c s dd dd
f S )N.iúÿÿÿr# r r r r r r 5 r c s ddd
f S r% r r r r r r 6 r c s ddd
f S r' r r r r r r 7 r c s ddd d
f S r( r r r r r r 8 r c s ddd d
f S )N.r$ r# r r r r r r 9 r c s d d
df dkdf S ©Nr é r r r r r r < r c s d d
df dkdf S )Nr r* .r r r r r r = r c s d d dkf S )N.r r* r r r r r r > r c s dddg S )NTr r r r r r ? r c s d S )N))TTTr r r r r r @ r r r* c s S ©Nr r ©r Úidxr r r D r c s
|
f S r+ r r r, r r r E r c s ddg S r) r r r r r r F r c s d S )N)©r r* r. r r r r r r G r c s ddg S r) r r r r r r H r c s t ddgg¡ S r) )ÚnpÚarrayr r r r r I r c s d S )Nr r r r r r r L r c s d S )N©r r r r r r r M r c s d S ©Nr r r r r r r N r c s d d
S r+ r r r r r r O r )r r Ú
|
IndexErrorr r r r, r Útest_validate_index sB r4 c sB ddddddddddddddddddd} dd | ¡ D ]Ä\}g}d kr~d
|
d d
}d d d
}|||g7 }d D ]z|D ]n D ]ì|dks |dkrºjtks |dkrÎjtks |dkrâjtks |dkröjtks |dkrnjtkrnjtkr&ttks^jtkr@tt ks^jtkrntt
|
t fkrnt qt t fdd q|D ]n D ]^ D ]NjtkrÈjttttfksxjtkrèjttttfksxjtkr jtksxjtkrjtksxjtkr0jtksxjtkrHjtksxjtkr`jtksxjtkrjtkrt t fdd q¢ d ¡rÎtjjjkrÎt t fdd q¢|dkrjtkrðjtksÊjtkrjtksÊ|dkr*jtkr*jtksÊ|dkrLjtkrLjtksÊ|dkrjtkrnjtksÊjtkrjtksÊ|dkr¨jtkr¨jtksÊ|dkrÚjtkrÚjtkrÚt nt t fdd q¢qqqqq:ddddd}| ¡ D ]f\} D ]T|dkr@jtksV|dkrdjtkrdt nt t fdd q&q fdd}dD ]d D ]z| D ]ltt
|
t fkrÄjtksÞtt kröjtkröt tfdd nt t fdd q¢qq| D ] | D ] jtkrPjttttfksèjtkrpjttttfksèjtkrjtksèjtkr jtksèjtkr¸jtksèjtkrÐjtksèjtksèjtkr(t t fdd t t fd d t t fd d q*jd ksVjd ksVjd jd krÀt tfd!d t tfd"d tjjjkrªt t fd#d nt tfd$d nv ¡ ¡ tjjjkr t t fd%d n6jd jd kr,t tfd&d n
|
¡ q*q d S )'NÚnumericZinteger_or_booleanr ÚintegerZfloating)Ú__add__Ú__and__Ú__eq__Ú __floordiv__Ú__ge__Ú__gt__Ú__le__Ú
|
__lshift__Ú__lt__Ú__mod__Ú__mul__Ú__ne__Ú__or__Ú__pow__Ú
|
__rshift__Ú__sub__Ú __truediv__Ú__xor__c s sL t D ]} td| dV qtD ]} td| dV q tD ]} td| dV q4d S )Nr* ©ÚdtypeFç ð?)r r r r )Údr r r Ú _array_valsj s z#test_operators.<locals>._array_vals)r9 rB r= r; r? r< Z__rr Z__i)r* rK FÚbooleanc s t S r+ ©Úgetattrr ©Ú_opr Úsr r r r z test_operators.<locals>.<lambda>c s t S r+ rO r ©rR ÚxÚyr r r § r c s t S r+ rO r rT r r r r c s t S r+ rO r rT r r r º r )Ú__abs__Ú
|
__invert__Ú__neg__Ú__pos__c s t S r+ rO r )r Úopr r r Í r c 3 sJ D ]
|
} | V qt D ].}td|dV td|dV td|dV qd S )Nr rI )r r )r r )r r )r rL )rM r r Ú_matmul_array_valsÑ s
|
z*test_operators.<locals>._matmul_array_vals)Ú
|
__matmul__Ú __rmatmul__Ú __imatmul__c s t S r+ rO r rQ r r r á r c s t S r+ rO r rQ r r r ã r c s
|
¡S r+ ©r] r ©rU rV r r r ð r c s
|
¡S r+ ©r^ r ra r r r ñ r c s
|
¡S r+ ©r_ r ra r r r ò r r r* r c s
|
¡S r+ r` r ra r r r ô r c s
|
¡S r+ rb r ra r r r õ r c s
|
¡S r+ rc r ra r r r ÷ r c s
|
¡S r+ rc r ra r r r ù r c s
|
¡S r+ rc r ra r r r þ r c s
|
¡S r+ rc r ra r r r r )ÚitemsrJ r r r r r Útyper ÚintÚfloatrP r Ú TypeErrorr r r r r Ú
|
startswithr Ú
|
ValueErrorÚshaper] r^ r_ )Zbinary_op_dtypesZdtypesÚopsZropZiopZunary_op_dtypesr\ r )rM rR r r[ rS rU rV r Útest_operatorsQ s¬ î
|
|
|
|
ÿÿþþýýüüûû ÷
|
÷
|
ö
|
|
ö õ õ
|
ÿÿþþýýüüûûúúùù ÿþ"ÿÿþþþýýýüüüûûúúúùùù (ü
|
ÿþýü
|
|
ÿÿ" ÿÿþþýýüüûûúù .
|
|
rm c s& t d t dt dt dks(ttdks8ttdksHtt ¡dksZtttdd ttdd ttdd ttdd tt fd d tt fd
|
d tt fd d tt fd d tt fd d tt fdd tt fdd ttfdd d S )NFr ç c S s t tdgS )NF)r r r r r r r r z0test_python_scalar_construtors.<locals>.<lambda>c S s t tdgS r2 )rf r r r r r r r c S s t tdgS )Nrn )rg r r r r r r r c S s t tdg¡S r2 )ÚoperatorÚindexr r r r r r r c s t S r+ ©r r ©Úir r r r c s t S r+ rq r ©Úfr r r r c s t S r+ ©rf r ©Úbr r r r c s t S r+ rv r rt r r r r c s t S r+ ©rg r rw r r r r c s t S r+ ry r rr r r r r c s
|
t ¡S r+ ©ro rp r rw r r r ! r c s
|
t ¡S r+ rz r rt r r r " r )
|
r r ÚAssertionErrorrf rg ro rp r rh rj r r )rx ru rs r Ú test_python_scalar_construtors s& r| c sp t d jdksttt d¡ s.ttt fdd ttt dd sZttt fdd d S )Nr Úcpuc s
|
d¡S )NÚgpu)Ú to_devicer r r r r * r z&test_device_property.<locals>.<lambda>©Údevicec s t ddS )Nr~ r )r r r r r r - r ) r r r{ r r r r rj r r r r r Útest_device_property% s r c s t d t d} tt fdd t| jts2t| jjdksBtt jtsRt jjdksbtt| jtsrt| jjdkstd S )N)r* r r ©r r c s j S r+ )ÚTr r r r r 2 r z'test_array_properties.<locals>.<lambda>)r r )r* r r ) r r rj Ú
|
isinstancer r
|
r{ rk ZmTrw r r r Útest_array_properties/ s r c C sf t dtd} t | ¡| jks ttj| tjd}t t |tj dtjd¡¡sRt|j tjksbtd S )Nr rI )
|
r r r/ r Z_arrayr{ Úfloat64r r rJ )r rx r r r Útest___array__< s
|
"r c C s* t d} | d d d
f }|jdks&td S )Nr& )r* r& )r rk r{ )r Z indexed_ar r r Útest_allow_newaxisC s r c C s* t d} t t¡ | d W 5 Q R X d S )N©r r r )Nr r ©r ÚpytestÚraisesr3 r r r r Ú(test_disallow_flat_indexing_with_newaxisH s r c C s2 t d} t t¡ | d tdf W 5 Q R X d S )Nr T)r r r r3 r r r r r Ú test_disallow_mask_with_newaxisM s r rk r )r& r rp ÚstringFTc C s* t | }t t¡ || W 5 Q R X d S r+ r )rk rp r r r r Útest_error_on_invalid_indexR s r c C s t d} | td d S )Nr T)r r r r r r Ú!test_mask_0d_array_without_errorsY s r rs r& r* c C sP t d}t t¡ |d| f W 5 Q R X t t¡ || df W 5 Q R X d S )Nr .r )rs r r r r Ú)test_error_on_invalid_index_with_ellipsis] s
|
r c C sX t dtd} | | jdks tt dtd} t dtd}t t¡ | | W 5 Q R X dS )aq
|
Indexing operations convert array keys before indexing the internal array
|
|
Fails when array_api array keys are not converted into NumPy-proper arrays
|
in __getitem__(). This is achieved by passing array_api arrays with 0-sized
|
dimensions, which NumPy-proper treats erroneously - not sure why!
|
|
TODO: Find and use appropriate __setitem__() case.
|
)r r rI r1 N)r Úbool_rk r{ r r r3 )r Úkeyr r r Ú!test_array_keys_use_private_arrayg s
|
r )-ro Z numpy.testingr Únumpyr/ r Ú r r r r r r Z _array_objectr
|
Z_dtypesr r r r r r r r r r r r r r4 rm r| r r r r r r ÚmarkZ parametrizer r Úslicer r r r r r Ú<module> s4 88 6
|
ÿ
|
|
