G*ͭ d Z ddlZddlZddlZddlZddlmZmZm Z m
Z
m Z m Z m
Z
d Zd Zd Zd Zd Zd Zd
Zd Zd Zd
Zd Zd Zd Zd Zd Z d Z d Z d Z d Z G d d Z!d Z"d Z#dS )zTests 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_raisesassert_raises_regexassert_array_equalassert_almost_equalassert_array_almost_equalc t j ddgddgg } t t | g dd d f t j t j t j d dd }t |d dt j d f |d dd f j d S )
N )r r r 2
)
npmatrixr
isinstanceasmatrixarangereshaper arrayT)mxs z/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/matrixlib/tests/test_interaction.pytest_fancy_indexingr s
Aq6Aq6"##A JqAAA 2 2333 BIbMM))!R0011A2A2rx||#$aBik22222 c ddg} ddg}t j | | g }t j j || | }t t
|t j d S )Nr r r r
)r r
polynomial polyutils mapdomainr r )dom1dom2r ress r test_polynomial_mapdomainr' se
q6D
q6D
4, A
-
!
+
+AtT
:
:C JsBI & &'''''r c t j g dg } t j | d }t j g dg }t || t t | t j u d S )Nr r r axisr r r )r r sortr r typeaactualexpecteds r test_sort_matrix_noner3 ) sq
999+ A
WQT
"
"
"Fy)))%%H""" DNNbi '(((((r c t j g dg } t j | dd }t j g dg }t || t t | t j u d S )Nr) r r* r, )r r partitionr r r. r/ s r test_partition_matrix_noner6 2 su 999+ A
\!QT
*
*
*Fy)))%%H""" DNNbi '(((((r c t j ddgt } t j ddggt }t t j | d | t t j d| | d S Nr r dtyper
)r r objectr dotarrdesireds r %test_dot_scalar_and_matrix_of_objectsrA < sp
)QF&
)
)
)Ci!Q///GQ )))3 )))))r c t j d t j d z dz D ]} t j d| d }t j ddgdd gg| }t j dd
gd d gg| }t t j || | t t j || | d S )
N
AllIntegerAllFloat?r
r9 r r r r; )r typecodesr r r inner)dtscar? r@ s r test_inner_scalar_and_matrixrM E s l<(2<
+CCcI 2 2hq###B'i!Q!Q(333)aVaW -R888RXc3''111RXc3''1111 2 2r c t j ddgt } t j ddggt }t t j | d | t t j d| | d S r8 )r r r< r rJ r> s r 'test_inner_scalar_and_matrix_of_objectsrO O st
)QF&
)
)
)Ci!Q///G#q!!7+++!S!!7+++++r c
t j ddgddgg } t j d dd j }t j | |d gg dgdgddgg }t
t |j d t j u t
t |j d t j u t |j d j d t j d ddd }t t t j | |d gg dgdgddgg t j | |d gg dgdgg d g }t
t |j d t j u t
t |j d t j u t |j d j d
d S ) Nr r r
r readonly writeonlyallocate)r r )rR rS
no_subtype)r r r )r r r r r nditerr r. operandsndarrayr shaper RuntimeError)r0 bis r !test_iter_allocate_output_subtyper\ X s Aq6Aq6"##A
! Q""$A
1a, <* Z/HI K KA DA 29 ,--- DA rz 1222A $f--- ! Q1%%A, Aq$< <* Z/HIK K K 1a, <* :::< = =A DA 2: -... DA ry 0111A $i00000r c V t j ddgddgg } t j t j t j fD ]j} || }t t
| t j u || d }t t
| t j u kd S )Nr r r
r F)subok)r r
zeros_like ones_like
empty_liker r. )r0
like_functionrZ cs r rb rb s s
Aq6Aq6"##A bmC * *
M! Q29$%%% M!5 ) ) )Qry()))) * *r c t j g dg dgd } | ddd }t | |u | dd }t | | t t | t j | ddd }t | | t | |u t t | t j u d S )
Nr, )r
r r f4r9 TF)r^ copyi4)rf )r r astyper r r. r0 rZ s r test_array_astyperj ~ s 999iii(555A T..A AFOOO
E""AAa")$$$
U//AA a DGG29 $%%%%%r c t j ddgddgg } t t dt j | | g d S )Nr r r
r z shape too large to be a matrix)r r r
ValueErrorstack)r s r
test_stackrn sL Aq6Aq6"##A
$D1a&* * * * *r c t j ddgt } t j ddggt }t t j | d | t t j d| | d S r8 )r r r< r multiplyr> s r test_object_scalar_multiplyrq st
)QF&
)
)
)Ci!Q///GS!$$g...Q$$g.....r c t j t j d } t j t j fD ]} || d }t t
|t j t |j dk || d }t t
|t j t |j dk || }t t j | t j | d< t j t j fD ]}t j d 5 }t j d || d }t t
|t j t t j
t j | t t | dk d d d n # 1 swxY w Y t j d 5 }t j d || d }t t
|t j t t j |d
o5t j |d ot j |d t t | dk d
t t! |d j t$ d d d n # 1 swxY w Y t j d 5 }t j d || }t t j | t |t j k t t | dk d d d n # 1 swxY w Y d S )Nr
r r* r r
r r
r T)recordalways)r r )r r )r r zno warning raised)r r eyenanminnanmaxr r rX isscalarnanwarningscatch_warnings simplefilteranyisnanlen
issubclasscategoryRuntimeWarning)matfr& ws r test_nanfunctions_matricesr s
)BF1II
Ci
# " "a!nnn
3 **+++ V#$$$a!nnn
3 **+++ V#$$$aff C !!!!
VCFi
# ! !
$D
1
1
1 !Q !( + + +!Ca...C JsBI.. / / / rx}}--- . . . CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! !
$D
1
1
1 ?Q !( + + +!Ca...C JsBI.. / / / BHSY'' 0T0C0C,C 0 HSY / //
1
1
1 CFFaK!4 5 5 5 Jqt}n== > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?
$D
1
1
1 !Q !( + + +!C&&C BK$$ % % % C26M " " " CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! !#! !s9 B G..G2 5G2 C4LL L .A>N88N< ?N< c x t j t j d } t j t j t j t j t j t j t j fD ]} || d }t t |t j t |j dk || d }t t |t j t |j dk || }t t j
| t j t j fD ]} || d }t t |t j t |j dk || d }t t |t j t |j dk || }t t |t j t |j dk d S ) Nr
r r* rs r rt )r
r
)r rG )r r rw nanargmin nanargmaxnansumnanprodnanmeannanvarnanstdr r rX rz nancumsum
nancumprod)r r r& s r "test_nanfunctions_matrices_generalr s
)BF1II
ClBL")RZj")RY0 " "a!nnn
3 **+++ V#$$$a!nnn
3 **+++ V#$$$aff C !!!!
\2=
( ' 'a!nnn
3 **+++ V#$$$a!nnn
3 **+++ V#$$$aff
3 **+++ X%&&&& ' 'r c t j t j dd } t | d t j | d t j ddgddgg }t j ddgddgg }t j |d| }t t | t j t |dd gg d S )
Nr r r r r
r )r+ weightsg @g
@) r r randomrandr meanaverager r. )yr0 r rs r test_average_matrixr s
")..A&&''Aqvvayy"*Q"2"2333
Aq6Aq6"##A
Aq6Aq6"##A
11a(((Aa")$$$c6]O$$$$$r c t j dd } | | z }t j || }t j | }t j | }t j || }t || d S )Nr r )r linspacetrapzr r )r r r mxmymrs r test_trapz_matrixr sn AqA AA
AA 1 B 1 B "b BA r c
t t j t j d t j sJ t t j t j d d t j sJ d S )Nr )to_begin)r r ediff1dr rF r r test_ediff1d_matrixr sa bj1 .. : :;; : bj1 :::BI F FGG FGGr c d } t j ddgddgg }t j ddgddgg }t j | d| }t t |t j t || t j | d| }t t |t j t || d S )Nc | dz S Nr rF )rows r doublez,test_apply_along_axis_matrix..double
s Qwr r r r r
r r; )r r apply_along_axisr r r )r r r2 results r test_apply_along_axis_matrixr s Aq6Aq6"##Ay1a&1a&)**H
A
.
.F Jvry ) )***vx(((
A
.
.F Jvry ) )***vx(((((r c * t j ddg } t j | }t t t j | | t j t t t j || t j t t t j | | t j t t t j || t j d S r )r onesr r r. kronrW r )r0 r s r test_kron_matrixr s
AA
AAbgamm$$bj111bgamm$$bi000bgamm$$bi000bgamm$$bi00000r c e Zd Zd Zd Zd ZdS )TestConcatenatorMatrixc ddgddgt j df }t j df }t t | t j t t | t j t t j | g dg t t j | dgdgdgdgg t
t fd d S ) Nr r r
r r rc )r r r
r c . t j d f S )Nrc)r r_ri s r z4TestConcatenatorMatrix.test_matrix..2 s "%a
*; r )r r r r. r r r rl )selfab_rab_cr0 rZ s @@r test_matrixz"TestConcatenatorMatrix.test_matrix% s
F
FuS!QY uS!QY T$ZZ+++T$ZZ+++RXd^^lll^444RXd^^qcA3aS%9:::j";";";";";<<<<>22222r N)__name__
__module__ __qualname__r r r rF r r r r # sA
=
=
= / / /
3 3 3 3 3r r c H t j t 5 } t t j ddg t j ddg d d d n # 1 swxY w Y t | j }t j
d }t || d S )Nr r zl
Arrays are not equal
(shapes (2,), (1, 2) mismatch)
x: array([1, 2])
y: matrix([[1, 2]])) pytestraisesAssertionErrorr r r r strvaluetextwrapdedent)exc_infomsg
msg_references r %test_array_equal_error_message_matrixr E s ~ & & :(RXq!f%%ry!Q'8'8999: : : : : : : : : : : : : : :
hn
C O % M m$$$$$s 9A A#&A#c t j ddgg } t j dt j gg }t j dt j gg }t j t j t j gg }t j ddgt j t j gg }t t
fD ]D}| ||||fD ]:} ||| t j | } ||| ||| ;Ed S )Ng ?g @)r r r{ infr r r )m1m2m3m4m5 assert_funcr r0 s r test_array_almost_equal_matrixr T s
RH: B RL> " "B R"&M? # #B RVRV$% & &B RHrvrv ./ 0 0B02EE RR# A K1 A K1 K1 r )$__doc__r r r| numpyr
numpy.testingr r r r r r r r r' r3 r6 rA rM rO r\ rb rj rn rq r r r r r r r r r r rF r r r s
K K K K K K K K K K K K K K K K K K
3 3 3 ( ( () ) )) ) )* * *2 2 2, , ,1 1 16* * *& & &&* * */ / /&! &! &!R' ' '8
%
%
% H H H ) ) )&1 1 1 3 3 3 3 3 3 3 3D % % % r