M1j d Z ddlmZmZmZmZmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZmZmZ ddlZddlZd 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"d Z#d Z$d Z%d Z&d Z'd Z( G d d Z) G d d Z* G d d Z+ G d d Z,dS ) z"Test functions for matrix module
) assert_equalassert_array_equalassert_array_max_ulpassert_array_almost_equal
assert_raisesassert_)arangeaddfliplrflipudzerosoneseyearraydiag histogram2dtri mask_indices triu_indicestriu_indices_from tril_indicestril_indices_fromvanderNc N t | }t j || }|S N)r r
outer)ndatas t/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_twodim_base.pyget_matr s# !99D 9T4 D K c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestEyec
N t t d t g dg dg dg dg t t dd t g dg dg dg dgd t t d d k t dt d S )
N r r r r r' r r r r r' r r r r r' fdtype r' )r r r boolselfs r
test_basiczTestEye.test_basic s SVVLLL (LL (LL (LL * + + , , , S#&&&LLL (LL (LL (LL * ,/0 0 1 1 1 SVVq[$''' ) ) ) ) )r! c j t t t j d t t ddgddgg t t t j d t j d t j d t g dg dg d S ) N r, r' r r% )Mkr( r) )r r npuint64intr r0 s r test_uint64zTestEye.test_uint64) s S1 S11151a&1a&9I3J3JKKKS1 1 1 FFFLLL,,,788 : : : : :r! c
t t dd t g dg dg dg dg t t dd t g dg d g dg dg d S )
Nr% r' r6 r( r) r* r r r r r& r r r r0 s r test_diagzTestEye.test_diag/ s Sa[[[LLL (LL (LL (LL * + + , , , Sb\\\LLL (LL (LL (LL * + + , , , , ,r! c
t t dd t g dg dg dg dg t t dd t g dg dg d g d S )
Nr% r. r' r r r r' r )r r r' r r r r& r( r) r? r0 s r test_2dzTestEye.test_2d< s SAYYIII %II %II %II ' ( ( ) ) ) SAYYLLL (LL (LL * + + , , , , ,r! c
t t ddd t g dg dg dg t t ddd t g d g d g d
g d g d S ) Nr. r% r4 r< r) r* r= rD rB rC r? r0 s r test_diag2dzTestEye.test_diag2dH s SA^^^LLL (LL (LL * + + , , ,
SA___III %II %II %II ' ( ( ) ) ) ) )r! c t t ddd ddgddgg t t ddd ddgddgg t t ddd ddgddgg t t ddd ddgddgg t t ddd ddgddgddgg t t ddd ddgddgddgg t t ddd ddgddgddgg t t ddd ddgddgddgg t t ddd ddgddgddgg d S )Nr4 r' r r> rG r. r r r0 s r test_eye_boundszTestEye.test_eye_boundsT s SAq\\QFQF#3444SAr]]aVaV$4555SAq\\QFQF#3444SAr]]aVaV$4555SAq\\QFQFQF#;<<<SAq\\QFQFQF#;<<<SAr]]aVaVaV$<===SAr]]aVaVaV$<===SAr]]aVaVaV$<=====r! c R t t ddd ddgddgg d S )Nr4 S3r, 1r! rK r0 s r test_stringszTestEye.test_strings_ s? SAT***SkC;/ 1 1 1 1 1r! c \ t t ddt ddgddgg d S )Nr4 r, TF)r r r/ r0 s r test_boolzTestEye.test_boolc s4 SAT***dE]UDM,JKKKKKr! c t ddd }t dddd }t || |j j sJ |j j rJ |j j rJ |j j sJ d S )Nr% r. r> r< F)r6 order)r r flags c_contiguous f_contiguous)r1 mat_cmat_fs r
test_orderzTestEye.test_orderf s AqBAqBc***UE"""{'''';++++;++++{''''''r! N) __name__
__module__ __qualname__r2 r: r@ rE rH rL rP rR r[ r! r r# r# s ) ) ) : : : , , ,
,
,
,
)
)
) > > >1 1 1L L L( ( ( ( (r! r# c . e Zd Zd ZddZd Zd Zd ZdS )TestDiagc dt d z d }t d }t d D ]}|| |||f< t t | | t d }| }t d D ]"}|| |||dz f< || ||dz |f< #t t |d | t t |d | d S ) Nd l)rd rd ) rf r4 r< rG )r astyper
ranger r copy)r1 valsbr6 cs r test_vectorzTestDiag.test_vectorq s fQii'',, &MMq A1gAadGGT$ZZ### &MM
FFHHq " "A q'AaQhK q'Aa!eQhKKT$!___a(((T$"%%%q)))))r! Nc |(dt d z dz d }t d }t d D ]}|||f ||< t t | | |dz }t d D ]}|||dz f ||< t t |d |d d t d D ]}||dz |f ||< t t |d |d d d S )
Nrc rd r' re )rd r r. r4 rG )r rg r
rh r r )r1 rj rk r6 s r test_matrixzTestDiag.test_matrix s <'!**$q(0055D $KKq A1:AaDDT$ZZ###
Eq " "A1q5>AaDDT$]]AbqbE***q " "AAq>AaDDT$^^QrrU+++++r! c z t dt d z dz dd }| | d S )Nrc rd r' rT re )rU r- )r r ro )r1 rj s r test_fortran_orderzTestDiag.test_fortran_order s@ cGAJJ&*3cBBB r! c ddgddgddgg}t t |d g t t |d dg t t |d ddg t t |d ddg t t |d
dg t t |d g d S ) Nr' r4 r. r% rd r< r r> rG rJ )r r )r1 As r test_diag_boundszTestDiag.test_diag_bounds s VaVaV $T!q\\\2&&&T!q\\\A3'''T!q\\\Aq6***T!r]]]QF+++T!r]]]QC(((T!r]]]B'''''r! c B t t t dggg d S )Nr' )r
ValueErrorr r0 s r test_failurezTestDiag.test_failure s j$1#00000r! r )r\ r] r^ rm ro rq ru rx r_ r! r ra ra p sd * * *
,
,
,
, ( ( (1 1 1 1 1r! ra c e Zd Zd ZdS )
TestFliplrc & t t t t d t d }|d d d d df }t t | | g dg dg}g dg dg}t t | | d S )Nr% r> r r' r4 r. r% rd )r4 r' r )rd r% r. )r rw r r r r r1 ark s r r2 zTestFliplr.test_basic s j&$q''222 AJJ
aaa2gJVAYY"""
YY
YY
YY
YY
VAYY"""""r! Nr\ r] r^ r2 r_ r! r rz rz s# # # # # #r! rz c e Zd Zd ZdS )
TestFlipudc t d }|d d dd d f }t t | | g dg dg}g dg dg}t t | | d S )Nr% r> r| r} )r r r r~ s r r2 zTestFlipud.test_basic s AJJ
dddAAAgJVAYY"""
YY
YY
YY
YY
VAYY"""""r! Nr r_ r! r r r s# # # # # #r! r c e Zd Zd Zd Zd Zd Zd Zd Zd Z e
j dd d
g d Z
d S )
TestHistogram2dc t g d }t g d }t j ddd }t j ddd }t ||||f d }t g dg dg dg d g d
g dg dg dg dg }t |j | t ||| d }t |j | t t
t d t
t d \ }}}t |t dd t |t j dd d t |t j dd d d S )
N)g̙
}?g% ?g$F ?g;cjY?gI+?)gp5x?g?_?gp ?gbd?gg<=?r r'
) r r r r' r r r r r ) r r r r r r r' r r ) r r r r r r r r r ) r' r r' r r r r r r ) r r' r r r r r r r ) r r7 linspacer r Tlistrh r )r1 xyxedgesyedgesHanswers r test_simplezTestHistogram2d.test_simple s L L L
N
N H H H
J
JQ2&&Q2&& 1vv . / / 2
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
* + + 13''' 1f % %a (13''' 'U2YYeBii I I661c"bkk***62;q!R#8#899962;q!R#8#899999r! c t g d }t g d }t ||dddgddggd \ }}}t g d g d
g d g d g d
g dg }t ||dz d t |t j ddd t |t j ddd d S )N)r' r' r4 r. r% r% r% rd )r' r. r4 r r' r4 r. r% )rs rd r rs rd T)rh density) r r r r )r r' r r' r )r r r' r r )r' r r r r )r r' r' r' r r r r r r' g @r. rf )r r r r r7 r r1 r r r xedyedr s r test_asymzTestHistogram2d.test_asym s *** + + *** + +!
q&!Q!Q 0$@ @ @ 3
__
__
__
__
__
"!VBY2223 Aq! 4 45553 Aq! 4 455555r! c t g d }t g d }t ||g dg dgd \ }}}t g dg dg dg dz }t ||d d S )
N) r' r4 r. r' r4 r. r' r4 r. ) r' r' r' r4 r4 r4 r. r. r. )r' r4 r. rd T)r )r' r' ?)r r ?g "@r. )r r r r s r test_densityz TestHistogram2d.test_density s --- . . --- . .!
q<<< .> > > 3
"
%
' ( ((*+ "!VQ/////r! c t j d dz dz }t ||dddgddgf \ }}}t |d d S )Nrc ?g .A)r% rd r r' )rh )r7 randomrandr r )r1 rr r r s r test_all_outliersz!TestHistogram2d.test_all_outliers s^ INN3 " $s *!!Q1v1v6FGGG 31a r! c t g g ddgddgf \ }}}t |t dgg t g g d \ }}}t |t j d d S )Nr r' binsr r% )r% r% )r r r r7 r
)r1 r edge1edge2s r
test_emptyzTestHistogram2d.test_empty s} %b"QFQF3CDDD5% Qtf
...%b"15555% Q 0 011111r! c
t g d }t g d }d}t |||df \ }}}t g dg dg dg dg dg dg dg dg dg d g
}t || t |t g d
t ||d|f \ }}}t g d g d g d
g dg }t || t |t g d
d S )N)r g >??? 1q%j11 2r
5
5
5
5
5
5
5
5
5
5
5
5
78 8
1f%%%2u%=%=%= > >?????r! c j G d d }ddg} | }t || }t ||f||fi fk t || }t ||f||fi fk t ||| }t ||f||ft | f t |||dg }t ||f||ft |dg f t t t |||g t ||| }t ||f||ft | f d S )Nc e Zd Zd ZdS )5TestHistogram2d.test_dispatch..ShouldDispatchc |||fS r r_ )r1 functiontypesargskwargss r __array_function__zHTestHistogram2d.test_dispatch..ShouldDispatch.__array_function__ s dF**r! N)r\ r] r^ r r_ r! r ShouldDispatchr s#
+
+
+
+
+r! r r' r4 r rd )weights)r r dictr Exception)r1 r xys_dr s r
test_dispatchz TestHistogram2d.test_dispatch s + + + + + + + + V n R ~'#rB77888 C ~'"cB77888 BS ) ) )^%Bx3 @AAA Bc1X . . .^%BxC81D1D1DEFFFib"C5AAAA B , , ,^%Bxc1B1B1BCDDDDDr! )x_leny_len)r r ) c t j | t j | }}t j t d 5 t || d d d d S # 1 swxY w Y d S )Nz"x and y must have the same length.)match)r7 r pytestraisesrw r )r1 r r r r s r test_bad_lengthz TestHistogram2d.test_bad_length, s wu~~rwu~~1
]:!EG G G 1 s A""A&)A&N)r\ r] r^ r r r r r r r r mark parametrizer r_ r! r r r s : : :46 6 6 0 0 0! ! !
2 2 2 @ @ @>E E E( [ /(H1EFF GF r! r c e Zd Zd ZdS )TestTric t g dg dg dg }t t d | t t dt | t d S )NrB r' r' r r' r' r' r. r, )r r r r/ rg )r1 outs r
test_dtypezTestTri.test_dtype5 sq YYY YY YY ! ! 3q663'''3q---szz$/?/?@@@@@r! N)r\ r] r^ r r_ r! r r r 4 s( A A A A Ar! r c t j d t j d z D ]} t j d| }t j | }t j | }t |ddgddgg t ||j t |j |j t |j |j d S )NAllFloat
AllInteger)r4 r4 r, r' r ) r7 typecodesr triltriur r r r- )r- r rk rl s r test_tril_triu_ndim2r = s j)BL ,FF ' ' GF% ( ( ( GAJJ GAJJ11v1v .///1ac"""QWag&&&QWag&&&&' 'r! c T t j d t j d z D ] } t j ddgddggddgddggddgddggg| }t j ddgddggddgddggddgddggg| }t j ddgddggddgddggddgddggg| }t j | }t j | }t || t || t
|j |j t
|j |j d S )Nr r r' r r, )r7 r r r r r r r- )r- r a_tril_desireda_triu_desireda_triu_observeda_tril_observeds r test_tril_triu_ndim3r I s j)BL ,FF 5 5 HVaV VaV VaV
VaV VaV VaV #
VaV VaV VaV #
'!** '!**?N;;;?N;;;_*AG444_*AG4444+5 5r! c R t j ddt j gg dt j ddgg } t j g dg dt j ddgg }|j }t t j | | t t j | | d S )Nr' r rB r )r7 r infr r r r )arrout_trilout_trius r test_tril_triu_with_infr b s
(Q26N IIVQN$ % %C x" &!Q ) * *H zHrws||X...rws||X.....r! c t j d D ]w} | dk r t j d| }t t j | j |j t t j | j |j xt j ddgddggd }t t j | j |j t t j | j |j t j dd
}t t j | j |j t t j | j |j d S ) NAllV)r. r. r, z2001-01-01T12:00z2002-02-03T13:56z2004-01-01T12:00z2003-01-03T13:45
datetime64zf4,f4)r7 r r
r r r- r r )rl r s r test_tril_triu_dtyper o s9 \%
4 4 88 hvQ'''RWS\\'333RWS\\'3333
(');<');<>% ' ' 'C #SY/// #SY///
(6
)
)
)C #SY/// #SY/////r! c T t dt j } t j d dd }t || t
g d t dt j d }t || t
g d d S )Nr. r )r r' r4 r% rd r' )r' r4 rd )r r7 r r reshaper r )iur iu1s r test_mask_indicesr s a ! !B
! Q""Aque$6$6$6 7 7888
q"'1
%
%CqvuYYY / /00000r! c
" t d } t dd }t dd }t ddd }t j g dg dg d g d
g }t j d d dd }t || t g d
t || t g d d|| < t |t g dg dg dg dg d||< t |t g dg dg dg dg d||< t |t g dg dg dg dg d||< t |t g dg d g d g d g d S ) Nr% r4 r< rd mr6 r r' r4 r. r% rd rs rf r r r r
r' )
r' rd rs r r r r r r r )
r' rs rf r r r r r r> )r> r4 r. r% )r> r> rf r )r> r> r> r r> r> r> r> )r> r4 r. r% rd )r> r> r r r )r> r> r> r r )r> r> r> r> r )r r r r% )r r r r )r r r r% rd )r r r r r )r r r r r )r r7 r r r r )il1il2il3il4r rk s r test_tril_indicesr sP
q//C
qA
C
qA
C
qA
#
#
#C
,,, ,,!//"""$ % %A !R A&&A qv A A ABBD D Dqv B B BCCE E E AcFq mmm ,nn . . . . . . 0 1 12 2 2
AcFq . . . 0 0 0 2 2 2 2 2 2 4 5 56 6 6 AcFq 0 0 0 2 2 2 2 2 2 2 2 2 4 5 56 6 6
AcFq 3 3 3 6 6 6 7 7 7 7 7 7 9 : :; ; ; ; ;r! c e Zd Zd ZdS )TestTriuIndicesc
" t d }t dd }t dd }t ddd }t j g dg dg d g d
g }t j d d dd }t || t g d
t || t g d d||< t |t g dg dg dg dg d||< t |t g dg dg dg dg d||< t |t g dg dg dg dg d||< t |t g dg d g d g dg d S ) Nr% r4 r< rd r r r r r r r' r )
r' r4 r. r% rs rf r r r r )r' r4 r. r% rd rf r r r r r r r r r> r )rd r> r> r> )r r r> r> )r r r r> )r> r> r> r> r> )rs r> r> r> r> )r r r> r> r> )r r r r> r> r )r> r> r r )rd r> r> r )r> r> r r r )rs r> r> r r )r r r> r> r )r r7 r r r r )r1 r iu2iu3iu4r rk s r test_triu_indicesz!TestTriuIndices.test_triu_indices sV 1oo1"""1"""1Q''' Hlll"ll%oo&&&(
)
)
Ia $ $Q * * 1S6 !B!B!BCC E E E1S6 ": ": ": ; ; < < <
#1 "2"2"2"1//"1//"2"2"2"4 5 5 6 6 6
#1 "6"6"6"5"5"5"6"6"6"6"6"6"8 9 9 : : : #1 "4"4"4"2"2"2"1//"2"2"2"4 5 5 6 6 6
#1 "9"9"9"7"7"7"7"7"7"6"6"6"8 9 9 : : : : :r! N)r\ r] r^ r r_ r! r r r s# .: .: .: .: .:r! r c e Zd Zd ZdS )TestTrilIndicesFromc t t t t j d t t t t j d d S N)r4 )r4 r4 r4 )r rw r r7 r r0 s r test_exceptionsz#TestTrilIndicesFrom.test_exceptions @ j"3RWT]]CCCj"3RWY5G5GHHHHHr! Nr\ r] r^ r r_ r! r r r ( I I I I Ir! r c e Zd Zd ZdS )TestTriuIndicesFromc t t t t j d t t t t j d d S r )r rw r r7 r r0 s r r z#TestTriuIndicesFrom.test_exceptions r r! Nr r_ r! r r r r r! r c e Zd Zd Zd ZdS )
TestVanderc f t j g d }t | }t j g dg dg dg dg }t ||d d dd f |j d }t d D ]2}t || }t ||d d ||z
|f 3d S ) N)r r' rG r. r )r' r' r' r' r' )r ir% rG r' )Q r r. r' r' rs N)r7 r r r shaperh )r1 rl vpowersr r s r r2 zTestVander.test_basic s H]]] # # 1II???*??---,,,. / /
1fQQQUm,,, LOq 4 4AqAA q&AaCE"2 3 3 3 3 4 4r! c P t g dt j }t | }t j g dg dg dg }t || t ddg }t |d }t j g d
g d g }t || d S ) N)r r r, )y r r' ) r% r' ) r r' ? ? ? r. r )y @r) r' )y r* r' )r r7 int8r r )r1 rl r" expecteds r test_dtypeszTestVander.test_dtypes s ---rw / / / 1II8\\\ *]] )\\ + , , 1h''' 66" # # 1NNN8]]] +^^ - . .
1h'''''r! N)r\ r] r^ r2 r- r_ r! r r r s2
4
4
4 ( ( ( ( (r! r )-__doc__
numpy.testingr r r r r r numpyr r
r r r
r r r r r r r r r r r r r7 r r r# ra rz r r r r r r r r r r r r r r_ r! r r1 sd
U( U( U( U( U( U( U( U(p,1 ,1 ,1 ,1 ,1 ,1 ,1 ,1^
#
#
#
#
#
#
#
# # # # # # # # #y y y y y y y y xA A A A A A A A ' ' '5 5 52
/
/
/0 0 0,1 1 1,; ,; ,;^/: /: /: /: /: /: /: /:dI I I I I I I II I I I I I I I ( ( ( ( ( ( ( ( ( (r!