U
|
P ±d+L ã @ sT d Z ddlmZmZmZmZmZmZ ddlm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZ ddlZddlm Z ddl Z dd Z G dd d Z G d
|
d d Z G d d d Z!G dd dZ"G dd dZ#G dd dZ$dd Z%dd Z&dd Z'dd Z(d d Z)d d Z*G d d! d!Z+G d"d# d#Z,G d$d% d%Z-G d&d' d'Z.dS )(z"Test functions for matrix module
|
|
é )Ú assert_equalÚassert_array_equalÚassert_array_max_ulpÚassert_array_almost_equalÚ assert_raisesÚassert_)ÚarangeÚaddÚfliplrÚflipudÚzerosÚonesÚeyeÚarrayÚdiagÚ histogram2dÚtriÚ mask_indicesÚ triu_indicesÚtriu_indices_fromÚ tril_indicesÚtril_indices_fromÚvanderN)Úrequires_array_functionc C s t | }t ||¡}|S ©N)r r Úouter)ÚnÚdata© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_twodim_base.pyÚget_mat s r c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dS )ÚTestEyec
|
C s t tdtddddgddddgddddgddddgg t tdddtddddgddddgddddgddddggd t tddktdtd d S )Né é r Úf©Údtypeé )r r r Úbool©Úselfr r r Ú
|
test_basic s&
|
|
|
ýÿ
|
|
|
ýýÿ
|
ÿzTestEye.test_basicc C sj t tt d¡tdtddgddgg t tt d¡t d¡t d¡dtddddgddddgg d S )Né r% r# r r" )ÚMÚk)r r ÚnpZuint64Úintr r) r r r Ú test_uint64/ s ( ÿzTestEye.test_uint64c
|
C s t tdddtddddgddddgddddgddddgg t tdddtddddgddddgddddgddddgg d S )Nr" r# ©r. r éÿÿÿÿ©r r r r) r r r Ú test_diag5 s
|
|
|
ýÿ
|
|
|
ýÿzTestEye.test_diagc C sj t tddtdddgdddgdddgdddgg t tddtddddgddddgddddgg d S )Nr" r' r# r r4 r) r r r Útest_2dB s
|
|
ýÿ
|
|
|
þÿzTestEye.test_2dc C sr t tddddtddddgddddgddddgg t tddddtdddgdddgdddgdddgg d S )Nr' r" r, r2 r r# éþÿÿÿr4 r) r r r Ú test_diag2dN s
|
|
þÿ
|
ýÿzTestEye.test_diag2dc C s0 t tdddddgddgg t tdddddgddgg t tdddddgddgg t tdddddgddgg t tdddddgddgddgg t tdddddgddgddgg t tdddddgddgddgg t tdddddgddgddgg t tdddddgddgddgg d S )Nr, r# r r3 r7 r' éýÿÿÿ©r r r) r r r Útest_eye_boundsZ s $$$$zTestEye.test_eye_boundsc C s$ t tddddddgddgg d S )Nr, ZS3r% ó 1ó r: r) r r r Ú test_stringse s ÿzTestEye.test_stringsc C s$ t tddtdddgddgg d S )Nr, r% TF)r r r( r) r r r Ú test_booli s zTestEye.test_boolc C s\ t dddd}t ddddd}t|| |jjs4t|jjr@t|jjrLt|jjsXtd S )Nr" r' r3 r2 ÚF)r. Úorder)r r ÚflagsÚ c_contiguousÚAssertionErrorÚ f_contiguous)r* Zmat_cZmat_fr r r Ú
|
test_orderl s
|
zTestEye.test_orderN) Ú__name__Ú
|
__module__Ú __qualname__r+ r1 r5 r6 r8 r; r> r? rF r r r r r! s r! c @ s6 e Zd Zdd Zd ddZdd Zdd Zd
|
d ZdS ) ÚTestDiagc C s´ dt d d¡}td}tdD ]}|| |||f< q"tt|| td}| ¡ }tdD ],}|| |||d f< || ||d |f< q^tt|dd| tt|dd| d S ) Néd é Úl)rL rL )é rN r, r2 r7 )r Úastyper Úranger r Úcopy)r* ÚvalsÚbr. Úcr r r Ú test_vectorw s zTestDiag.test_vectorNc C sÒ |d kr dt d d d¡}td}tdD ]}|||f ||< q.tt|| |d }tdD ]}|||d f ||< qbtt|d|d d
tdD ]}||d |f ||< qtt|d |d d
d S )
|
NrK rL r# rM )rL r r' r, r7 )r rO r rP r r )r* rR rS r. r r r Ú test_matrix
s zTestDiag.test_matrixc C s( t dtd d ddd}| |¡ d S )NrK rL r# r@ rM )rA r& )r r rV )r* rR r r r Útest_fortran_order s zTestDiag.test_fortran_orderc C s ddgddgddgg}t t|ddg t t|dddg t t|ddddg t t|d dddg t t|d
|
ddg t t|d dg d S ) Nr# r, r' r" rL é r2 r r3 r7 r9 )r r )r* ÚAr r r Útest_diag_bounds s zTestDiag.test_diag_boundsc C s t ttdggg d S )Nr# )r Ú
|
ValueErrorr r) r r r Ú test_failure¡ s zTestDiag.test_failure)N)rG rH rI rU rV rW rZ r\ r r r r rJ v s
|
|
rJ c @ s e Zd Zdd ZdS )Ú
|
TestFliplrc C sv t tttd td}|d d
d d d
f }tt|| dddgdddgg}dddgdddgg}tt|| d S ©Nr" r3 r r# r, r' rL )r r[ r
|
r r r ©r* ÚarS r r r r+ ¦ s ÿÿzTestFliplr.test_basicN©rG rH rI r+ r r r r r] ¥ s r] c @ s e Zd Zdd ZdS )Ú
|
TestFlipudc C sf t d}|d d d
d d
f }tt|| dddgdddgg}dddgdddgg}tt|| d S r^ )r r r r_ r r r r+ ³ s ÿÿzTestFlipud.test_basicNra r r r r rb ² s rb c @ sb e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze d d Z
|
e j dddg¡dd ZdS )ÚTestHistogram2dc C s t dddddg}t dddd d
|
g}t d d d ¡}t d d d ¡}t||||fd }t d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g d d d d d d d d d g g }t|j| t|||d }t|j| tttd ttd \}}}t|td d t|t d dd¡ t|t d dd¡ d S )NgÌí
|
}°Ú?g%Úáñå ç?g$Fû ?gí;c¸jYÓ?gI+¤åÈÂ?gÐp5x£·?gó?ã_×Ç?gÒpÀª Ö?gúñb£dÙ?gºg<ü=á?r r# é
|
é é ) r r/ Úlinspacer r ÚTÚlistrP r )r* ÚxÚyZxedgesZyedgesÚHÚanswerr r r Ú test_simple¿ s8 ÿ ÿøÿ
|
zTestHistogram2d.test_simplec C sà t ddddddddg}t ddddddddg}t||dddgddggd d
|
\}}}t d ddddgdddddgdddddgdddddgdddddgdddddgg}t||d d t|t ddd ¡ t|t ddd¡ d S )Nr# r, r' r" rL r )rX rL rX T)rP Údensityç g @rN )r r r r r/ rg ©r* rj rk rl ÚxedÚyedrm r r r Ú test_asymÙ s* ÿ ûÿzTestHistogram2d.test_asymc
|
C s t dddddddddg }t dddddddddg }t||ddddgddddggdd\}}}t dddgdddgdddggd }t||d d S )
|
Nr# r, r' rL T)ro ç à?ç Ð?g "@)r r r rq r r r Ú test_densityé s ÿ
|
þþz TestHistogram2d.test_densityc C sD t j d¡d d }t||dddgddgfd\}}}t|d d S )NrK ç ð?g .A)r" rL r r# )rP )r/ ÚrandomZrandr r )r* Úrrl rr rs r r r Útest_all_outliersó s "z!TestHistogram2d.test_all_outliersc C sZ t g g ddgddgfd\}}}t|tdgg t g g dd\}}}t|t d¡ d S )Nr r# ©Zbinsrp r" )r" r" )r r r r/ r )r* r` Zedge1Zedge2r r r Ú
|
test_emptyø s zTestHistogram2d.test_emptyc C sp t dddddddg}t ddd d
|
d d dg}d }t|||df\}}}t ddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgg
|
}t|| t|t dddddg t||d|f\}}}t ddddddddddg
|
ddddddddddg
|
ddddddddddg
|
ddddddddddg
|
g}t|| t|t dddddg d S )Nr g<Dç·?gô±³Ï ªä?g!,ø§3{À?glÊyTÞ?g¥fì¦.ã?r# g( @TT\Â?g±ÁÂIZß?gYöN»Ó?gK?~ÂÜ?g&W-À©UÄ?) r g¹?gÉ?g333333Ó?gÙ?ru g333333ã?gffffffæ?gé?gÍÌÌÌÌÌì?r# r" g @rp rx rv ru g è?)r r r )r* rj rk Úedgesrl ZxeZyerm r r r Ú test_binparameter_combinationÿ sN
|
ÿÿ
|
ÿÿ
|
|
|
|
|
|
|
|
|
|
÷ÿ
|
ýÿ
|
z-TestHistogram2d.test_binparameter_combinationc C sþ G dd d}ddg}| }t ||}t||f||fi fk t ||}t||f||fi fk t |||d}t||f||ft|df t |||dgd}t||f||ft|dgdf ttt |||gd t |||d}t||f||ft|df d S )Nc @ s e Zd Zdd ZdS )z5TestHistogram2d.test_dispatch.<locals>.ShouldDispatchc S s
|
|||fS r r )r* ÚfunctionÚtypesÚargsÚkwargsr r r Ú__array_function__! s zHTestHistogram2d.test_dispatch.<locals>.ShouldDispatch.__array_function__N)rG rH rI r r r r r ÚShouldDispatch s r
r# r, r| rL )Úweights)r r Údictr Ú Exception)r* r
ZxyZs_drz r r r Ú test_dispatch s
|
|
z TestHistogram2d.test_dispatch)Úx_lenÚy_len)rd rf )é é c C s> t |¡t |¡ }}tjtdd t|| W 5 Q R X d S )Nz"x and y must have the same length.)Úmatch)r/ r ÚpytestZraisesr[ r )r* r r rj rk r r r Útest_bad_length3 s
|
ÿz TestHistogram2d.test_bad_lengthN)rG rH rI rn rt rw r{ r} r r r r ÚmarkZ parametrizer r r r r rc ¾ s
|
|
rc c @ s e Zd Zdd ZdS )ÚTestTric C sJ t dddgdddgdddgg}ttd| ttdtd| t¡ d S )Nr# r r' r% )r r r r( rO )r* Úoutr r r Ú
|
test_dtype< s
|
þzTestTri.test_dtypeN)rG rH rI r r r r r r ; s r c C s~ t jd t jd D ]d} t jd| d}t |¡}t |¡}t|ddgddgg t||j t|j|j t|j|j qd S )NÚAllFloatÚ
|
AllInteger)r, r, r% r# r ) r/ Ú typecodesr ÚtrilÚtriur rh r r& )r& r` rS rT r r r Útest_tril_triu_ndim2D s
|
|
r c C s
|
t jd t jd D ]ð} t jddgddggddgddggddgddggg| d}t jddgddggddgddggddgddggg| d}t jddgddggddgddggddgddggg| d}t |¡}t |¡}t|| t|| t|j|j t|j|j qd S )Nr r r# r r% )r/ r r r r r r r& )r& r` Za_tril_desiredZa_triu_desiredZa_triu_observedZa_tril_observedr r r Útest_tril_triu_ndim3P s8 ýüýüýü
|
|
|
|
r c C st t ddt jgdddgt jddgg¡} t dddgdddgt jddgg¡}|j}tt | ¡| tt | ¡| d S )Nr# r )r/ r Úinfrh r r r )ÚarrZout_trilZout_triur r r Útest_tril_triu_with_infi s
|
þ
|
þr c C sÌ t jd D ]D} | dkrq
|
t jd| d}tt |¡j|j tt |¡j|j q
|
t jddgddggd d}tt |¡j|j tt |¡j|j t jdd
|
d}tt |¡j|j tt |¡j|j d S ) NZAllÚV)r' r' r% z2001-01-01T12:00z2002-02-03T13:56z2004-01-01T12:00z2003-01-03T13:45Z
|
datetime64zf4,f4)r/ r r r r r& r r )rT r r r r Útest_tril_triu_dtypev s
|
ÿþr c C sf t dtj} t d¡ dd¡}t|| tddddddg t dtjd}t|| tdddg d S ) Nr' re r r# r, r" rL é )r r/ r r Úreshaper r )Úiur` Úiu1r r r Útest_mask_indices s
|
r¥ c C sÌ t d} t ddd}t ddd}t dddd}t ddddgdd d
|
d gd d ddgddddgg¡}t dd¡ dd¡}t|| tddd d d dddddg
|
t|| tdd d
|
dddddddg
|
d|| < t|tddddgddd
|
d gddddgddddgg d||< t|tdddddgddd d d gdddddgdddddgg d||< t|tddddgddddgddddgddddgg d||< t|tdddddgddddd gdddddgdddddgg d S )Nr" r, r2 rL ©Úm©r. r§ r# r' rX rN r¡ re rd rf é é é é é é é é r r3 r éöÿÿÿ)r r/ r r r¢ r )Zil1Zil2Zil3Zil4r` rS r r r Útest_tril_indices s`
|
|
|
ýÿÿ
|
|
|
ýÿ ýÿ
|
|
|
ýÿ ýÿr² c @ s e Zd Zdd ZdS )ÚTestTriuIndicesc C sÔ t d}t ddd}t ddd}t dddd}t ddddgdd d
|
d gd d ddgddddgg¡}t dd¡ dd¡}t|| tddddd d
|
d dddg
|
t|| tdddddd
|
d d d dddddg d||< t|tddddgddddgd d ddgddddgg d||< t|tdddddgd ddddgdddddgdddddgg d||< t|tddddgddddgd d ddgddddgg d||< t|tdddddgd ddddgdddddgdddddgg d S )Nr" r, r2 rL r¦ r¨ r# r' rX rN r¡ re rd rf r© rª r« r¬ r r® r r r3 r¯ r° r± )r r/ r r r¢ r )r* r¤ Ziu2Ziu3Ziu4r` rS r r r Útest_triu_indicesÆ sn
|
|
|
ýÿ ÿÿ
|
|
|
ýÿ ýÿ
|
|
|
ýÿ ýÿz!TestTriuIndices.test_triu_indicesN)rG rH rI r´ r r r r r³ Å s r³ c @ s e Zd Zdd ZdS )ÚTestTrilIndicesFromc C s( t ttt d¡ t ttt d¡ d S ©N)r, )r, r, r, )r r[ r r/ r r) r r r Útest_exceptionsø s z#TestTrilIndicesFrom.test_exceptionsN©rG rH rI r· r r r r rµ ÷ s rµ c @ s e Zd Zdd ZdS )ÚTestTriuIndicesFromc C s( t ttt d¡ t ttt d¡ d S r¶ )r r[ r r/ r r) r r r r· ÿ s z#TestTriuIndicesFrom.test_exceptionsNr¸ r r r r r¹ þ s r¹ c @ s e Zd Zdd Zdd ZdS )Ú
|
TestVanderc
|
C s´ t ddddg¡}t|}t dddddgdddddgdddddgdd d
|
ddgg¡}t||d d
dd
f |jd }td D ].}t||d }t||d d
|| |
f qd S ) Nr r# r7 r' r iøÿÿÿr" éQ é re rX ©ÚN)r/ r r r ÚshaperP )r* rT ÚvZpowersr§ r r r r r+ s ý
|
zTestVander.test_basicc C s t dddgtjd}t|}t dddgdddgdddgg¡}t|| t d d
|
g}t|d d }t d d dgdd
|
dgg¡}t|| d S )Nrf iôÿÿÿrª r% éy r# é é© y ð? ð?y ð? ð¿r' r½ y @y À)r r/ Zint8r r )r* rT rÀ Úexpectedr r r Ú test_dtypes s þ
|
ÿzTestVander.test_dtypesN)rG rH rI r+ rÅ r r r r rº s rº )/Ú__doc__Z numpy.testingr r r r r r Únumpyr r r
|
r r r r r r r r r r r r r r r/ Z numpy.core.tests.test_overridesr r r r! rJ r] rb rc r r r r r r¥ r² r³ rµ r¹ rº r r r r Ú<module> s, L X/ }
|
/2
|
