idIdZddlmZmZmZmZmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZddlZddlZdZGddZGdd ZGd d ZGd d Z GddZ!GddZ"dZ#dZ$dZ%dZ&dZ'dZ(GddZ)GddZ*GddZ+GddZ,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_fromvanderNcNt|}tj||}|SN)r r outer)ndatas U/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/lib/tests/test_twodim_base.pyget_matr s# !99D 9T4 D KcDeZdZdZdZdZdZdZdZdZ dZ d Z d S) TestEyec Nttdtgdgdgdgdgttddtgdgdgdgdgdttdd ktdtdS) Nrrrrr'rrrrr'rrrrr'fdtyper')rrrboolselfs r test_basiczTestEye.test_basicsSVVLLL(LL(LL(LL*++ , , , S#&&&LLL(LL(LL(LL*,/00 1 1 1 SVVq[$''' ) ) ) ) )r!c jtttjdtt ddgddggtttjdtjdtjdt gdgdgdS) Nr,r'rr%)Mkr(r))rrnpuint64intrr0s r test_uint64zTestEye.test_uint64)sS1S11151a&1a&9I3J3JKKKS111FFFLLL,,,788 : : : : :r!c ttddtgdgdgdgdgttddtgdgd gdgdgdS) Nr%r'r6r(r)r*rrrrr&rrrr0s r test_diagzTestEye.test_diag/sSa[[[LLL(LL(LL(LL*++ , , , Sb\\\LLL(LL(LL(LL*++ , , , , ,r!c ttddtgdgdgdgdgttddtgdgdgd gdS) Nr%r.r'rrrr'r)rrr'rrrr&r(r)r?r0s rtest_2dzTestEye.test_2d<sSAYYIII%II%II%II'(( ) ) ) SAYYLLL(LL(LL*++ , , , , ,r!c ttdddtgdgdgdgttdddtgd gd gd gd gdS) Nr.r%r4r<r)r*r=rDrBrCr?r0s r test_diag2dzTestEye.test_diag2dHsSA^^^LLL(LL(LL*++ , , , SA___III%II%II%II'(( ) ) ) ) )r!cttdddddgddggttdddddgddggttdddddgddggttdddddgddggttdddddgddgddggttdddddgddgddggttdddddgddgddggttdddddgddgddggttdddddgddgddggdS)Nr4r'rr>rGr.rrr0s rtest_eye_boundszTestEye.test_eye_boundsTsSAq\\QFQF#3444SAr]]aVaV$4555SAq\\QFQF#3444SAr]]aVaV$4555SAq\\QFQFQF#;<<<SAq\\QFQFQF#;<<<SAr]]aVaVaV$<===SAr]]aVaVaV$<===SAr]]aVaVaV$<=====r!cRttdddddgddggdS)Nr4S3r,1r!rKr0s r test_stringszTestEye.test_strings_s?SAT***SkC;/ 1 1 1 1 1r!c\ttddtddgddggdS)Nr4r,TF)rrr/r0s r test_boolzTestEye.test_boolcs4SAT***dE]UDM,JKKKKKr!ctddd}tdddd}t|||jjsJ|jjrJ|jjrJ|jjsJdS)Nr%r.r>r<F)r6order)rrflags c_contiguous f_contiguous)r1mat_cmat_fs r test_orderzTestEye.test_orderfsAqBAqBc***UE"""{'''';++++;++++{''''''r!N) __name__ __module__ __qualname__r2r:r@rErHrLrPrRr[r!rr#r#s))) ::: , , , , , , ) ) ) > > >111LLL(((((r!r#c.eZdZdZddZdZdZdZdS)TestDiagc dtdzd}td}tdD]}|||||f<t t ||td}|}tdD]"}|||||dzf<||||dz|f<#t t |d|t t |d|dS) Ndl)rdrd)rfr4r<rG)r astyper rangerrcopy)r1valsbr6cs r test_vectorzTestDiag.test_vectorqsfQii'',, &MMq  A1gAadGGT$ZZ### &MM FFHHq " "Aq'AaQhKq'Aa!eQhKKT$!___a(((T$"%%%q)))))r!Nc|(dtdzdzd}td}tdD]}|||f||<t t |||dz}tdD]}|||dzf||<t t |d|ddtdD]}||dz|f||<t t |d |dddS) Nrcrdr're)rdrr.r4rG)r rgr rhrr)r1rjrkr6s r test_matrixzTestDiag.test_matrixs <'!**$q(0055D $KKq  A1:AaDDT$ZZ### Eq " "A1q5>AaDDT$]]AbqbE***q " "AAq>AaDDT$^^QrrU+++++r!cztdtdzdzdd}||dS)Nrcrdr'rTre)rUr-)rr ro)r1rjs rtest_fortran_orderzTestDiag.test_fortran_orders@cGAJJ&*3cBBB r!cddgddgddgg}tt|dgtt|ddgtt|dddgtt|d ddgtt|d dgtt|d gdS) Nr'r4r.r%rdr<rr>rGrJ)rr)r1As rtest_diag_boundszTestDiag.test_diag_boundssVaVaV $T!q\\\2&&&T!q\\\A3'''T!q\\\Aq6***T!r]]]QF+++T!r]]]QC(((T!r]]]B'''''r!cBtttdgggdS)Nr')r ValueErrorrr0s r test_failurezTestDiag.test_failures j$1#00000r!r)r\r]r^rmrorqrurxr_r!rrarapsd * * * , , , ,(((11111r!raceZdZdZdS) TestFliplrc&ttttdt d}|dddddf}t t||gdgdg}gdgdg}t t||dS)Nr%r>rr'r4r.r%rd)r4r'r)rdr%r.)rrwr rr rr1arks rr2zTestFliplr.test_basicsj&$q''222 AJJ aaa2gJVAYY""" YY YY  YY YY VAYY"""""r!Nr\r]r^r2r_r!rrzrzs# # # # # #r!rzceZdZdZdS) TestFlipudctd}|dddddf}tt||gdgdg}gdgdg}tt||dS)Nr%r>r|r})r rr r~s rr2zTestFlipud.test_basics AJJ dddAAAgJVAYY""" YY YY  YY YY VAYY"""""r!Nrr_r!rrrs######r!rceZdZdZdZdZdZdZdZdZ e j dd d gd Z d S) TestHistogram2dc tgd}tgd}tjddd}tjddd}t||||fd}tgdgdgdgd gd gdgdgdgdg }t |j|t|||d}t |j|tt tdt td\}}}t |tddt |tjdd d t |tjdd d dS) N)g̙ }?g% ?g$F?g;cjY?gI+?)gp5x?g?_?gp?gbd?gg<=?rr' ) rrrr'rrrrr) rrrrrrr'rr) rrrrrrrrr) r'rr'rrrrrr) rr'rrrrrrr ) rr7linspacerrTlistrhr)r1xyxedgesyedgesHanswers r test_simplezTestHistogram2d.test_simples  L L L N N  H H H J JQ2&&Q2&& 1vv. / / 2 ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( * + + 13''' 1f % %a (13''''U2YYeBiiII661c"bkk***62;q!R#8#899962;q!R#8#899999r!c tgd}tgd}t||dddgddggd\}}}tgd gd gd gd gd gdg}t||dz dt|t jdddt|t jddddS)N)r'r'r4r.r%r%r%rd)r'r.r4rr'r4r.r%)rsrdrrsrdT)rhdensity)rrrr)rr'rr'r)rrr'rr)r'rrrr)rr'r'r'rrrrrr'g @r.rf)rrrrr7rr1rrrxedyedrs r test_asymzTestHistogram2d.test_asyms *** + + *** + +! q&!Q!Q 0$@@@ 3    __ __ __ __ __   "!VBY2223 Aq! 4 45553 Aq! 4 455555r!ctgd}tgd}t||gdgdgd\}}}tgdgdgdgdz }t||d dS) N) r'r4r.r'r4r.r'r4r.) r'r'r'r4r4r4r.r.r.)r'r4r.rdT)r)r'r'?)rr?g"@r.)rrrrs r test_densityzTestHistogram2d.test_densitys --- . . --- . .! q<<<.>>> 3 " % '(((*+ "!VQ/////r!ctjddzdz}t||dddgddgf\}}}t |ddS)Nrc?g.A)r%rdrr')rh)r7randomrandrr)r1rrrrs rtest_all_outliersz!TestHistogram2d.test_all_outlierss^ INN3  " $s *!!Q1v1v6FGGG 31a     r!ctggddgddgf\}}}t|tdggtggd\}}}t|tjddS)Nrr'binsrr%)r%r%)rrrr7r )r1redge1edge2s r test_emptyzTestHistogram2d.test_emptys}%b"QFQF3CDDD5%Qtf ...%b"15555%Q 0 011111r!c tgd}tgd}d}t|||df\}}}tgdgdgdgdgdgdgdgdgdgd g }t||t|tgd t||d|f\}}}tgd gd gd gdg}t||t|tgd dS)N)rg>???1q%j11 2r 5 5 5 5 5 5 5 5 5 5 5 5 788 1f%%%2u%=%=%=>>?????r!c jGdd}ddg}|}t||}t||f||fifkt||}t||f||fifkt|||}t||f||ft|ft|||dg}t||f||ft|dgfttt|||gt|||}t||f||ft|fdS)NceZdZdZdS)5TestHistogram2d.test_dispatch..ShouldDispatchc |||fSrr_)r1functiontypesargskwargss r__array_function__zHTestHistogram2d.test_dispatch..ShouldDispatch.__array_function__sdF**r!N)r\r]r^rr_r!rShouldDispatchrs# + + + + +r!rr'r4rrd)weights)rrdictr Exception)r1rxys_drs r test_dispatchzTestHistogram2d.test_dispatchs + + + + + + + +Vn R ~'#rB77888 C ~'"cB77888 BS ) ) )^%Bx3@AAA Bc1X . . .^%BxC81D1D1DEFFFib"C5AAAA B , , ,^%Bxc1B1B1BCDDDDDr!)x_leny_len)rr)ctj|tj|}}tjtd5t ||ddddS#1swxYwYdS)Nz"x and y must have the same length.)match)r7rpytestraisesrwr)r1rrrrs rtest_bad_lengthzTestHistogram2d.test_bad_length,swu~~rwu~~1 ]:!EGGG   1                     sA""A&)A&N)r\r]r^rrrrrrrrmark parametrizerr_r!rrrs:::4666 000!!! 222@@@>EEE( [/(H1EFFGFr!rceZdZdZdS)TestTrictgdgdgdg}ttd|ttdt|tdS)NrBr'r'rr'r'r'r.r,)rrrr/rg)r1outs r test_dtypezTestTri.test_dtype5sqYYYYYYY !! 3q663'''3q---szz$/?/?@@@@@r!N)r\r]r^rr_r!rrr4s(AAAAAr!rctjdtjdzD]}tjd|}tj|}tj|}t |ddgddggt ||jt|j|jt|j|jdS)NAllFloat AllInteger)r4r4r,r'r) r7 typecodesrtriltriurrrr-)r-rrkrls rtest_tril_triu_ndim2r=sj)BL,FF'' GF% ( ( ( GAJJ GAJJ11v1v.///1ac"""QWag&&&QWag&&&&''r!cTtjdtjdzD]}tjddgddggddgddggddgddggg|}tjddgddggddgddggddgddggg|}tjddgddggddgddggddgddggg|}tj|}tj|}t ||t ||t |j|jt |j|jdS)Nrrr'rr,)r7rrrrrrr-)r-ra_tril_desireda_triu_desireda_triu_observeda_tril_observeds rtest_tril_triu_ndim3rIsj)BL,FF55 HVaV VaV VaV      VaV VaV VaV #  VaV VaV VaV #  '!**'!**?N;;;?N;;;_*AG444_*AG4444+55r!cRtjddtjggdtjddgg}tjgdgdtjddgg}|j}t tj||t tj||dS)Nr'rrBr)r7rinfrrrr)arrout_trilout_trius rtest_tril_triu_with_infrbs (Q26NIIVQN$ % %Cx"&!Q)**HzHrws||X...rws||X.....r!ctjdD]w}|dkr tjd|}ttj|j|jttj|j|jxtjddgddggd }ttj|j|jttj|j|jtjdd }ttj|j|jttj|j|jdS) NAllV)r.r.r,z2001-01-01T12:00z2002-02-03T13:56z2004-01-01T12:00z2003-01-03T13:45 datetime64zf4,f4)r7rr rrr-rr)rlrs rtest_tril_triu_dtyperos9\% 44 88 hvQ'''RWS\\'333RWS\\'3333 (');<');<>% ' ' 'C#SY///#SY/// (6 ) ) )C#SY///#SY/////r!cTtdtj}tjddd}t ||t gdtdtjd}t ||t gddS)Nr.r)rr'r4r%rdr')r'r4rd)rr7rr reshaperr)iuriu1s rtest_mask_indicesrs a ! !B ! Q""Aque$6$6$677888 q"'1 % %CqvuYYY//00000r!c "td}tdd}tdd}tddd}tjgdgdgd gd g}tjd d dd}t ||tgd t ||tgdd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgdS)Nr%r4r<rdmr6rr'r4r.r%rdrsrfrrrr  r') r'rdrsrrrrrrr) r'rsrfrrrrrr>)r>r4r.r%)r>r>rfr)r>r>r>rr>r>r>r>)r>r4r.r%rd)r>r>rrr)r>r>r>rr)r>r>r>r>r)rrrr%)rrrr)rrrr%rd)rrrrr)rrrrr)rr7rr rr)il1il2il3il4rrks rtest_tril_indicesrsP q//C qA   C qA   C qA # # #C ,,,,,!//"""$ % %A !R  A&&AqvAAABBDDDqvBBBCCEEEAcFqmmm,nn......011222 AcFq...000222222455666 AcFq000222222222455666 AcFq3336667777779::;;;;;r!ceZdZdZdS)TestTriuIndicesc "td}tdd}tdd}tddd}tjgdgdgd gd g}tjd d dd}t ||tgd t ||tgdd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgd||<t |tgdgdgdgdgdS)Nr%r4r<rdrrrrrrr'r) r'r4r.r%rsrfrrrr)r'r4r.r%rdrfrrrrrrrrr>r)rdr>r>r>)rrr>r>)rrrr>)r>r>r>r>r>)rsr>r>r>r>)rrr>r>r>)rrrr>r>r)r>r>rr)rdr>r>r)r>r>rrr)rsr>r>rr)rrr>r>r)rr7rr rr)r1riu2iu3iu4rrks rtest_triu_indicesz!TestTriuIndices.test_triu_indicessV1oo1"""1"""1Q''' Hlll"ll%oo&&&( ) ) Ia   $ $Q * * 1S6 !B!B!BCC E E E1S6 ":":":;; < < < #1 "2"2"2"1//"1//"2"2"2"455 6 6 6 #1 "6"6"6"5"5"5"6"6"6"6"6"6"899 : : :#1 "4"4"4"2"2"2"1//"2"2"2"455 6 6 6 #1 "9"9"9"7"7"7"7"7"7"6"6"6"899 : : : : :r!N)r\r]r^rr_r!rr r s#.:.:.:.:.:r!r ceZdZdZdS)TestTrilIndicesFromcttttjdttttjddSN)r4)r4r4r4)rrwrr7rr0s rtest_exceptionsz#TestTrilIndicesFrom.test_exceptions@j"3RWT]]CCCj"3RWY5G5GHHHHHr!Nr\r]r^rr_r!rrr(IIIIIr!rceZdZdZdS)TestTriuIndicesFromcttttjdttttjddSr)rrwrr7rr0s rrz#TestTriuIndicesFrom.test_exceptionsrr!Nrr_r!rrrrr!rceZdZdZdZdS) TestVandercftjgd}t|}tjgdgdgdgdg}t||ddddf|jd}t dD]2}t||}t||dd||z |f3dS) N)rr'rGr.r)r'r'r'r'r')rir%rGr')Qrr.r'r'rsN)r7rrrshaperh)r1rlvpowersrrs rr2zTestVander.test_basics H]]] # # 1II???*??---,,,.// 1fQQQUm,,, LOq 4 4AqAA q&AaCE"2 3 3 3 3 4 4r!cPtgdtj}t|}tjgdgdgdg}t ||tddg}t|d }tjgd gd g}t ||dS) N)rrr,)yrr')r%r')rr'???r.r)y@r)r')yr*r')rr7int8rr)r1rlr"expecteds r test_dtypeszTestVander.test_dtypess ---rw / / / 1II8\\\*]])\\+,, 1h''' 66" # # 1NNN8]]]+^^-.. 1h'''''r!N)r\r]r^r2r-r_r!rrrs2 4 4 4(((((r!r)-__doc__ numpy.testingrrrrrrnumpyr r r r r rrrrrrrrrrrrr7rr r#rarzrrrrrrrrrr rrrr_r!rr1sd   U(U(U(U(U(U(U(U(p,1,1,1,1,1,1,1,1^ # # # # # # # # # # # # # # # #yyyyyyyyxAAAAAAAA ' ' '5552 / / /000,111,;,;,;^/:/:/:/:/:/:/:/:dIIIIIIIIIIIIIIII((((((((((r!