f }} L d dl Zd dlZd dlmZmZmZ d dlmZm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ G d d Z G d d Z G d
d Z G d d
Z G d d Z G d d Z G d d Z G d d Z G d d ZdS ) N)matrixasmatrixbmat)assert_ assert_equalassert_almost_equalassert_array_equalassert_array_almost_equal
assert_raises) matrix_power)matc e Zd Zd Zd Zd ZdS )TestCtorc t j ddgddgg }t | }t t j |j |k t
d }t
||g||gg }t j g dg dg dg dg }t t j |j |k t t j |j |k t j dd gd
d gg }t g d g d
g }t t j t
||g |k t j d }t | } t | j dk d S )N A,A;A,Ar r r r r r r r r r r r r r r r )r r ) nparrayr r allAr arangeshape)
selfr! mABCDEAEresultvecmvecs
x/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/matrixlib/tests/test_defmatrix.py
test_basiczTestCtor.test_basic
sk Hq!fq!f% & &
AYYrtqy!!""" OO 1a&1a&! " " Hlll"ll"ll"ll$
%
% qsax !!!qsax !!! Hq!fq!f% & &<<< 677tQF||x/00111illc{{
f$%%%%% c < t t t d d S )Ninvalid)r
ValueErrorr r$ s r- test_exceptionszTestCtor.test_exceptions# s j&)44444r/ c
t j ddgddgg }t j ddgddgg }t j g d g d
g d g d
g }t j g d g d g d
g dg }t t j t d |k t t j t dd|i |k t t t dd|i t t j t dd|id|i |k t d||d||d }t t j ||k d S )Nr r r r r r r r r r r r )r r r r )r r r r r r! )ldict)gdict)r6 r7 zA,B;C,D)r! r& )r' r( )r r r r r r TypeError)r$ r! r& Aresult mixresultb2s r- test_bmat_nondefault_strz!TestCtor.test_bmat_nondefault_str' s Hq!fq!f% & & Hq!fq!f% & &(LLL (LL (LL (LL * + + Hlll *ll *ll *ll , - - tI'122333tIc!W555@AABBBiyQ@@@@ F4 #aQ@@@GK L L N N N
)a==Q1
F
F
FrY''(((((r/ N)__name__
__module__ __qualname__r. r4 r< r/ r- r r sA & & &,5 5 5) ) ) ) )r/ r c \ e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
ZdS )TestPropertiesc 6 t g dg dg dg dg }t g d }t g d j }d}t || d t || d
t || t |t j |d t |t j |d
t |t j | 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 r r )r r r )r r r r axisr N)r Tr sumr r )r$ Msum0sum1sumalls r- test_sumzTestProperties.test_sum< s
LLL << << <<!
"
" mmm$$mmm$$&4A ///4A ///VQUUWW%%%4!2!2!23334!2!2!2333VRVAYY'''''r/ c t g dg dg }t | d t | d t g dg t | d t dgdgg t t j | d t t j |d t g dg t t j |d t dgdgg t g d
}t | dk d S ) Nr r r r r r i r )r
r r x rG )r r r )r r prodr r r$ xys r- test_prodzTestProperties.test_prodO s+ IIIyyy) * *QVVXXs###QVVAYY } 5 5666QVVAYYcU| 4 4555RWQZZ%%%RWQQ'''
)>)>???RWQQ'''!se )=)=>>> 999 A
r/ c X t g dg dg }t | d t | d t g dg t | d t dgdgg t t j | d t t j |d t g dg t t j |d t dgdgg d S )NrQ rR r r r r rG )r r maxr r$ rX s r- test_maxzTestProperties.test_max\ IIIyyy) * *QUUWWa QUU1XXvyyyk 2 2333QUU1XXvsQCj 1 1222RVAYY"""RVAA&&& {(;(;<<<RVAA&&&aSz(:(:;;;;;r/ c X t g dg dg }t | d t | d t g dg t | d t dgdgg t t j | d t t j |d t g dg t t j |d t dgdgg d S )NrQ rR r r r rG )r r minr r] s r- test_minzTestProperties.test_minf r_ r/ c t j d d }t | dk t t j | d t j ddg k t t j | d t j ddg k d S )Nr )r r r r r r )r r" reshaper ptpr r r] s r- test_ptpzTestProperties.test_ptpp s IaLL ( (1 quuQxx28QF#3#3344555quuQxx28QF#3#334455555r/ c r t j d d }| t j }t | d | d t | d | d d S )N )r r r )ddofr )r r" rd viewr r var)r$ rX mxs r- test_varzTestProperties.test_varv s IaLL ( (
VVBI
QUUU]]BFFFNN333QUUU]]BFFFNN33333r/ c b dd l m} t j ddgddgg }t | }t t j | | |j t t j t j t j
| |j k t t j t j t j
| |j k t t j ||j
k |d|z z }t | }t t j | | |j t t j t j t j
| |j k t t j t j t j
| |j k d S )Nr ? @ @ @y @) numpy.linalglinalgr r r r allcloseinvIr transposerI Hr! conj)r$ rt r! r% r& mBs r- r. zTestProperties.test_basic| s %%%%%% Hr2h2h
!
!
AYY FJJqMM2400111rx Q24 78899:::rx Q24 78899:::qBDy!!"""
1H
AYY FJJqMM2400111rx Q24 78899:::rx Q 4 4 6 6"$ >??@@AAAAAr/ c t t j d dd }t ddgddgdd gg }t |j | d S )
Nr r r g:gn)t ?g#qg#q?gn)t ?g6Bq)r r r" rd r rw )r$ rX xpinvs r- test_pinvzTestProperties.test_pinv sl 29Q<<''1-- . .z2$z2$k24 5 5 AC'''''r/ c t j d dd }t | }t | dz }t t j ||dz k t t j |t |dz k t t j |t |dz
k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j ||k t t j t | dk t t j t |dk d S )Nd rS 皙?r )r r" rd r r r anyabs)r$ r! r% r{ s r- test_comparisonsz TestProperties.test_comparisons s/ IcNN " "2r * *
AYY
AYY_rQsU{##$$$rVAcE]]*++,,,BF2# .///000rBw rRx !!!rRx !!!BF27OO#$$$BF27OO#$$$rRx !!!rRx !!!BF27OO#$$$rRx !!!BF28$$$%%%rRx !!!BF3r77Q;'''(((s26{{##$$$$$r/ c t j d dd }t | }d|d< t |d |d k d S )Nr rS ir r )r r" rd r r )r$ r! r% s r-
test_asmatrixz TestProperties.test_asmatrix sT IcNN " "2r * *
a[[$$2d8#$$$$$r/ c t ddgddgg }t | t d k t | t d k d S )Nr r r ?)r r rJ meanr$ r! s r- test_noaxiszTestProperties.test_noaxis sd QFQF# $ $6!99$%%%F3KK'(((((r/ c p t ddgddgg }t t | dk d S )Nr r z matrix([[1, 0],
[0, 1]]))r r reprr s r- test_reprzTestProperties.test_repr s< QFQF# $ $Q>>?????r/ c n t d }t dgdgdgg }t || d S )NzTrue; True; FalseTF)r r )r$ r! r& s r- test_make_bool_matrix_from_strz-TestProperties.test_make_bool_matrix_from_str s@ & ' ' TFTFUG, - -1a r/ N)r= r> r? rO rZ r^ rb rf rm r. r~ r r r r r r@ r/ r- rB rB ; s ( ( (& < < << < <6 6 6 4 4 4 B B B"( ( (% % %0% % % ) ) )
@ @ @! ! ! ! !r/ rB c e Zd Zd ZdS ) TestCastingc t j d dd }t | }| }t j dt j dz }||z }t |j j t j k t t j
||k t t j
||dz k | }t j dt j }||z }t |j j t j k t t j
||k d S )Nr rS )rS rS r ) r r" rd r copyonesfloat64r dtypetyper
complex128)r$ r! r% r{ OmCs r- r. zTestCasting.test_basic s IcNN " "2r * *
AYY
WWYY GHbj ) )C /
!V
+,,,rRx !!!rRV|$$%%%
WWYY GHbm , ,
!V
.///rRx !!!!!r/ Nr= r> r? r. r@ r/ r- r r s# " " " " "r/ r c & e Zd Zd Zd Zd Zd ZdS ) TestAlgebrac dd l m} t j ddgddgg }t | }t j d }t
d D ]A}t t j ||z j | t j
|| }B| | }t j d }t
d D ]B}t t j || z j | t j
|| }Ct t j ||z j t j
|| t t j ||z j ||z t t j d|z j d|z t | }|dz }t t j |j d|z d S ) Nr ro rp rq rr r r r ) rs rt r r r identityranger ru r! dotrv )r$ rt r! r% r& iAinvmA2s r- r. zTestAlgebra.test_basic s %%%%%% Hr2hR) * *
AYY KNNq A BKq Q// 0 0 0q! AAzz!}} KNNq A BKr a00 1 1 1q$AA R"WK1 66777 R"WK!a%11222 QrTHqs,,---Qii q CE1Q3''(((((r/ c t d }| }|dz }| }|dz }| }|dz }t ||dz t |t j || t t j || t j d dS )z " "
VVXX
q
VVXX
r
WWYY
q!"ad+++!"bfRnn555!"&Q--;;;;;r/ c t ddgddgg }t j t j fD ]#} |d }t |dz ||z $d S Nr r r r )r r int8uint8r
)r$ r scalar_ttwos r- test_scalar_type_powz TestAlgebra.test_scalar_type_pow sf QFQF# $ $"(+ 8 8H(1++C %a1fa3h 7 7 7 7 8 8r/ c t ddgddgg }t t 5 d|z ddd n # 1 swxY w Y t t 5 |t z ddd dS # 1 swxY w Y dS )z:Check that 'not implemented' operations produce a failure.ro rp rq rr N)r r r8 objectr s r- test_notimplementedz TestAlgebra.test_notimplemented s RHH
9
%
% FF 9
%
%
fhhJJ s! <A A A::A>A>N)r= r> r? r. r r r r@ r/ r- r r sP ) ) )2 < < <8 8 8 r/ r c e Zd Zd ZdS )TestMatrixReturnc z t dgd }dddgfdddt j dg d }g d
}t | D ]}| d s||v r t || }t
|t j j re|
d | d ||v r || }nd } || }t t | t u d
|z t t |j t u t t |j t u t dg \ }} t t | t j u t t | t j u d S )Nro f8)r )intc) ro r )r r )astypeclipcompressrepeatrd swapaxesr )&argminchoosedumpdumpsfillgetfieldgetAgetA1itemnonzeroputputmaskresize searchsortedsetflagssetfieldsort partition argpartitiontaketofiletolisttostringtobytesr r rJ argmaxr ra r\ r rk re rV stdctypesitemset_r@ z%sr )r r r dir
startswithgetattr
isinstance collectionsabcCallabler r r r realimagr ndarray)
r$ a
methodargsexcluded_methodsattribfargsbcds
r- test_instance_methodsz&TestMatrixReturn.test_instance_methods s C5 % % % 8SE??
!ff
:
:F %%
3C)C)C6""A![_ 566
:s Z''%f-DD DAtHQ6)4&=999QV &'''QV &'''se}}$$&&1Q2:%&&&Q2:%&&&&&r/ N)r= r> r? r r@ r/ r- r r s# &' &' &' &' &'r/ r c e Zd Zd ZdS ) TestIndexingc t t j dt }t j dt }g d|d d df< |dk |d d df< t |ddgddgddgg d S )Nr r )r r )g?g?g333333?r r r r r zerosfloatr rW s r- r. zTestIndexing.test_basic9 s RXfe,, - - HVU # #!//!!!Q$c'!!!Q$Q!Q!Q!Q011111r/ Nr r@ r/ r- r r 8 s# 2 2 2 2 2r/ r c f e Zd Z eddgddgg Zd Zd Zd Zd Zd Z d
Z
d Z d Z d
Z
dS )TestNewScalarIndexingr r r r c N | j }|d }t |j d d S )Nr r )r r ndimr$ r rX s r- test_dimesionsz$TestNewScalarIndexing.test_dimesionsD s* F
aDQVQ r/ c n | j }t j ||g }t |j g d d S )N)r r r )r r r r r# r s r- test_array_from_matrix_listz1TestNewScalarIndexing.test_array_from_matrix_listI s8 F HaV QWiii(((((r/ c d | j }t | ddgddgg d S r )r r r r$ r s r- test_array_to_listz(TestNewScalarIndexing.test_array_to_listN s4 FQXXZZ1a&1a&!122222r/ c | j }|dg df }t t |t t |t g dg |ddg }t t |t t |t ddgddgg |dgdggddgddggf }t t |t t |t ddgddgg d S )Nr )r r r )r r r r r r r )r r r r r r s r- test_fancy_indexingz)TestNewScalarIndexing.test_fancy_indexingR s F
alO
1f%%&&&Q }--...
q!fI
1f%%&&&QQ!Q 011222
sQCjAq6Aq6** +
1f%%&&&QQ!a 12233333r/ c t g dg dg }t |d d t g dg t |d d j d t |d j d t |d d df j d t d }t |d d t |d d t |d d df j |j d S )NrQ rR r )r r r r r )r r r# r] s r- test_matrix_elementz)TestNewScalarIndexing.test_matrix_element^ s IIIyyy) * *QqT!Wfiii[ 1 1222QqT!W]F+++QqTZ(((Qqqq!tW]F+++ 1IIQtWa QqT1 Qqqq!tW]AG,,,,,r/ c t t j dt }t |d |d d d S )Nr r r r r] s r- test_scalar_indexingz*TestNewScalarIndexing.test_scalar_indexingj s= RXfe,, - -QtWad1g&&&&&r/ c 4 t t j 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 r r r] s r- test_row_column_indexingz.TestNewScalarIndexing.test_row_column_indexingn s RVAYY 1QqqqS6QF8,,,1QqqqS6QF8,,,1QQQT7aS1#J///1QQQT7aS1#J/////r/ c : t j d }d|_ t | }t |d d t j ddg f |d d df t |t j g d d d f |dd d f d S )Nr r TFr )TFF)r r" r# r r r r$ r! rX s r- test_boolean_indexingz+TestNewScalarIndexing.test_boolean_indexingu s IaLL QKK1QQQ$ 7 7 78!AAAqD'BBB1RX&:&:&: ; ;AAA =>!AAA#GGGGGr/ c t j d }d|_ t | }t |d d ddgf |d d d d df t |g dd d f |d d dd d f d S )Nr r r r r )r r r )r r" r# r r r s r- test_list_indexingz(TestNewScalarIndexing.test_list_indexing| s IaLL QKK1QQQAY<111ddd74441YYYqqq[>1TTrT!!!V955555r/ N)r= r> r? r r r r r r r r
r r r r@ r/ r- r r A s AA A
) ) )
3 3 3
4
4
4
-
-
-' ' '0 0 0H H H6 6 6 6 6r/ r c e Zd Zd Zd ZdS ) TestPowerc & t j ddgddgg }t t t |d t j u t
| }t t t |d t u d S Nr r r )r r r r r r r
r r s r- test_returntypezTestPower.test_returntype s{ Hq!fq!f% & &\!Q''((BJ6777 FF\!Q''((F233333r/ c Z t t ddgddggd ddgddgg d S r )r r r3 s r- test_listzTestPower.test_list s< NOOOOOr/ N)r= r> r? r r r@ r/ r- r r s7 4 4 4 P P P P Pr/ r c e Zd Z ej dgdgg Z edgdgg Zd Zd Z d Z
d Z d Z d Z
d Zd
Zd Zd S )
TestShaper r c n t | j j d t | j j d d S )Nr
)r r r# r r3 s r-
test_shapezTestShape.test_shape s0 TV\6***TV\6*****r/ c t t j | j j d t t j | j j d d S )Nr )r r ravelr r# r r3 s r- test_numpy_ravelzTestShape.test_numpy_ravel sF RXdf%%+T222RXdf%%+T22222r/ c t | j j d t | j j d d S Nr! )r r )r r r" r# r r3 s r- test_member_ravelzTestShape.test_member_ravel sB TV\\^^)4000TV\\^^)622222r/ c t | j j d t | j j d d S r% )r r flattenr# r r3 s r- test_member_flattenz TestShape.test_member_flatten sF TV^^%%+T222TV^^%%+V44444r/ c t j g dg dg }t t j | g d t t j |d g d t t j |j g d t t j |j d g d t g dg dg }t t j | g d t t j |d g d t t j |j g d t t j |j d g d d S NrQ rR )r r r r r r F)order)r r r r r r r! )r r r r" rI r r] s r- test_numpy_ravel_orderz TestShape.test_numpy_ravel_order sZ Hiii+ , ,RXa[["4"4"4555RXas+++-?-?-?@@@RXac]]$6$6$6777RXac---/A/A/ABBB IIIyyy) * *RXa[["4"4"4555RXas+++-?-?-?@@@RXac]]$6$6$6777RXac---/A/A/ABBBBBr/ c t t g dg dg }t | g dg t | d g dg t |j g dg t |j d g dg d S r+ )r r r" rI r] s r- test_matrix_ravel_orderz!TestShape.test_matrix_ravel_order s IIIyyy) * *QWWYY!3!3!3 4555QWW3W''*<*<*<)=>>>QSYY[[#5#5#5"6777QSYYSY)),>,>,>+?@@@@@r/ c t t j | j | j t t j | j | j d S N)r r may_share_memoryr r" r( r3 s r- test_array_memory_sharingz#TestShape.test_array_memory_sharing [ #DFDFLLNN;;<<<B'0@0@AAABBBBBr/ c t t j | j | j t t j | j | j d S r2 )r r r3 r r" r( r3 s r- test_matrix_memory_sharingz$TestShape.test_matrix_memory_sharing r5 r/ c 4 t j d d t j }t j |d }t
|j d t t |t j d S )NrS )r r r rG r )
r r" rd rj r expand_dimsr r r r )r$ r expandeds r- test_expand_dims_matrixz!TestShape.test_expand_dims_matrix sx
IbMM ! !& ) ) . .ry 9 9>!!,,,X]A&&&Jx33344444r/ N)r= r> r? r r r r r r r# r& r) r. r0 r4 r7 r; r@ r/ r- r r s 1#s AaSzA+ + +3 3 33 3 35 5 5
C
C
CA A AC C CC C C5 5 5 5 5r/ r )collections.abcr numpyr r r r
numpy.testingr r r r r
r rs r numpy.matrixlibr
r rB r r r r r r r r@ r/ r- r@ s@ ( ( ( ( ( ( ( ( ( ( & % % % % % ,) ,) ,) ,) ,) ,) ,) ,)^C! C! C! C! C! C! C! C!J" " " " " " " "&8 8 8 8 8 8 8 8v'' '' '' '' '' '' '' ''T2 2 2 2 2 2 2 2@6 @6 @6 @6 @6 @6 @6 @6FP P P P P P P P65 65 65 65 65 65 65 65 65 65r/