U
|
O ±dQ ã @ s: d dl Z d dlZd dlmZmZmZmZ G dd dZdS )é N)Úassert_Ú assert_raisesÚassert_array_equalÚ HAS_REFCOUNTc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestTakec C s¼ ddgddgg}ddgddgg}d d
|
d g}d dg}t jd t jdt jt t jdt jdt jdg}dtddd dd ddd}ttt dg¡f}|D ] }t jt |t j
|
¡r²|n||d} t | j ¡ }
|
|D ]æ} | jd krüd| j |
|
d _d| j |
|
d _|D ]²} |D ]¦} || | }|tkrP| jd krP| d | ¡ tt| j| | dd n\| jd kr| d | ¡ | j| | dd}t||
|
| n$| j| | dd}t|jd| j k qq qÎqd S )Né é é é ó 1ó 2ó 3ó 4ÚraiseÚwrapÚclipéÿÿÿÿr ©Údtype)r r )r r
|
)r r r )Ú Úi2r )r )ÚmodeZaxis)ÚnpÚemptyZintpÚtupleÚ
|
IndexErrorÚintÚobjectr ÚarrayZ
|
issubdtypeÚnumberÚlistÚTÚcopyÚsizeÚshapeÚputr Útaker r )ÚselfÚaZa_strÚmodesÚindicesZ index_arraysZ real_indicesÚtypesÚtÚtaZtresultZ index_arrayr ÚindexZ
|
real_indexÚres© r0 ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_item_selection.pyÚ test_simple
|
sF
|
þþ
|
|
|
ÿ zTestTake.test_simplec C sÀ dd t dD }dD ]¤}t |¡}t ddddddg¡}|j||d d
|d
|
~trnttd d |D t |d ¡d d d
}|j||d d
|d
|
~trttd d |D qd S )Nc S s g | ]
|
}t qS r0 )r )Ú.0Úir0 r0 r1 Ú
|
<listcomp>2 s z-TestTake.test_refcounting.<locals>.<listcomp>é
|
)r r r r r
|
é r é )Úoutr c s s | ]}t |¡d kV qdS ©r N©ÚsysÚ getrefcount©r3 Úor0 r0 r1 Ú <genexpr>9 s z,TestTake.test_refcounting.<locals>.<genexpr>c s s | ]}t |¡d kV qdS r: r; r> r0 r0 r1 r@ ? s )Úranger r r& r r Úall)r' Zobjectsr r( Úbr0 r0 r1 Útest_refcounting1 s
|
zTestTake.test_refcountingc C s* t d¡}d d¡}tt|jd|d d S )Nr6 s äÚUTF8r7 )r )r ZarangeÚdecoder Ú
|
ValueErrorr& )r' ÚdÚkr0 r0 r1 Útest_unicode_modeA s
|
|
zTestTake.test_unicode_modec C sB t ddddddg¡}| ¡ }| t jg t jd¡ t|| d S )Nr r r
|
r8 é r6 r )r r r" Ú partitionÚint16r )r' Z
|
a_originalr( r0 r0 r1 Útest_empty_partitionF s z TestTake.test_empty_partitionc C sP t ddddddg¡}| t jg t jd¡}t dddd dd
|
g¡}t|| d S ) Nr r r
|
r8 rK r6 r r r r7 )r r Z argpartitionrM r )r' r( rC r0 r0 r1 Útest_empty_argpartitionP s z TestTake.test_empty_argpartitionN)Ú__name__Ú
|
__module__Ú __qualname__r2 rD rJ rN rO r0 r0 r0 r1 r s
|
'
|
r ) r< Únumpyr Z numpy.testingr r r r r r0 r0 r0 r1 Ú<module> s
|
