bg"ddlZddlmZddlmZmZmZmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlmZddlmZGddeejZGd d ZGd d ZGd dZdS)N) assert_warns)assert_ assert_equal assert_raisesassert_array_equal) masked_array masked_valuesmaskedallequalMaskTypegetmask MaskedArraynomasklogaddhypotdividemr_)picklec4eZdZefdZdZedZdS)MMatrixc\tj|}tj|||}|S)N)datamask)npmatrixr__new__)clsrrmat_datas {/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/matrixlib/tests/test_masked_matrix.pyrzMMatrix.__new__s+ioo#Cc=== cptj||tj||dSN)rr__array_finalize__r)selfobjs r"r&zMMatrix.__array_finalize__s2 $$T3///&tS111r#cH|t}d|_|S)NF)viewr _sharedmask)r'_views r"_serieszMMatrix._seriess  +&&! r#N)__name__ __module__ __qualname__rrr&propertyr-r#r"rr sU &  Xr#rc>eZdZdZdZdZdZdZdZdZ dZ d S) TestMaskedMatrixc 8tjgdgdg}t|gdgdg}t|gdgdg}t|}t|t |t t |dt |dut |d|dkt |dtut|d|dt|d d df|d d dft|ddd f|ddd ft|dd|ddt|d d|d dd |d<d |d<t||d |d d df<d |d d df<t||t|d<t||t|d d df<t|||d ddf|d ddf<t|d<t tt|tj gdgdgtgdgd|d ddf<t tt|d tgdt tt|d tgdtgdgd|d ddf<t tt|d tgdt t|d tgdtjtj ddz}t|d}t||t ttgdt|jtd|jdS)N)r9r8r7rr)rr7rrr7r)r7r7)rr8rr7r8 c)rr7)r7r7r?@)rrrr7rdtype)rrrstrreprrtyper rr 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$$$44R1abb5 1abb5 R4R1abb5 Rad81aaa44bh 999/E&F&FGGHHH 999551aaa4Qiii)@)@AABBBAiii)@)@AABBB 999551aaa4Aiii)@)@AABBBA YYY 7 788999 Yry||c) * * 2s # #RoooXFFF"" # # #S"-(((((r#c ttjtt dgddz}t dt jdzD]}t jt j||}t|j |j t||tt|j tjdS)N )r7rr7rrr8r<r7)protocol)rrrlistrangerHIGHEST_PROTOCOLloadsdumpsr_maskr isinstancer!)r'aproto a_pickleds r"test_pickling_subbaseclassz+TestMaskedMatrix.test_pickling_subbaseclassLs 4b ??33///A:M N N N1f59:: < rAgF) rrrrflatr ralleyerrHr)r'testcontroltestflatrZbb01s r" test_flatzTestMaskedMatrix.test_flat_sBIyyyk22CCCTYq\1%%%TYq\6***ty1~a1f566777BIyyyk22CCC ;;; ry)))55IIIFFFT7###BIyyyk22CCC9yyy) T7### 26!99--A 6 6 6 FeSXrx"b 33444SXrx%(899:::::r#ctjgdgdgdg}tj|}tjgdgdgdgtj}t ||}|d k}|d k}t | t |t|d tjgd t|d tjgd j t|d tjgd t|d tjgd j t | t |t|d tjgdt|d tjgdj t|d tjgdt|d tjgdj dS)N)gp= ף?gp= ף?g?)gQ?gQ?g)\(?)gףp= ?gףp= ?gffffff?)TFF)FFF)TTFrCr<g?r)FFTr7)TTT) rrHrbool_rrrhanyrT)r'xXrdmXmXbigmXsmalls r"test_allany_onmatricesz'TestMaskedMatrix.test_allany_onmatricesxsj 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#cLttjgdgd}|}t ||t t |tjt|d<|}t |gdgdS)N)r7r8r9r:)rrrrr<)rr)r8r9r:)rrr compressedrrrYr )r'rZrms r"test_compressedz TestMaskedMatrix.test_compresseds <<<00||| D D D LLNNQ 1bi(()))$ LLNNQ $$$$$r#cttjgdgdg}|}t |jdt |jj|jdS)N)r7r8r9r:r@rr7rrrr<)r7r@)rrrravelrrbrX)r'rZaravels r" test_ravelzTestMaskedMatrix.test_ravelse ???33???:K L L LV\6***V\'11111r#ctttjdtjd}tj|}t|dtfdtfg}d|j d<| tdftj }t||tt|tj tt|t dS)NrQrZrmrCr=rr8)rSziprrIrandomrandrHrfloatrr*rrrrYr)r'iteratorrrZrjs r" test_viewzTestMaskedMatrix.test_viewsBIbMM29>>"+=+=>>??x!! 3,e )E F F Fq vvuaj"),,T4    4++,,,Jt[11122222r#N) r.r/r0rOr]reroryr|rrr2r#r"r4r4s*)*)*)X<<<555;;;2GGG4%%%222 3 3 3 3 3r#r4c,eZdZdZdZdZdZdZdS)TestSubclassingcjtjdd}t|gd}||f|_dS)Nr@rrCr~r<)rrIrrr'rtmxs r" setup_methodzTestSubclassing.setup_methods; Iaw ' ' ' Q___ - - -G r#cr|j\}}tt|jtjdSr%)rrrYr!rrrs r"test_maskedarray_subclassingz,TestSubclassing.test_maskedarray_subclassings/)B 28RY//00000r#c2|j\}}tjd5tt t |t tt |tj|ddddS#1swxYwYdS)Nignore)r)rrerrstaterrYrrrrs r"test_masked_unary_operationsz,TestSubclassing.test_masked_unary_operationss)B [ ) ) ) , , Js2ww00 1 1 1 Q + + + , , , , , , , , , , , , , , , , , ,sAB  BBc |j\}}ttt||tttt||tt t||||zttt||jtjtt5tttj ||tdddn #1swxYwYttt||tttt||tdSr%) rrrYrrrr!rrrDeprecationWarningouterrrs r"test_masked_binary_operationsz-TestSubclassing.test_masked_binary_operationss\)B 3r2;;00111 3r1::w//000SQZZA&&& 3r2;;,bi88999 , - - < < JsyR00':: ; ; ; < < < < < < < < < < < < < < < 5R=='22333 5Q<<1122222s6DD #D c|j\}}t|j|j}t t t ||tt t t ||ttt ||t ||dS)Nr<) rr __array__rrrYrrr)r'rtrxmxs r"test_masked_binary_operations2z.TestSubclassing.test_masked_binary_operations2s)B27,,..RW=== 6"b>>733444 6"a=='22333VB^^VC%5%566666r#N)r.r/r0rrrrrr2r#r"rrs_ 111 ,,, 3 3 377777r#rceZdZdZdZdS)TestConcatenatorcFttjjddS)NctdS)Nz 1, 2; 3, 4rr2r#r"z6TestConcatenator.test_matrix_builder..s S->r#)rrmaMAError)r's r"test_matrix_builderz$TestConcatenator.test_matrix_builders bem%>%>?????r#cZtd}tjtjd}t ||t t|t|t t|jt|jdS)N)rr7r8r9) rrrrHr_rrrGr)r'actualexpecteds r" test_matrixzTestConcatenator.test_matrixs\"5;;ru\23368,,, T&\\4>>222T&+&&X](;(;<<<<rs&&&&&&4444444444444444444444444444444444444444 k29&K3K3K3K3K3K3K3K3\(7(7(7(7(7(7(7(7T==========r#