U
|
P ±dÅ/ ã @ s d Z ddlZddlZddlZddlZddlmZmZm Z m
|
Z
|
m Z m Z m Z dd Zdd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Z d"d# Z d$d% Z d&d' Z d(d) Z G d*d+ d+Z!d,d- Z"d.d/ Z#dS )0zTests of interaction of matrix with other parts of numpy.
|
|
Note that tests with MaskedArray and linalg are done in separate files.
|
é N)Úassert_Ú assert_equalÚ assert_raisesÚassert_raises_regexÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalc C s~ t ddgddgg¡} tt| dddgd d
f t j t t d¡ dd¡¡}t|d d
t d ¡f |d d
d f j d S )
|
Né é é é r é2 é é
|
éÿÿÿÿ)
|
ÚnpÚmatrixr Ú
|
isinstanceÚasmatrixÚarangeÚreshaper ÚarrayÚT)ÚmÚx© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/matrixlib/tests/test_interaction.pyÚtest_fancy_indexing s "r c C sD ddg} ddg}t | | g¡}t jj || |¡}tt|t j d S )Nr r r r )r r Z
|
polynomialZ polyutilsZ mapdomainr r )Zdom1Zdom2r Úresr r r Útest_polynomial_mapdomain s
|
r c C sR t dddgg¡} t j| d d}t dddgg¡}t|| tt|t jk d S ©Nr
|
r r ©Úaxis)r r Úsortr r Útype©ÚaÚactualÚexpectedr r r Útest_sort_matrix_none) s
|
|
r) c C sT t dddgg¡} t j| dd d}t dddgg¡}t|| tt|t jk d S r )r r Ú partitionr r r$ r% r r r Útest_partition_matrix_none2 s
|
|
r+ c C sN t jddgtd} t jddggtd}tt | d¡| tt d| ¡| d S ©Nr r
|
©Zdtyper é )r r Úobjectr Údot©ÚarrÚdesiredr r r Ú%test_dot_scalar_and_matrix_of_objects< s r4 c C s t jd t jd d D ]n} t jd| dd }t jddgdd gg| d}t jdd
|
gd d gg| d}tt ||¡| tt ||¡| qd S ) NZ
|
AllIntegerZAllFloatú?r r- r r r
|
r r. é é )r Z typecodesr r r Úinner)ÚdtZscar2 r3 r r r Ú test_inner_scalar_and_matrixE s r: c C sN t jddgtd} t jddggtd}tt | d¡| tt d| ¡| d S r, )r r r/ r r8 r1 r r r Ú'test_inner_scalar_and_matrix_of_objectsO s r; c C s2 t ddgddgg¡} t d¡ dd¡j}t | |d gg dgdgddgg¡}tt|jd t jk tt|jd t j k t
|
|jd j d t d¡ ddd¡}t t t j| |d gg dgdgddgg t | |d gg dgdgddd gg¡}tt|jd t j k tt|jd t jk t
|
|jd j d
|
d S ) Nr r
|
r r ÚreadonlyZ writeonlyÚallocate)r
|
r
|
Z
|
no_subtype)r r
|
r
|
)r r r r r Znditerr r$ ZoperandsÚndarrayr Úshaper Ú RuntimeError)r& ÚbÚir r r Ú!test_iter_allocate_output_subtypeX s( ÿÿÿÿrC c C sh t ddgddgg¡} t jt jt jfD ]<}|| }tt|t jk || dd}tt|t jk q&d S )Nr r
|
r r F)Úsubok)r r Z
|
zeros_likeZ ones_likeZ
|
empty_liker r$ )r& Ú like_functionrA Úcr r r rE s s rE c C s t jdddgdddggdd} | jdd d
|
d }t| |k | jd d
|
d }t| | tt|t j | jdd
|
d
|
d }t| | t| |k tt|t jk d S )Nr r r
|
r r r Zf4r- TF)rD ÚcopyÚi4)rG )r r Zastyper r r$ ©r& rA r r r Útest_array_astype~ s
|
|
rJ c C s. t ddgddgg¡} ttdt j| | g d S )Nr r
|
r r z shape too large to be a matrix)r r r Ú
|
ValueErrorÚstack)r r r r Ú
|
test_stack s
|
ÿrM c C sN t jddgtd} t jddggtd}tt | d¡| tt d| ¡| d S r, )r r r/ r Úmultiplyr1 r r r Útest_object_scalar_multiply s rO c
|
C s t t d¡¡} t jt jfD ]n}|| dd}tt|t j t|jdk || dd}tt|t j t|jdk || }tt |¡ q t j | d< t jt jfD ]X}t
|
j ddR}t
|
d ¡ || dd}tt|t j tt t |¡¡ tt|dk W 5 Q R X t
|
j dd}t
|
d ¡ || dd}tt|t j tt |d
|
¡ort |d ¡ ort |d ¡ tt|dkd tt|d jt W 5 Q R X t
|
j ddD}t
|
d ¡ || }tt |¡ t|t j k tt|dk W 5 Q R X q¢d S )Nr r r! ©r r r ©r r T)ÚrecordÚalways)r r )r r )r
|
r zno warning raised)r r ÚeyeZnanminZnanmaxr r r? ÚisscalarÚnanÚwarningsÚcatch_warningsÚ simplefilterÚanyÚisnanÚlenÚ
|
issubclassÚcategoryÚRuntimeWarning)ÚmatÚfr Úwr r r Útest_nanfunctions_matrices¢ sB
|
|
|
$ÿ
|
rc c C s0 t t d¡¡} t jt jt jt jt jt jt j fD ]n}|| dd}t
|
t |t j t
|
|j dk || dd}t
|
t |t j t
|
|j dk || }t
|
t |¡ q0t jt jfD ]~}|| dd}t
|
t |t j t
|
|j dk || dd}t
|
t |t j t
|
|j dk || }t
|
t |t j t
|
|j dk q¬d S ) Nr r r! rP r rQ )r r )r r6 )r r rT Z nanargminZ nanargmaxZnansumZnanprodZnanmeanZnanvarZnanstdr r r? rU Z nancumsumZ
|
nancumprod)r` ra r r r r Ú"test_nanfunctions_matrices_generalË s0 ÿ rd c C s t t j dd¡¡} t| d¡t | d¡ t ddgddgg¡}t ddgddgg¡}t j|d|d}tt|t j t|dd gg d S )
|
Nr r r r
|
r r )r" Úweightsg @g«ªªªªª
|
@) r r ÚrandomZrandr ZmeanZaverager r$ )Úyr& rb Úrr r r Útest_average_matrixç s ri c C sN t dd¡} | | }t || ¡}t | ¡}t |¡}t ||¡}t|| d S )Nr r )r ZlinspaceZtrapzr r )r rg rh ZmxZmyÚmrr r r Útest_trapz_matrixô s
|
|
rk c C s@ t t t d¡¡tjs tt tjt d¡ddtjs<td S )Nr )Zto_begin)r r Zediff1dr ÚAssertionErrorr r r r Útest_ediff1d_matrix s rm c C s dd } t ddgddgg¡}t ddgddgg¡}t | d|¡}tt|t j t|| t | d|¡}tt|t j t|| d S ) Nc S s | d S ©Nr
|
r )Úrowr r r Údouble
|
s z,test_apply_along_axis_matrix.<locals>.doubler r r
|
r r r. )r r Zapply_along_axisr r r )rp r r( Úresultr r r Ú test_apply_along_axis_matrix s
|
rr c C s| t ddg¡} t | ¡}ttt | | ¡t j ttt ||¡t j ttt | |¡t j ttt || ¡t j d S rn )r Zonesr r r$ Zkronr> r )r& r r r r Útest_kron_matrix s
|
rs c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestConcatenatorMatrixc s¢ ddg ddgt jd f }t jd f }tt|t j tt|t j tt |¡ddddgg tt |¡dgdgdgdgg tt fdd d S ) Nr r
|
r r rh rF c s t jd f S )NÚrc)r Úr_r rI r r Ú<lambda>2 ó z4TestConcatenatorMatrix.test_matrix.<locals>.<lambda>)r rv r r$ r r r rK )ÚselfZab_rZab_cr rI r Ú test_matrix% s z"TestConcatenatorMatrix.test_matrixc C s@ t jdddgdf }tt|t j tt |¡dddgg d S )Nrh r r
|
r )r rv r r$ r r )ry rh r r r Útest_matrix_scalar4 s z)TestConcatenatorMatrix.test_matrix_scalarc C sp t dg¡}t dg¡}t dg¡}t dg¡}t jd }t ||g||gg¡}t|| tt|t| d S )Nr r
|
r r z
|
a, b; c, d)r r rv Zbmatr r$ )ry r& rA rF Údr' r( r r r Útest_matrix_builder9 s
|
|
z*TestConcatenatorMatrix.test_matrix_builderN)Ú__name__Ú
|
__module__Ú __qualname__rz r{ r} r r r r rt # s rt c C sV t t¡$} tt ddg¡t ddg¡ W 5 Q R X t| j}t
|
d¡}t|| d S )Nr r
|
zl
|
Arrays are not equal
|
|
(shapes (2,), (1, 2) mismatch)
|
x: array([1, 2])
|
y: matrix([[1, 2]])) ÚpytestZraisesrl r r r r ÚstrÚvalueÚtextwrapÚdedent)Úexc_infoÚmsgZ msg_referencer r r Ú%test_array_equal_error_message_matrixE s
|
(
|
|
r c C s² t ddgg¡} t dt jgg¡}t dt j gg¡}t t jt jgg¡}t ddgt jt jgg¡}ttfD ]@}| ||||fD ],}||| t |¡}||| ||| q~qld S )Ng ð?g @)r r rV Úinfr r r )Úm1Úm2Zm3Zm4Zm5Z assert_funcr r& r r r Ú test_array_almost_equal_matrixT s
|
|
|
r )$Ú__doc__r r rW Únumpyr Z numpy.testingr r r r r r r r r r) r+ r4 r: r; rC rE rJ rM rO rc rd ri rk rm rr rs rt r r r r r r Ú<module> s6 $
|
|
|
)
|
"
|
