_`&r d dl Zd dlmZ d dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZ d dlmZ d dlmZ G d deej Z G d d
Z G d d Z G d
d Z dS ) N) assert_warns)assert_ assert_equal
assert_raisesassert_array_equal) masked_array
masked_valuesmaskedallequalMaskTypegetmask MaskedArraynomasklogaddhypotdividemr_)picklec 4 e Zd ZefdZd Zed ZdS )MMatrixc \ t j | }t j | || }|S )N)datamask)npmatrixr __new__)clsr r mat_datas |/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/matrixlib/tests/test_masked_matrix.pyr zMMatrix.__new__ s+ ioo#Cc=== c p t j | | t j | | d S N)r r __array_finalize__r )selfobjs r" r& zMMatrix.__array_finalize__ s2
$$T3///&tS111r# c H | t }d|_ |S )NF)viewr _sharedmask)r' _views r" _serieszMMatrix._series s +&& ! r# N)__name__
__module__ __qualname__r r r& propertyr- r# r" r r sU &
X r# r c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestMaskedMatrixc
8 t j g dg dg }t |g dg dg }t |g dg dg }t | }t | t | t t
|d t
|d u t |d |d k t |d t u t |d |d t |d d
d f |d d
d f t |d d d f |d d d f t |d d |d d t |d
d |d
d d |d< d |d< t || d
|d d
d f< d
|d d
d f< t || t |d< t || t |d d
d f< t || |d d d f |d d d f< t |d< t t t | t j g dg dg t g dg d |d
d d f< t t t | d
t g d t t t |d
t g d t g dg d |d
d d f< t t t |d
t g d t t |d
t g d t j t j d dz }t |d }t || t t t g dt |j t d|j d S )N ) r9 r8 r7 r r )r r7 r r r7 r )r7 r7 )r r8 r r7 r8 c )r r7 )r7 r7 r ? @)r r r r7 r dtype)r r r strreprr typer
r r r
arrayaranger r r
fill_value)r' x1x2x3x4s r" test_matrix_indexingz%TestMaskedMatrix.test_matrix_indexing s
Y 999-
.
.
"IIIyyy#9
:
:
:
"IIIyyy#9
:
:
:
"
B RRX$r$x..01114BtH$%%%4F"###RXr$x(((R122Y1abb5 ***R1Xr!!!Q$x(((RUBqqqE"""RVRV$$$44R 1abb5 1abb5 R 4R 1abb5 R ad81aaa44bh 999/E&F&FGGHHH 999551aaa4Q iii)@)@AABBBA iii)@)@AABBB 999551aaa4A iii)@)@AABBBA YYY 7 788999
Yry||c)
*
*
2s
#
#R oooXFFF" " # # #S"-(((((r# c t t j t t d g ddz }t dt
j dz D ]}t j t j || }t |j
|j
t || t t |j
t j d S )N
)r7 r r7 r r r8 r< r7 )protocol)r r r listranger HIGHEST_PROTOCOLloadsdumpsr _maskr
isinstancer! )r' aproto a_pickleds r" test_pickling_subbaseclassz+TestMaskedMatrix.test_pickling_subbaseclassL s 4b ??33///A:M N N N1f 5 9:: < rA g F) r r r r flatr
r alleyer rH r )r' testcontroltestflatrZ bb01s r" test_flatzTestMaskedMatrix.test_flat_ s BIyyyk 2 2CCCTYq\1%%%TYq\6***ty1~a1f 566777BIyyyk 2 2CCC ;;; ry))) 5 5IIIFFFT7###BIyyyk 2 2CCC9 yyy) T7### 26!99--A 6 6 6
FeSXrx"b
3 3444SXrx% (8 9 9:::::r# c t j g dg dg dg }t j | }t j g dg dg dgt j }t || }|d k }|d k }t | t | t | d
t j g d t | d t j g d j t | d
t j g d t | d t j g d
j t | t | t | d
t j g d t | d t j g d j t | d
t j g d t | d t j g d j d S )N)gp=
ף?gp=
ף?g?)gQ ?g Q ?g)\(?)gףp=
?gףp=
?gffffff?)TFF)FFF)TTFrC r< g ?r )FFTr7 )TTT)
r rH r bool_r r rh anyr T)r' xXrd mXmXbigmXsmalls r" test_allany_onmatricesz'TestMaskedMatrix.test_allany_onmatricesx sj H(((((((((*
+
+
IaLL H***+++)))+24(
<
<
< !!
$
$
$c8EIIKK UYYq\\29-A-A-A#B#BCCCUYYq\\29-A-A-A#B#B#DEEEUYYq\\29-A-A-A#B#BCCCUYYq\\29-?-?-?#@#@#BCCCGKKMM!"""
W[[^^RY/B/B/B%C%CDDDW[[^^RY/D/D/D%E%E%GHHHW[[^^RY/B/B/B%C%CDDDW[[^^RY/B/B/B%C%C%EFFFFFr# c L t t j g d g d }| }t || t t
|t j t |d< | }t |g dg d S )N)r7 r8 r9 r: )r r r r r< )r r )r8 r9 r: )r r r
compressedr r rY r
)r' rZ rm s r" test_compressedz TestMaskedMatrix.test_compressed s <<<00||| D D D
LLNNQ
1bi(()))$
LLNNQ $$$$$r# c t t j g d g dg }| }t |j d t |j j |j d S )N)r7 r8 r9 r: r@ r r7 r r r r< )r7 r@ )r r r ravelr rb rX )r' rZ aravels r"
test_ravelzTestMaskedMatrix.test_ravel se ???33???:K L L LV\6***V\'11111r# c t t t j d t j d }t j | }t |dt fdt fg }d|j d< |
t dft j }t || t t |t j t t |t d S )NrQ rZ rm rC r= r r8 )rS zipr rI randomrandrH r floatr r* r r r rY r )r' iteratorr rZ rj s r" test_viewzTestMaskedMatrix.test_view s BIbMM29>>"+=+=>>??x!! 3,e )E F F Fq vvuaj"),,T4
4++,,,Jt[11122222r# N) r. r/ r0 rO r] re ro ry r| r r r2 r# r" r4 r4 s *) *) *)X< < <5 5 5; ; ;2G G G4% % %2 2 2 3 3 3 3 3r# r4 c , e Zd Zd Zd Zd Zd Zd ZdS )TestSubclassingc j t j dd }t |g d }||f| _ d S )Nr@ r rC r~ r< )r rI r r r' rt mxs r" setup_methodz TestSubclassing.setup_method s; Iaw ' ' '
Q___
-
-
-G r# c r | j \ }}t t |j t j d S r% )r r rY r! r r r s r" test_maskedarray_subclassingz,TestSubclassing.test_maskedarray_subclassing s/ )B
28RY//00000r# c 2 | j \ }}t j d 5 t t t | t t t | t j | d d d d S # 1 swxY w Y d S )Nignore)r )r r errstater rY r r r r s r" test_masked_unary_operationsz,TestSubclassing.test_masked_unary_operations s )B
[
)
)
) , , Js2ww00 1 1 1 Q + + + , , , , , , , , , , , , , , , , , ,s A B BBc | j \ }}t t t || t t t t || t t t || ||z t t t || j t j t t 5 t t t j || t d d d n # 1 swxY w Y t t t || t t t t || t d S r% )
r r rY r r r r! r r r DeprecationWarningouterr r s r" test_masked_binary_operationsz-TestSubclassing.test_masked_binary_operations s\ )B
3r2;;00111
3r1::w//000SQZZA&&&
3r2;;,bi88999
,
-
- < < JsyR 0 0':: ; ; ; < < < < < < < < < < < < < < <
5R=='22333
5Q<<1122222s 6D D #D c | j \ }}t |j |j }t t t
|| t t t t
|| t t t
|| t
|| d S )Nr< ) r r __array__r r rY r r r )r' rt r xmxs r" test_masked_binary_operations2z.TestSubclassing.test_masked_binary_operations2 s )B27,,..RW===
6"b>>733444
6"a=='22333VB^^VC%5%566666r# N)r. r/ r0 r r r r r r2 r# r" r r s_
1 1 1
, , , 3 3 3 7 7 7 7 7r# r c e Zd Zd Zd ZdS )TestConcatenatorc F t t j j d d S )Nc t d S )Nz
1, 2; 3, 4r r2 r# r" z6TestConcatenator.test_matrix_builder.. s S -> r# )r r maMAError)r' s r" test_matrix_builderz$TestConcatenator.test_matrix_builder s bem%>%>?????r# c Z t d }t j t j d }t || t
t | t | t
t |j t |j d S )N)rr7 r8 r9 ) r r r rH r_r r rG r )r' actualexpecteds r" test_matrixz TestConcatenator.test_matrix s \"5;;ru\ 23368,,, T&\\4>>222T&+&&X](;(;<<<<r s & & & & & &4 4 4 4 4 4 4 4 4 4 4 44 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 k29 &K3 K3 K3 K3 K3 K3 K3 K3\(7 (7 (7 (7 (7 (7 (7 (7T= = = = = = = = = =r#