U
|
P ±dK< ã @ sÞ d dl Zd dlZd dlmZmZmZ d dlmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ G dd dZG dd d ZG d
|
d d ZG d d d ZG dd dZG dd dZG dd dZG dd dZG dd dZdS )é N)ÚmatrixÚasmatrixÚbmat)Úassert_Ú assert_equalÚassert_almost_equalÚassert_array_equalÚassert_array_almost_equalÚ assert_raises)Ú matrix_power)Úmatc @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestCtorc
|
C s t ddgddgg¡}t|}tt |j|k¡ td}t||g||gg}t ddddgddddgddddgddddgg¡}tt |j|k¡ tt |j|k¡ t ddgdd gg¡}tddddgdddd gg}tt t||g|k¡ t d¡}t|} t| jd
|
k d S ) Né é é é úA,A;A,Aé é é é )r r ) ÚnpÚarrayr r ÚallÚAr ÚarangeÚshape)
|
Úselfr ÚmAÚBÚCÚDÚEZAEresultZvecZmvec© r# ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/matrixlib/tests/test_defmatrix.pyÚ
|
test_basic s$
|
|
|
ý
|
zTestCtor.test_basicc C s t ttd d S )NÚinvalid)r
|
Ú
|
ValueErrorr ©r r# r# r$ Útest_exceptions# s zTestCtor.test_exceptionsc C s, t ddgddgg¡}t ddgddgg¡}t ddddgddddgddddgddddgg¡}t ddddgddddgddddgddddgg¡}tt td |k¡ tt td d
|
|id |k¡ tttd d
|
|id tt td d
|
|id
|
|id |k¡ td||d||dd }tt ||k¡ d S )Nr r r r r r r r r r )Úldict)Úgdict)r* r+ zA,B;C,D)r r )r r! )r r r r r r
|
Ú TypeError)r r r ZAresultZ mixresultZb2r# r# r$ Útest_bmat_nondefault_str' s(
|
|
|
ý
|
|
|
ý ÿz!TestCtor.test_bmat_nondefault_strN)Ú__name__Ú
|
__module__Ú __qualname__r% r) r- r# r# r# r$ r s r c @ st e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd ZdS ) ÚTestPropertiesc C sÄ t ddddgddddgddddgddddgg}t ddddg}t dd dd
|
gj}d }t||jdd t||jdd t|| ¡ t|tj|dd t|tj|dd t|t |¡ d S )zeTest whether matrix.sum(axis=1) preserves orientation.
|
Fails in NumPy <= 0.9.6.2127.
|
r r r r r r é r r é é ©ZaxisN)r ÚTr Úsumr r )r ÚMZsum0Zsum1Zsumallr# r# r$ Útest_sum< s
|
|
|
ýzTestProperties.test_sumc C sÐ t dddgdddgg}t| ¡ d t| d¡t dd d
|
gg t| d¡t dgd gg tt |¡d ttj|dd t dd d
|
gg ttj|dd t dgd gg t dddg}t| ¡ dk d S ) Nr r r r r r iÐ r é
|
é éx r5 )r r Úprodr r ©r ÚxÚyr# r# r$ Ú test_prodO s zTestProperties.test_prodc C s² t dddgdddgg}t| ¡ d t| d¡t dddgg t| d¡t dgdgg tt |¡d ttj|ddt dddgg ttj|ddt dgdgg d S © Nr r r r r r r r5 )r r Úmaxr ©r r? r# r# r$ Útest_max\ s zTestProperties.test_maxc C s² t dddgdddgg}t| ¡ d t| d¡t dddgg t| d¡t dgdgg tt |¡d ttj|ddt dddgg ttj|ddt dgdgg d S rB )r r Úminr rD r# r# r$ Útest_minf s zTestProperties.test_minc C sh t d¡ d¡}t| ¡ dk tt | d¡t ddg¡k¡ tt | d¡t ddg¡k¡ d S )Nr )r r r r r r )r r Úreshaper Úptpr r rD r# r# r$ Útest_ptpp s "zTestProperties.test_ptpc C sT t d¡ d¡}| t j¡}t|jdd|jdd t|jdd|jdd d S )Né )r r r )Zddofr )r r rH Úviewr r Úvar)r r? Zmxr# r# r$ Útest_varv s zTestProperties.test_varc C s
|
dd l m} t ddgddgg¡}t|}tt | |¡|j¡ tt t t
|
|¡|j k¡¡ tt t t
|
|¡|j k¡¡ tt ||j k¡ |d| }t|}tt | |¡|j¡ tt t t
|
|¡|j k¡¡ tt t t
|
|¡ ¡ |j k¡¡ d S )Nr ç ð?ç @ç @ç @y @)Ú numpy.linalgÚlinalgr r r r ÚallcloseÚinvÚIr Z transposer6 ÚHr Zconj)r rT r r r ÚmBr# r# r$ r% | s
|
ÿ zTestProperties.test_basicc C s@ t t d¡ dd¡}t ddgddgdd gg}t|j| d S )
|
Nr r r g·ù:ãè¿gnó)t ÇÑ?g#¬Çq¼¿g#¬Çq¼?gnó)t Çá?gç6BÇq¬¿)r r r rH r rW )r r? Zxpinvr# r# r$ Ú test_pinv s þzTestProperties.test_pinvc C sr t d¡ dd¡}t|}t|d }tt ||d k¡ tt |t|d k¡ tt |t|d k¡ tt ||k ¡ tt ||k¡ tt ||k¡ tt ||k ¡ tt ||k ¡ tt ||k¡ tt ||k¡ tt ||k¡ tt ||k¡ tt ||k¡ tt ||k¡ tt t|dk¡ tt t|dk¡ d S )Néd r: ç¹?r )r r rH r r r ÚanyÚabs)r r r rY r# r# r$ Útest_comparisons s& z TestProperties.test_comparisonsc C s: t d¡ dd¡}t|}d|d< t|d |d k d S )Nr[ r: iöÿÿÿ©r r )r r rH r r )r r r r# r# r$ Ú test_asmatrix¬ s z TestProperties.test_asmatrixc C s@ t ddgddgg}t| ¡ t dk t| ¡ t dk d S )Nr r r ç à?)r r r7 Úmean©r r r# r# r$ Ú test_noaxis² s zTestProperties.test_noaxisc C s( t ddgddgg}tt|dk d S )Nr r z matrix([[1, 0],
|
[0, 1]]))r r Úreprrd r# r# r$ Ú test_repr· s zTestProperties.test_reprc C s* t d}t dgdgdgg}t|| d S )NzTrue; True; FalseTF)r r )r r r r# r# r$ Ú test_make_bool_matrix_from_str» s z-TestProperties.test_make_bool_matrix_from_strN)r. r/ r0 r9 rA rE rG rJ rN r% rZ r_ ra re rg rh r# r# r# r$ r1 ; s
|
|
r1 c @ s e Zd Zdd ZdS )Ú TestCastingc C s¼ t d¡ dd¡}t|}| ¡ }t dt j¡d }|| }t|jj t jk tt
|
||k¡ tt
|
||d k¡ | ¡ }t dt j ¡}|| }t|jj t j k tt
|
||k¡ d S )Nr[ r: )r: r: r\ ) r r rH r ÚcopyZonesÚfloat64r ÚdtypeÚtyper Z
|
complex128)r r r rY ÚOZmCr# r# r$ r% Á s zTestCasting.test_basicN©r. r/ r0 r% r# r# r# r$ ri À s ri c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestAlgebrac C s& dd l m} t ddgddgg¡}t|}t d¡}tdD ]&}tt || j |¡ t
|
||¡}q<| |¡}t d¡}tdD ](}tt || j |¡ t
|
||¡}qtt || j t
|
||¡¡ tt || j || ¡ tt d| j d| ¡ t|}|d9 }tt |j d| ¡ d S ) Nr rO rP rQ rR r r r ) rS rT r r r ÚidentityÚranger rU r ÚdotrV )r rT r r r ÚiZAinvZmA2r# r# r$ r% Ô s$
|
|
|
zTestAlgebra.test_basicc C st t d}| ¡ }|dC }| ¡ }|dC }| ¡ }|dC }t||d t|t ||¡ tt ||¡t d¡ dS )z<Test raising a matrix to an integer power works as expected.z 1. 2.; 3. 4.r éÿÿÿÿN)r rj r r rs Úeye)r ÚmÚm2ÚmiZm4r# r# r$ Útest_powí s zTestAlgebra.test_powc C sD t ddgddgg}tjtjfD ] }|d}t|d || q d S ©Nr r r r )r r Zint8Zuint8r )r rw Zscalar_tZtwor# r# r$ Útest_scalar_type_powú s z TestAlgebra.test_scalar_type_powc C sR t ddgddgg}tt d| W 5 Q R X tt |t W 5 Q R X dS )z:Check that 'not implemented' operations produce a failure.rO rP rQ rR N)r r
|
r, Úobjectrd r# r# r$ Útest_notimplemented s ÿ
|
|
z TestAlgebra.test_notimplementedN)r. r/ r0 r% rz r| r~ r# r# r# r$ rp Ó s rp c @ s e Zd Zdd ZdS )ÚTestMatrixReturnc
|
& C s^ t dgdd}dddgfdddt dg¡d }d
|
d d d ddddddddddddddd d d d d d!d"d#d$d%d
|
d&d'd(d)d*d+d,d-d.g&}t|D ]x}| d/¡s||krqt||}t|tjj r|
|
d¡ | d¡ ||krÜ|| }nd0}|| }t t |t kd1| qt t |jt k t t |jt k t d2g ¡ \}} t t |tjk t t | tjk d S )3NrO Zf8)rl )Zintc)ç rO r )r r` )ÚastypeZclipÚcompressÚrepeatrH Zswapaxesrs ZargminÚchooseÚdumpÚdumpsÚfillZgetfieldZgetAZgetA1ÚitemÚnonzeroÚputZputmaskÚresizeZ searchsortedZsetflagsZsetfieldÚsortÚ partitionZ argpartitionZtakeZtofileÚtolistÚtostringÚtobytesr r] r7 ZargmaxrF rC rc rM rI r= ZstdÚctypesZitemsetÚ_r# z%sr )r r r ÚdirÚ
|
startswithÚgetattrÚ
|
isinstanceÚ collectionsÚabcÚCallabler r r rm ÚrealÚimagr Úndarray)
|
r ÚaZ
|
methodargsZexcluded_methodsÚattribÚfÚargsÚbÚcÚdr# r# r$ Útest_instance_methods s
|
ù
|
ù
|
|
|
|
z&TestMatrixReturn.test_instance_methodsN)r. r/ r0 r¤ r# r# r# r$ r s r c @ s e Zd Zdd ZdS )Ú TestIndexingc C sf t t dt¡}t dt¡}dddg|d d
df< |dk|d d
df< t|ddgddgddgg d S ) N©r r )r r gé?gÉ?g333333Ó?r rb r ©r r ÚzerosÚfloatr r> r# r# r$ r% 9 s
|
zTestIndexing.test_basicNro r# r# r# r$ r¥ 8 s r¥ c @ sh e Zd ZeddgddggZdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestNewScalarIndexingr r r r c C s | j }|d }t|jd d S )Nr r )r r Úndim©r r r? r# r# r$ Útest_dimesionsD s z$TestNewScalarIndexing.test_dimesionsc C s* | j }t ||g¡}t|jdddg d S )Nr )r r r r r r¬ r# r# r$ Útest_array_from_matrix_listI s z1TestNewScalarIndexing.test_array_from_matrix_listc C s$ | j }t| ¡ ddgddgg d S r{ )r r r ©r r r# r# r$ Útest_array_to_listN s z(TestNewScalarIndexing.test_array_to_listc C s¼ | j }|ddddgf }tt|t t|tdddgg |ddg }tt|t t|tddgddgg |dgdggddgddggf }tt|t t|tddgddgg d S )Nr r r r r )r r r r r r¬ r# r# r$ Útest_fancy_indexingR s z)TestNewScalarIndexing.test_fancy_indexingc C s´ t dddgdddgg}t|d d t dddgg t|d d jd t|d jd t|d d
df jd t d}t|d
|
d t|d d t|d d
df j|j d S ) Nr r r r r r r )r r ©r r r` )r r r rD r# r# r$ Útest_matrix_element^ s z)TestNewScalarIndexing.test_matrix_elementc C s* t t dt¡}t|d |d d d S )Nr¦ r` r r§ rD r# r# r$ Útest_scalar_indexingj s z*TestNewScalarIndexing.test_scalar_indexingc C s t t d¡}t|dd d
f ddgg t|dd d
f ddgg t|d d
df dgdgg t|d d
df dgdgg d S )Nr r r )r r rv r rD r# r# r$ Útest_row_column_indexingn s
|
z.TestNewScalarIndexing.test_row_column_indexingc C sv t d¡}d|_t|}t|d d
t ddg¡f |d d
df t|t dddg¡d d
f |dd d
f d S )Nr r¦ TFr )r r r r r r ©r r r? r# r# r$ Útest_boolean_indexingu s
|
|
,z+TestNewScalarIndexing.test_boolean_indexingc C sv t d¡}d|_t|}t|d d
ddgf |d d
d d d
f t|dddgd d
f |d d d
d d
f d S )Nr r¦ r r ru r )r r r r r r¶ r# r# r$ Útest_list_indexing| s
|
|
,z(TestNewScalarIndexing.test_list_indexingN)r. r/ r0 r r r r® r° r± r³ r´ rµ r· r¸ r# r# r# r$ rª A s rª c @ s e Zd Zdd Zdd ZdS )Ú TestPowerc C sP t ddgddgg¡}ttt|dt jk t|}ttt|dtk d S ©Nr r r )r r r rm r r r r r¯ r# r# r$ Útest_returntype
s zTestPower.test_returntypec C s, t tddgddggdddgddgg d S rº )r r r( r# r# r$ Ú test_list s zTestPower.test_listN)r. r/ r0 r» r¼ r# r# r# r$ r¹ s r¹ c @ sv e Zd Ze dgdgg¡ZedgdggZdd Zdd Z dd Z
|
d d
|
Z d d Z d d Z dd Zdd Zdd ZdS )Ú TestShaper r c C s t | jjd t | jjd d S )Nr² )r r r rw r( r# r# r$ Ú
|
test_shape s zTestShape.test_shapec C s, t t | j¡jd t t | j¡jd d S )N©r )r r Úravelr r rw r( r# r# r$ Útest_numpy_ravel s zTestShape.test_numpy_ravelc C s( t | j ¡ jd t | j ¡ jd d S ©Nr¿ )r r )r r rÀ r rw r( r# r# r$ Útest_member_ravel s zTestShape.test_member_ravelc C s( t | j ¡ jd t | j ¡ jd d S r )r r Úflattenr rw r( r# r# r$ Útest_member_flatten s z TestShape.test_member_flattenc C s. t dddgdddgg¡}tt |¡ddddddg tt j|ddddddddg tt |j¡ddddddg tt j|jd dddddddg tdddgdddgg}tt |¡ddddddg tt j|ddddddddg tt |j¡ddddddg tt j|jd dddddddg d S ©
|
Nr r r r r r ÚF)Úorderr )r r r rÀ r6 r rD r# r# r$ Útest_numpy_ravel_order¤ s " z TestShape.test_numpy_ravel_orderc C s t dddgdddgg}t| ¡ ddddddgg t|jddddddddgg t|j ¡ ddddddgg t|jjd dddddddgg d S rÆ )r r rÀ r6 rD r# r# r$ Útest_matrix_ravel_order° s
|
z!TestShape.test_matrix_ravel_orderc C s6 t t | j| j ¡ ¡ t t | j| j ¡ ¡ d S ©N)r r Úmay_share_memoryr rÀ rÄ r( r# r# r$ Útest_array_memory_sharing· s z#TestShape.test_array_memory_sharingc C s6 t t | j| j ¡ ¡ t t | j| j ¡ ¡ d S rË )r r rÌ rw rÀ rÄ r( r# r# r$ Útest_matrix_memory_sharing» s z$TestShape.test_matrix_memory_sharingc C sH t d¡ d¡ t j¡}t j|dd}t|jd tt |t j d S )Nr: )r r r r5 r )
|
r r rH rL r Z expand_dimsr r« r r )r r Úexpandedr# r# r$ Útest_expand_dims_matrix¿ s z!TestShape.test_expand_dims_matrixN)r. r/ r0 r r r r rw r¾ rÁ rà rÅ rÉ rÊ rÍ rÎ rÐ r# r# r# r$ r½ s r½ )Úcollections.abcr Únumpyr r r r Z numpy.testingr r r r r r
|
rS r Znumpy.matrixlibr r r1 ri rp r r¥ rª r¹ r½ r# r# r# r$ Ú<module> s / ;* C
|
