(@,J
d dl Z d dlmZmZ d dlZd dlZddlmZm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z d Z!d Z"d Z#d
Z$d Z%d Z&d
Z'd Z(d Z)ej* + dg d ej* + dg d d Z,d Z-ej* + d e.d e.dd e d e d dg g d Z/d Z0dS ) N)
assert_raisessuppress_warnings )onesasarrayreshape result_typeallequal)Array) _all_dtypes_boolean_dtypes_real_floating_dtypes_floating_dtypes_complex_floating_dtypes_integer_dtypes_integer_or_boolean_dtypes_real_numeric_dtypes_numeric_dtypesint8int16int32int64uint64boolc t d t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd
t t fd t t fd t t fd
t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t ddgg t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd t t fd! t t fd" d S )#N c d d S Nr as {/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/tests/test_array_object.pyz%test_validate_index.., s ae c d d S Nr" r# s r% r& z%test_validate_index..- s af r' c d dd S )Nr r" r# s r% r& z%test_validate_index... s a2h r' c d dd S )Nr, r" r# s r% r& z%test_validate_index../ s aBi r' c dd S r! r" r# s r% r& z%test_validate_index..0 s ae r' c dd S r) r" r# s r% r& z%test_validate_index..1 s af r' c dd d S )Nr r, r" r# s r% r& z%test_validate_index..2 s a2h r' c dd d S )Nr* r, r" r# s r% r& z%test_validate_index..3 s aBi r' c dd df S N. r" r# s r% r& z%test_validate_index..5 s aBQBi r' c dd df S N.r. r" r# s r% r& z%test_validate_index..6 s aCRCj r' c dd ddf S N.r5 r, r" r# s r% r& z%test_validate_index..7 s aEQrE l r' c dd ddf S )N.ir, r" r# s r% r& z%test_validate_index..8 s aFRF
m r' c ddd f S r4 r" r# s r% r& z%test_validate_index..9 s aABBi r' c ddd f S r7 r" r# s r% r& z%test_validate_index..: s aBCCj r' c ddd df S r9 r" r# s r% r& z%test_validate_index..; s aAErE l r' c ddd df S )N.r. r, r" r# s r% r& z%test_validate_index..< s aBFF
m r' c 4 d d df dk df S Nr r" r# s r% r& z%test_validate_index..? s a!!!A# ! n r' c 4 d d df dk df S )Nr rA .r" r# s r% r& z%test_validate_index..@ s a!!!A# #
&6 r' c , d d dk f S )N.r rA r" r# s r% r& z%test_validate_index..A s aQqT1W o r' c g d S )NTTTr" r# s r% r& z%test_validate_index..B s a(:(:(:&; r' c d S )N)rE r" r# s r% r& z%test_validate_index..C s
a(;&< r' r rA c S Nr" r$ idxs r% r& z%test_validate_index..G s af r' c f S rH r" rI s r% r& z%test_validate_index..H s
ag r' c ddg S r@ r" r# s r% r& z%test_validate_index..I aAi r' c d S )N)r rA rO r" r# s r% r& z%test_validate_index..J s a&7 r' c ddg S r@ r" r# s r% r& z%test_validate_index..K rM r' c > t j ddgg S r@ )nparrayr# s r% r& z%test_validate_index..L s a1a&(:(:&; r' c d S )Nr" r" r# s r% r& z%test_validate_index..O ae r' c d S )Nr r" r# s r% r& z%test_validate_index..P rU r' c d S Nr r" r# s r% r& z%test_validate_index..Q s ad r' c d d S rH r" r# s r% r& z%test_validate_index..R s ad r' )r r
IndexErrorr rI s @@r% test_validate_indexr\ s4
V A *mmmm,,,*nnnn---* . . . .///* / / / /000*mmmm,,,*nnnn---* . . . .///* / / / /000* / / / /000* 0 0 0 0111* 2 2 2 2333* 3 3 3 3444* / / / /000* 0 0 0 0111* 2 2 2 2333* 3 3 3 3444 * 4 4 4 4555* 6 6 6 6777* 5 5 5 5666* ; ; ; ;<<<* < < < <=== Aq6(
C*nnnnn---*ooooo...* / / / /000* 7 7 7 7888* / / / /000* ; ; ; ;<<< *mmmm,,,*mmmm,,,*llll+++*llll+++++r' c
i ddddddddd dd
dd dd d
dddddddddddddd
ddddddi} d t d }| D ]\ } g} dvr!d d d z }d d d z }|||gz
}d d d!|d"fD ]
|D ]ۊ
D ]ъ |dk sz|dk r j t v sf|dk r j t v sR|d
k r j t
v s>|dk r j t v s*|d#k r j t v s|dk rC j t v r4 j t v rt
t k s j t
v rt
t k s\ j t v r t
t t fv s1 j t v rt
t t t fv r j t
v r"
|k r t t
fd$ [t# 5 }| t& d% t)
d d d n # 1 swxY w Y t t*
fd& |D ]
D ] D ]يj t, k r!j t. t0 t2 t4 fv sىj t, k r!j t. t0 t2 t4 fv sj t
v rj t
vsj t
v rj t
vspj t v rj t vsTj t v rj t vs8j t v rj t vs j t v r*j t vr t t*
fd' )
d r?t9 j j j k r t t*
fd( }|dk r9j t v rj t v sj t v rj t v s|dk r j t v rj t v s|dk r j t v rj t v s|d
k r j t
v rj t
v s|dk r8j t
v rj t
v s`j t v rj t v sD|d#k r j t v rj t v s"|dk r7j t v r)j t v r t)
t t*
fd) ېݐddddd*}| D ]l\ } D ]\ |dk r j t v s|dk r' j t v r t) Ct t* fd+ ]m fd,}d-D ]
d.D ]
| D ] t
t t fv r j t v s&t
t k r) j t
v rt t:
fd/ nt t*
fd0 | D ] | D ]j t, k r!j t. t0 t2 t4 fv sj t, k r!j t. t0 t2 t4 fv sj t
v rj t
vspj t
v rj t
vsTj t v rj t vs8j t v rj t vs j t v sj t v rMt t* fd1 t t* fd2 t t* fd3 >j d4k s'j d4k s j d j d5 k rt t: fd6 t t: fd7 t9 j j j k rt t* fd8 t t: fd9 t9 j j j k rt t* fd: cj d5 j d k rt t: fd; ! d S )._array_valsl s & &A!1%%% % % % % * *A%q))) ) ) ) )! ( (A#Q''' ' ' ' ' ( (r' gꌠ9Y>)F)rb rl rf rd ri re __rr __irA rw y ?Fbooleanc 6 t S rH getattr_opr$ ss r% r& z test_operators.. C QRASAS r' z"invalid value encountered in powerc 6 t S rH r~ r s r% r& z test_operators.. s C 9K9K r' c 6 t S rH r~ r xys r% r& z test_operators.. r r' c 6 t S rH r~ r s r% r& z test_operators.. r r' c 6 t S rH r~ r s r% r& z test_operators.. r r' )__abs__
__invert____neg____pos__c 4 t S rH r~ )r$ ops r% r& z test_operators.. s B 1A1A r' c 3 K D ]} | V t D ];}t d| V t d| V t d| V ._matmul_array_vals s AGGGG ( (AvQ''' ' ' 'vQ''' ' ' 'vQ''' ' ' ' ' ( (r' )
__matmul__ __rmatmul__ __imatmul__)rA rw Fc 6 t S rH r~ r s r% r& z test_operators.. s ogaooa6H6H r' c 6 t S rH r~ r s r% r& z test_operators.. s _WQ__Q5G5G r' c . S rH r r r s r% r& z test_operators.. s a r' c . S rH r r s r% r& z test_operators.. q1A1A r' c . S rH r r s r% r& z test_operators.. r r' r" r c . S rH r r s r% r& z test_operators.. s !,,q// r' c . S rH r r s r% r& z test_operators.. s !--2B2B r' c . S rH r r s r% r& z test_operators.. Q]]15E5E r' c . S rH r r s r% r& z test_operators.. ammA6F6F r' c . S rH r r s r% r& z test_operators..
r r' c . S rH r r s r% r& z test_operators..
r r' )"intitemsrv r r r r r r typer r floatr complexr
OverflowErrorr filterRuntimeWarningr TypeErrorr r r r r
startswithr
ValueErrorshaper r r )binary_op_dtypesBIG_INTdtypesopsropiopsupunary_op_dtypesr ry r r$ r r r r s @@@@@@@r% test_operatorsr T s 9' %
. .
. i . > 9 % & 9 i 9!" z#$ '% *( ( ( $iiG&,,.. PU PU
Fd
Q Q Q"QRR&.C"QRR&.C C: CS"gu- J U J UA I
U I
U$ "M "MA 5 "i / /AG4N4N "n 4 4DX9X9X "i / /AG4N4N "&: : :qwJd?d?d "i / /AG4N4N "j 0 0QW@P5P5P W 7 7DGGtOO !? : :tAww#~~ !+@ @ @T!WWQVX[P\E\E\ !+C C CQT[]bdgShHhHh 7o55!w,, )-9S9S9S9S9S9S T T T T "3!4!4 3 #
>+O!Q !Q !Q /3 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 &i1K1K1K1K1K1KLLLL #U #UC ([]] "U "U!, !U !UA !"6 1 1ag$uV[A\6\6\#$7f#4#4T5RWY^D_9_9_#$7o#=#=!'Q`B`B`#$7o#=#=!'Q`B`B`#$7o#=#=!'Q`B`B`#$7o#=#=!'Q`B`B`#$7.>#>#>17RbCbCb#$7.>#>#>17RbCbCb -i9S9S9S9S9S9S T T T T !$u 5 5 U$/$A$AQW$L$L -i9S9S9S9S9S9S T T T T"(E//qw/7Q7QVWV]apVpVp9:O9S9SXYX_crXrXr$*n$<$<L`A`A`efel qE fE fE$*i$7$7AG. s T'5'*:*:%;%; r' c < t t dg S rY )r r r" r' r% r& z0test_python_scalar_construtors.. s S! %6%6 r' c < t t dg S )Nr )r r r" r' r% r& z0test_python_scalar_construtors.. s U7C5>>%:%: r' c < t t dg S )Nr )r r r" r' r% r& z0test_python_scalar_construtors..! s WWbT]]%;%; r' c F t j t dg S rY )operatorindexr r" r' r% r& z0test_python_scalar_construtors.." s X^GQCLL%A%A r' c " t S rH )r cs r% r& z0test_python_scalar_construtors..) s
SVV r' c " t S rH )r r s r% r& z0test_python_scalar_construtors..- s
U1XX r' c , t j S rH r r )bs r% r& z0test_python_scalar_construtors..4 X^A%6%6 r' c , t j S rH r )fs r% r& z0test_python_scalar_construtors..5 r r' c , t j S rH r r s r% r& z0test_python_scalar_construtors..6 r r' ) r r r r r r r r r )ir r r s @@@r% test_python_scalar_construtorsr s A
A A A 77e q66Q;;;; 88s???? >! ! ! ! ! ) ; ;<<<) 6 6777) : :;;;) ; ;<<<) A ABBB 77d1gg < < < <a < < < ! ! ! ! !) 6 6 6 6777) 6 6 6 6777) 6 6 6 677777r' c ` t d j dk sJ t t d sJ t t fd t t t d sJ t t fd d S )Nr cpuc . d S )Ngpu) to_devicer# s r% r& z&test_device_property..> s akk%&8&8 r' devicec & t d S )Nr r )r r# s r% r& z&test_device_property..A s ga&>&>&> r' )r r r
r r r r r r# s @r% test_device_propertyr 9 s V A 8u uQ[[''++ , ,,, ,* 8 8 8 8999 uWQu---q11 2 222 2* > > > >?????r' c t d t d } t t fd t | j t
sJ | j j dk sJ t j t
sJ j j dk sJ t | j t
sJ | j j dk sJ d S )N)rA r r r r c j S rH )Tr# s r% r& z'test_array_properties..F s ac r' )r r )rA r r )r r r
isinstancer r r mT)r r$ s @r% test_array_propertiesr C s YA V A*kkkk*** ac5 ! !!! ! 39 adE " """ " 4: " " " " adE " """ " 4: r' c h t dt } t j | | j u sJ t j | t j }t j t j |t j dt j sJ |j t j k sJ d S )Nr ru ) r r rR r _arrayfloat64r
r rv )r$ r s r% test___array__r P s V5!!!A
:a==AH $ $ $ $
1BJ'''A
6"(1bgfBJ ? ? ?@@ A AAA A 7bj r' c V t d } | d d d f }|j dk sJ d S )Nr5 )rA r5 )r r )r$ indexed_as r% test_allow_newaxisr W s7 QA$'
I ?f $ $ $ $ $ $r' c t d } t j t 5 | d d d d d S # 1 swxY w Y d S )Nr r r )Nr r r pytestraisesr[ r# s r% (test_disallow_flat_indexing_with_newaxisr \ s YA z " " *
?AAc t d } t j t 5 | d t d f d d d d S # 1 swxY w Y d S )Nr T)r r r r[ r r# s r% test_disallow_mask_with_newaxisr a s YA z " " $
s AAAr )r" )r5 r r )stringFTc t | }t j t 5 || d d d d S # 1 swxY w Y d S rH r )r r r$ s r% test_error_on_invalid_indexr f s
U A z " " % r c N t d } | t d d S )Nr" T)r r r# s r% !test_mask_0d_array_without_errorsr m s$ RAgdmmr' r r5 TrA c t d }t j t 5 |d| f d d d n # 1 swxY w Y t j t 5 || df d d d d S # 1 swxY w Y d S )Nr .r )r r$ s r% )test_error_on_invalid_index_with_ellipsisr q s
YA z " " #q& z " " !S& s# A AA$ A<<B B c $ t dt } | | j dk sJ t dt } t dt }t j t
5 | | ddd dS # 1 swxY w Y 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 ru rW N)r bool_r r r r[ )r$ keys r% !test_array_keys_use_private_arrayr { s
V5!!!A Q4: T A
vU
#
#
#C z " " # s / BB B )1r
numpy.testingr r numpyrR r r r r r r
r
_array_objectr _dtypesr
r r r r r r r r r r r r r r r r\ r r r r r r r r mark parametrizer r slicer r r" r' r% r s : : : : : : : :
> > > > > > > > > > > > > > > > ! ! ! ! ! ! &6, 6, 6,p{% {% {%|$8 $8 $8N@ @ @ ! ! !% % %
"7"7"788"9"9"9:: ;: 98
%%((EE!QKK A @ r'