1j: d dl Z d dlZd dlZd dlmZmZmZmZ G d d Z G d d Z
G d d Z dS ) N)assert_
assert_raisesassert_array_equal HAS_REFCOUNTc , e Zd Zd Zd Zd Zd Zd ZdS )TestTakec ddgddgg}ddgddgg}g d }d
dg}t j d t j t j t t j t j d
t j g}dt ddd dd ddd }t
t t j dg f}|D ]}t j t j |t j
r|n|| } t | j
}
|D ]!} | j d k r*d| j z |
d _ d| j z |
d _ |D ]} |D ]}
|| |
}|t u r@| j d k r5| d |
t# t | j | | d Y| j d k rE| d |
| | | d }t' ||
| | | | d }t) |j d| j z k #d S )N 1 2 3 4raisewrapclipr dtype)r
r
)r r
) i2r )r )modeaxis)npemptyintptuple
IndexErrorintobjectr array
issubdtypenumberlistTcopysizeshapeputr taker r )selfaa_strmodesindices index_arrays real_indicestypesttatresult index_arrayr index
real_indexress x/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_item_selection.py test_simplezTestTake.test_simple s VaV d|,)))q'"'222888rw7779 '(J!7!7%&1
%&1
/ / VRX}o 6 66 K KAr}Q : :E!!QOOOB2499;;''G +
K
K #q(('+k.?'?GAJ$'+k.?'?GAJ$ !
K
KD!( K K%1$%7%>
%33 8HA8M8M 'OOAu 5 5 5 )*bg{/3! = = = = = ( - 2 2 'OOAu 5 5 5"$''+Dq'"I"IC .sGJ4G H H H H"$''+Dq'"I"IC #CI 8I1I$I J J J J K
K
K K K c d t d D }dD ]}t j | }t j g d }| ||d d | ~t r&t t
d |D t j |dz d d d }| ||d d | ~t r&t t
d |D d S )
Nc * g | ]}t S )r# ).0is r=
z-TestTake.test_refcounting..4 s ///688///r?
)r r r )r r r
r rG )outr c 3 F K | ] }t j | d k V dS r Nsys getrefcountrC os r= z,TestTake.test_refcounting..; 1 EECOA . .! 3EEEEEEr? r c 3 F K | ] }t j | d k V dS rK rL rO s r= rQ z,TestTake.test_refcounting..A rR r? )ranger r$ r- r r all)r. objectsr r/ bs r= test_refcountingzTestTake.test_refcounting3 s //U2YY///- G GD!!A+++,,A
FF1!BQB%dF + + +
GEEWEEEEEFFF1%%ccc*A
FF1!BQB%dF + + +
GEEWEEEEEFFF G Gr? c t j d }d d }t t |j d| d S )NrF s äUTF8rG r )r arangedecoder
ValueErrorr- )r. dks r= test_unicode_modezTestTake.test_unicode_modeC sA IbMM v & &j!&!!444444r? c t j g d }| }| t j g t j t || d S )Nr r r
rH rF r )r r$ r) partitionint16r )r.
a_originalr/ s r= test_empty_partitionz TestTake.test_empty_partitionH sb X 1 1 122
OO
BHRrx0001111j)))))r? c t j g d }| t j g t j }t j g d }t || d S )Nrc r )r r
r r r
rG )r r$ argpartitionrf r )r. r/ rW s r= test_empty_argpartitionz TestTake.test_empty_argpartitionR sf H((( ) )
NN28Bbh777 8 8 H''' ( (1a r? N)__name__
__module__ __qualname__r> rX ra rh rk rB r? r= r r sf %K %K %KNG G G 5 5 5
* * *! ! ! ! !r? r c T e Zd Zej d eej d dgz d Z ej d eej d dd dgz ej dg d d Z
dS )
TestPutMaskr Alli,Oc * | dk r|dz
}t j d | }t j dd t }t j d|j }| }t j
||| t || |d t | | t || || d S )Nm8[ns] r r r* )
lowerr r\ astyperandomrandintboolzerosr r) putmaskr len)r. r valsmaskarrr~ s r= r> zTestPutMask.test_simple\ s ;;==C W E y%%E%22y ..55d;;ht4:...
3d###3t9d:CII:&6t&<===3u:udU|44444r? r
Nr r c t j d| }| }t j dd t }t j ||g t || d S )Nrw r r rx ) r r~ r) r{ r| rz r} r, r )r. r r r arr_copyr s r=
test_emptyzTestPutMask.test_emptym sy ht5)))88::y ..55d;; sD" 3)))))r? rl rm rn pytestmark parametrizer' r typecodesr> r rB r? r= rp rp [ s [ Wdd2<+>&?&?5'&IJJ5 5 KJ5 [ Wdd2<+>&?&?&Cug&MNN [ V%>%>%>??* * @? ON* * *r? rp c e Zd Zej d eej d dd dgz ej dg d d Z ej d eej d dd dgz ej dg d d Z
dS )
TestPutr rq r
Nrr r r c | dk r|dz
}t j d | }t j d|j }| }|dk rst j t | dz
d d d z }d
|d < t | d z
|d < | }d
|d < d|d < nQt j t | dz
d d }|}|dk r|t | z }t j
|||| t || |d t | t j t | t }d||< t || |d | d S )Nrt ru rv r rw r r i r
r r ii) r r r[ F)ry r r\ rz r~ r r) r{ permutationr r, r onesr} sum) r. r r r r r~ indxindx_put untoucheds r= r> zTestPut.test_simplez s ;;==C W E y%%E%22 ht4:...
6>>9((SA 66uu=ADDH3xx!|DHyy{{H HRL HRLL 9((SA 66uu=DHv~~#c#hh.
sHd....3t9d:CII:&6777GCHHD111 $3y>51A)--//1A+BCCCCCr? c t j d| }| }t j |g dg t || d S )Nrw r )r
r r )r r~ r) r, r )r. r r r r s r= r zTestPut.test_empty sU ht5)))88:: sIIIr"""3)))))r? r rB r? r= r r y s [ Wdd2<+>&?&?&Cug&MNN [ V%>%>%>?? D D @? ON DB [ Wdd2<+>&?&?&Cug&MNN [ V%>%>%>??* * @? ON* * *r? r ) rM r numpyr
numpy.testingr r r r r rp r rB r? r= r s
M! M! M! M! M! M! M! M!`* * * * * * * *<,* ,* ,* ,* ,* ,* ,* ,* ,* ,*r?