U
|
P ±dGQ ã @ sô d dl Z d dlZd dlmZmZmZmZmZm Z m
|
Z
|
d dl m Z m Z mZmZmZmZmZmZmZmZmZ G dd dZG dd dZG dd d ZG d
|
d d ZG d d d ZG dd dZ dd Z G dd dZ dd Z G dd dZ dd Z!dS )é N)Úassert_Ú assert_equalÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚ assert_raisesÚassert_raises_regex) ÚmgridÚogridÚ ndenumerateÚ fill_diagonalÚ diag_indicesÚdiag_indices_fromÚ index_expÚndindexÚr_Ús_Úix_c @ sh e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j ddddg¡dd Z dd ZdS )ÚTestRavelUnravelIndexc C sª t t dd¡d t tjdddd tt tjddd W 5 Q R X tt tjddd W 5 Q R X tt tjddd W 5 Q R X tt tjddd
|
W 5 Q R X t t dd¡d t t dd¡d t t d d¡d tttjd d tttjd d tttjdd tttjdd tttjdd tttjdd tttjdd tttjdd t t dd¡dddg t t dddgd¡d t dddgdddgg¡}t t |d¡dd d g t tj|dd d d d d!g t tj|d"d#d$dd%d&g t tj|d'd(d$d)d!d!g t t d*d+¡d, t t t dd d g¡d¡dddgdddgg t tjt d d d!g¡dd d dddgdddgg t t d,d+¡ddddg d S )-Né )r r ©é r )ÚindicesÚshape)r Úhape)r éþ )é é^ )Zims)Zdims)r éB éÿÿÿÿg à?é )éýÿÿÿr )r r )r r! )r r )ç¹?g é. )r é é r r$ r% é )é r% é é) é% ÚF©Úorderé é )r r% Úclip©Úmodeé é ©r r )r0 Úwrapé )r$ r r r )r% r' é é iU )r ÚnpÚ unravel_indexr Ú TypeErrorÚravel_multi_indexÚ
|
ValueErrorÚarray)ÚselfÚarr© rB úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_index_tricks.pyÚ
|
test_basic sh ÿþ
|
|
|
|
ÿ ÿ ÿÿÿþz TestRavelUnravelIndex.test_basicc C s d}d}t t|tjg d t t|tjdd t t|tjt g ¡d tt tjg tdd¡g g g g t t|tjg g fd t t|tjg dgfd t t|tjt g ¡t g ¡fd tt tjg tdtjg tdfd¡g tt tjg g gtdd¡g d S ) Nz9indices must be integral: the provided empty sequence waszonly int indices permitted)é
|
r$ r& rB ©Údtype©rE r$ Úabc)r& r$ )r r< r: r; r? r Úintr= )r@ Zmsg1Zmsg2rB rB rC Útest_empty_indicesF s> ÿÿÿÿ
|
ÿ ÿÿÿÿz(TestRavelUnravelIndex.test_empty_indicesc C sê t jt jkrPddgddgddgddgdd gddgd
|
dgf}tt |d ¡d d g ttt jdd d
|
gd
|
gf}t t j¡j d }tt ||df¡d
|
g ttt j||d df tt j||dfddd
|
g ttt j||d dfdd d S )Nr é r$ r& éu r4 r iK i r )r) r' éx é$ i
|
r8 r% l ¼zá l nm )l ÿÿ l r+ r, )
|
r: ÚintpÚint64r r= r r> r; ZiinfoÚmax)r@ rA Z dummy_arrZhalf_maxrB rB rC Útest_big_indicesZ s@ ÿ
|
þ ÿ
|
ÿ ÿ
|
ÿz&TestRavelUnravelIndex.test_big_indicesc
|
C s t jt jt jt jt jt jfD ]v}t jddddddgddddddgg|d}d}d |d |d }tt ||¡| t|t
|
||¡ |d d
|
|d }tt j ||d d | t|t j
|
||d d t jddddddgddddddgddddd d
|
gg|d}d}dd |d |d |d }tt ||¡| t|t
|
||¡ |d d
|
|d d |d }tt j ||d d | t|t j
|
||d d q d S )Nr r r r$ r r% rF )r& r8 r8 r& r+ r, r9 )r& r8 rE rE ) r: Úint16Zuint16Úint32Zuint32rQ Zuint64r? r r= r; )r@ rG Zcoordsr ZuncoordsrB rB rC Ú test_dtypesr sB ÿ
|
ÿ ÿ,þ ÿz!TestRavelUnravelIndex.test_dtypesc C sx t tjddddgdddt ddddgd¡ t tjddddgdd dt ddd
|
dgd¡ tttjddddgd d S ) Nr& r r r )r r$ r' r7 r6 r1 r% )r6 Úraiser0 rW r )r r: r= r r> ©r@ rB rB rC Útest_clipmodes s þÿý
|
ÿz$TestRavelUnravelIndex.test_clipmodesc C s2 t dddgd¡\}}t|jj t|jj d S )Nr r r$ )r r& )r: r; r ÚflagsZ writeable)r@ ÚxÚyrB rB rC Útest_writeability s z'TestRavelUnravelIndex.test_writeabilityc C sB t dd¡}t|d ttdt jdgd ttdt jdgd d S )Nr rB z0d arrayz out of boundsr )r: r; r r r> ©r@ r[ rB rB rC Útest_0d s
|
ÿz TestRavelUnravelIndex.test_0dr2 r0 r6 rW c C s^ t jt jdt jdd|d}|jdks*ttt" t jt jdt jdd|d W 5 Q R X d S )N)r$ r rF ©r r r r1 ©r )r$ r )r: r= ÚzerosrP r ÚAssertionErrorr r> )r@ r2 ÚresrB rB rC Útest_empty_array_ravel¨ s ÿ
|
ÿz,TestRavelUnravelIndex.test_empty_array_ravelc C sd t t jdt jdd¡}t|dks(ttdd |D s>ttt t dgd¡ W 5 Q R X d S )Nr rF r` r$ c s s | ]}|j d kV qdS )ra N)r )Ú.0ÚarB rB rC Ú <genexpr>¶ s zATestRavelUnravelIndex.test_empty_array_unravel.<locals>.<genexpr>r ) r: r; rb rP Úlenrc Úallr r> )r@ rd rB rB rC Útest_empty_array_unravel² s
|
|
z.TestRavelUnravelIndex.test_empty_array_unravelN)Ú__name__Ú
|
__module__Ú __qualname__rD rK rS rV rY r] r_ ÚpytestÚmarkÚ parametrizere rk rB rB rB rC r s 7
|
r c @ s^ e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z dd Z dd Z dS )ÚTestGridc C s² t ddd
}t ddd
}t|jdk t|jdk t|d dk t|d d t|d dk t|d |d dd t|d |d d d t|d |d d
|
d d S ) Nr r ù $@r" )rE )é r é ggfffffþ?g Çq ÇqÌ?)r r r r )r@ rg ÚbrB rB rC rD ¼ s zTestGrid.test_basicc C s8 t jdddd\}}t|d t|tddd
d d S )Nr rE T)Zretstepg9/§àåÄ?y I@r/ )r: Zlinspacer r r )r@ r\ ÚstrB rB rC Útest_linspace_equivalenceÈ s
|
z"TestGrid.test_linspace_equivalencec C sV t ddd
ddd
f }t ddd
ddd
f }t|jdk t|jd k t|d
|
d
|
d d
f t d d ¡ t|d d d
d
|
f dt d d ¡ t|d
|
dd d
f t d d ¡d t|d d d
df dt d d ¡ d t|d
|
dd d
f |d
|
d
|
d d
f dt dd ¡ d t|dd d
df |dd d
d
|
f dt dd ¡ d d S )Nr r rs éþÿÿÿr r" gÉ?)r rE rE )r rt rt r rE Údru rt )r r r r r: Úonesr )r@ Úcrz rB rB rC Útest_ndÍ s $&$($ ÿ$ ÿzTestGrid.test_ndc C s\ t ddd
ddd
f }tddd
ddd
f }tj| }t||D ]\}}t|| qDd S )Nr r rs ry r )r r
|
r: Zbroadcast_arraysÚzipr )r@ Z grid_fullZ grid_sparseZgrid_broadcastÚfrv rB rB rC Ú test_sparseÛ s
|
|
zTestGrid.test_sparsezstart, stop, step, expected)NrE rs )éÈ rE )iöÿÿÿrt N)i é c C sJ t |||
|||
f }t |||
}t|j|d t|j|d d S )Nr r )r r Úsize)r@ ÚstartÚstopÚstepÚexpectedZgridZ
|
grid_smallrB rB rC Ú test_mgrid_size_none_handlingä s z&TestGrid.test_mgrid_size_none_handlingc C s t ddd
f }t t d¡t d¡t d¡
f }t|jtjk t|| t ddd
}t t d¡t d¡t d¡
}t|jtjk t|| d S )Nr" ç
ëQ¸ Õ?)r r: Úfloat32r rG Úfloat64r )r@ Úgrid64Zgrid32rB rB rC Útest_accepts_npfloatingô s "
|
z TestGrid.test_accepts_npfloatingc C sò t ddd
f }t t d¡t d¡t d¡
f }t|jtjk t|| t dt d¡d
}t dt d¡d
f }t|j|j kotjkn t||d t ddd
}t t d¡t d¡t d¡
}t|jtjk t|| d S )Nr" r r r y 333333 @)r r: Ú
|
longdoubler rG r r )r@ r Zgrid128Z
|
grid128c_aZ
|
grid128c_brB rB rC Útest_accepts_longdouble s( ÿ
|
" ÿz TestGrid.test_accepts_longdoublec C sø t tddd
f tddt d¡
f t tddd
tddt d¡
tddd
}tddd
f d }t|j|j kotjkn t|| tddt d¡
}tddt d¡
f d }t|j|j koâtj kn t|| d S )Nr" g333333Ó?y @y ffffff
|
@r )
|
r r r: Ú complex64r rG r r Z clongdoubler )r@ Zgrid64_aZgrid64_bZ grid128_aZ grid128_brB rB rC Ú test_accepts_npcomplexfloating s ÿ ÿ"
|
"z'TestGrid.test_accepts_npcomplexfloatingN)rl rm rn rD rx r} r ro rp rq r r r r rB rB rB rC rr » s þ
|
rr c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestConcatenatorc C s` t td t ddddddg¡ t d¡}t|dd|f }t |ddddddddddddg d S ) N)r r r$ r r& r% r r r$ r r& r% r )r r r: r? r{ )r@ rv r| rB rB rC Útest_1d/ s
|
zTestConcatenator.test_1dc C s" t ddd
f }t|jdk d S )Ng333333$@r rE Úf8)r r rG ©r@ ÚgrB rB rC Útest_mixed_type5 s z TestConcatenator.test_mixed_typec C s6 t dt dg¡t dddg¡df }t|jdk d S )Ng333333$Àr r r$ r g $@r )r r: r? r rG r rB rB rC Útest_more_mixed_type9 s $z%TestConcatenator.test_more_mixed_typec C sB t ddd
}t|jdk t ddt d¡
}t|jdk d S )Nr rO y Y@)éd )r r r r: r r rB rB rC Útest_complex_step= s z"TestConcatenator.test_complex_stepc C s¾ t j dd¡}t j dd¡}td||f }t|jdk t|d d
d d
f | t|d d
dd
f | t||f }t|jdk t|d d
d d
f | t|dd
d d
f | d S )Nr& Ú1)r& rE )rE r& )r: ÚrandomÚrandr r r r )r@ rv r| rz rB rB rC Útest_2dF s zTestConcatenator.test_2dc C sp t tdt d¡df dddg t tdddgt d¡f ddddg t tt d¡dddgf ddddg d S )Nr r r r$ )r r r: r? rX rB rB rC r_ R s &zTestConcatenator.test_0dN) rl rm rn r r r r r r_ rB rB rB rC r . s r c @ s e Zd Zdd ZdS )ÚTestNdenumeratec C s4 t ddgddgg¡}ttt|ddddg d S ) Nr r r$ r ))r r r ))r r r )r r$ ))r r r )r: r? r Úlistr ©r@ rg rB rB rC rD Y s
|
ÿzTestNdenumerate.test_basicN)rl rm rn rD rB rB rB rC r X s r c @ s e Zd Zdd Zdd ZdS )ÚTestIndexExpressionc C sJ t d¡}t|d d
|td d
t|d d
|td d
d S )Nr r )r: Úaranger r r r¡ rB rB rC Útest_regression_1` s
|
z%TestIndexExpression.test_regression_1c C s t j ddd¡}t|d d
d d
ddgf |td d
d d
ddgf t|d d
d d
ddgf |td d
d d
ddgf d S )Nr r& r% r$ r r )r: r r r r r r¡ rB rB rC Ú test_simple_1f s :z!TestIndexExpression.test_simple_1N)rl rm rn r¤ r¥ rB rB rB rC r¢ _ s r¢ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestIx_c C sb t td¡\}t|jt j t g ¡\}t|jt j t t jg t jd¡\}t|jt j d S )Nr rF )r: r Úranger rG rP r? r r¡ rB rB rC r¤ n s zTestIx_.test_regression_1c s d}t tjfD ]| tj fdd|D }tt||D ]P\\}}t|j | tt fddt|jD tt
|
|j tj ¡ q8qd S )N)r r& r$ r c s g | ] } |qS rB rB )rf Úsz)ÚfuncrB rC Ú
|
<listcomp>~ s z0TestIx_.test_shape_and_dtype.<locals>.<listcomp>c 3 s" | ]\}}| kr|d kV qdS )r NrB )rf ÚjÚsh)ÚkrB rC rh s z/TestIx_.test_shape_and_dtype.<locals>.<genexpr>) r§ r: r£ r Ú enumerater~ r r r rj Z
|
issubdtyperG Úinteger)r@ ÚsizesZarraysrg r¨ rB )r© r rC Útest_shape_and_dtypez s z TestIx_.test_shape_and_dtypec C s0 ddddg}t |¡\}tt |¡d | d S )NTFr )r: Znonzeror r )r@ Zbool_aZint_arB rB rC Ú test_bool s zTestIx_.test_boolc C s& dddgdddgg}t ttj| d S )Nr r r$ r r& r% )r r> r: r )r@ Zidx2drB rB rC Ú test_1d_only s zTestIx_.test_1d_onlyc C sR d}t |¡}t||}t|d j|df t|d jd|f t|j|f d S )Nr& r r )r: r£ r r r )r@ Zlength_of_vectorr[ ÚoutrB rB rC Útest_repeated_input s
|
|
zTestIx_.test_repeated_inputN)rl rm rn r¤ r± r² r³ rµ rB rB rB rC r¦ m s
|
|
r¦ c
|
C sN t jt dddgg¡ddt dddgg¡f } t| ddddddddgg d S )Nr r r$ r r r& r% )r: Zc_r? r )rg rB rB rC Útest_c_ s .r¶ c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )ÚTestFillDiagonalc C sB t dt¡}t|d t|t dddgdddgdddgg¡ d S )N)r$ r$ r& r ©r: rb rJ r r r? r¡ rB rB rC rD s
|
þÿzTestFillDiagonal.test_basicc C sz t dt¡}t|d t|t dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
|
¡ d S )NrH r& r r¸ r¡ rB rB rC Útest_tall_matrix¦ s
|
÷ÿz!TestFillDiagonal.test_tall_matrixc C s| t dt¡}t|dd t|t dddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg
|
¡ d S )NrH r& Tr r¸ r¡ rB rB rC Útest_tall_matrix_wrap¶ s ÷ÿz&TestFillDiagonal.test_tall_matrix_wrapc C sl t dt¡}t|d t|t ddddddddddg
|
ddddddddddg
|
ddddddddddg
|
g¡ d S )N)r$ rE r& r r¸ r¡ rB rB rC Útest_wide_matrixÆ s
|
þÿz!TestFillDiagonal.test_wide_matrixc C sF t dt¡}t|d t dddg¡}tt |dk¡||||f d S )N)r$ r$ r$ r$ r r r r )r: rb rJ r r? r Úwhere)r@ rg ÚirB rB rC Útest_operate_4d_arrayÏ s
|
z&TestFillDiagonal.test_operate_4d_arrayc C s0 t dt¡}ttd t|d W 5 Q R X d S )Nr$ ú at least 2-dr& ©r: rb rJ r r> r r¡ rB rB rC Útest_low_dim_handlingÕ s z&TestFillDiagonal.test_low_dim_handlingc C s0 t dt¡}ttd t|d W 5 Q R X d S )N)r$ r$ r' r$ ú equal lengthr rÀ r¡ rB rB rC Útest_hetero_shape_handlingÛ s z+TestFillDiagonal.test_hetero_shape_handlingN)
|
rl rm rn rD r¹ rº r» r¾ rÁ rà rB rB rB rC r· s r· c C sÈ t d} t ddddgddddgd d
|
d d gd dddgg¡}d|| < t|t ddddgddddgd d
|
dd gd dddgg¡ t dd}t dt¡}d||< t|t ddgddggddgddggg¡ d S )Nr r r r$ r& r% r' r8 r9 rE ru r7 r/ é é é r )r r r r )r r: r? r rb rJ )Zdirg Zd3rB rB rC Útest_diag_indicesã s8
|
|
|
ý
|
|
|
ýÿ
|
|
ÿÿþÿrÇ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestDiagIndicesFromc C s< t j d¡}t|\}}t|t d¡ t|t d¡ d S )Nr5 r )r: r r r r£ )r@ r[ Úrr| rB rB rC Útest_diag_indices_from s z*TestDiagIndicesFrom.test_diag_indices_fromc C s, t d¡}ttd t| W 5 Q R X d S )Nr' r¿ )r: r{ r r> r r^ rB rB rC Útest_error_small_input s
|
z*TestDiagIndicesFrom.test_error_small_inputc C s. t dt¡}ttd t| W 5 Q R X d S )N)r$ r$ r r$ r )r: rb rJ r r> r r^ rB rB rC Útest_error_shape_mismatch s z-TestDiagIndicesFrom.test_error_shape_mismatchN)rl rm rn rÊ rË rÌ rB rB rB rC rÈ ÿ s rÈ c C s° t tddd} dd tt d¡D }t| | t td} t| | t td} t| t td t t } t| dg t td} t| dg t td g } t| g d S )
|
Nr r r$ c S s g | ] \}}|qS rB rB )rf ÚixÚerB rB rC rª s z test_ndindex.<locals>.<listcomp>)r r r$ )r$ rB r )r r r r: rb r r )r[ r rB rB rC Ú test_ndindex s
|
|
|
rÏ )"ro Únumpyr: Z numpy.testingr r r r r r r Znumpy.lib.index_tricksr r
|
r r r r r r r r r r rr r r r¢ r¦ r¶ r· rÇ rÈ rÏ rB rB rB rC Ú<module> s $4 .s**G
|
