`{PddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZGddZGddZGdd ZGd d ZGd d ZGddZdZGddZdZGddZ dZ!dS)N)assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal assert_raisesassert_raises_regex) mgridogrid ndenumerate fill_diagonal diag_indicesdiag_indices_from index_expndindexr_s_ix_ceZdZdZdZdZdZdZdZdZ e j dgd d Z d Zd S) TestRavelUnravelIndexc ttjdddttjdddtt5tjdddddn #1swxYwYtt5tjdddddn #1swxYwYtt5tjdd dddn #1swxYwYtt5tjdd dddn #1swxYwYttjdddttjddd ttjd ddtt tjd dtttjd dtt tjddtt tjddtt tjddtt tjddtt tjddtttjddttjddgdttjgdddtjgdgdg}ttj|dgdttj|ddgdttj|dd gd!ttj|d"d# gd$ttjd%d&d'ttjtjgddgdgdgttjtjgdddgdgdgttjd'd&gd%dS)(N)rrr)indicesshape)rhape)r)^)ims)dims)rBg?)r)rr)rr&)rr)皙?g.)r%)rrr%)r)r*r*)r%r)r*))%Forder)r. )r%r*clipmode)r-r%r%)r5wrap) r4r4)r)rr%r)r*r, iU)rnp unravel_indexr TypeErrorravel_multi_index ValueErrorarray)selfarrs u/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_index_tricks.py test_basicz TestRavelUnravelIndex.test_basics/R%a00&999 R%a,2444&, . . . 9 % % 5 5  QV 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 59 % % - -  QV , , , , - - - - - - - - - - - - - - -9 % % 0 0  Sh / / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 09 % % 1 1  Sx 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 R)&&991===R%c844g>>>R)'8<>>j""6HHHj""6GGGj""6HHHj""6GGGi!5y&IIIR%oyAA999MMM  I 6 6 I I Ih 999-..R)#v66 EEE  fC 8 8 8,,, H H H  f6 : : :LLL J J JR)#v > >  XqM E E Es L L Lj  )hqj!_C I I I I I IrTc &tjtjtjtjtjtjfD]}tjgdgdg|}d}d|dz|dz}ttj |||t|tj |||dd|dzz}ttj ||d |t|tj ||d tjgdgdgd g|}d }d d|dz|dzz|dz}ttj |||t|tj |||dd|dd|dzzzz}ttj ||d |t|tj ||d dS)N)rrrrr)r%)rr*rr)rrrL)r+r=r=rrr+r0r1)rr)rrr>r+)r+r=rJrJr) r?int16uint16int32uint32r[uint64rDrrBr@)rErMcoordsruncoordss rG test_dtypesz!TestRavelUnravelIndex.test_dtypesrs5h 28i296 O OEX###%7%7%78GGGFE{6!9,H -fe<>> J J J !1(E!M!M!M N N NX###%7%7%79K9K9KLFE1VAY;vay01&);H -fe<>> J J J !1(E!M!M!M N N N N/ O OrTc<ttjgdddtjgddttjgdddtjgddtttjgdddS)N)r+rr$r)r%r)r,r<r;r6)rrr*r)r;raiser5rk)rrrr)rr?rBrrCrEs rGtest_clipmodesz$TestRavelUnravelIndex.test_clipmodess   F K K K  } = = ? ? ? R)---0PQQQ),,, FF H H H  ,mmm] L L L L LrTctjgdd\}}t|jjt|jjdS)Nrrr))r%r+)r?r@rflags writeable)rExys rGtest_writeabilityz'TestRavelUnravelIndex.test_writeabilitysH 6221!"""!"""""rTctjdd}t|dttdtjdgdttdtjdgddS)NrrKz0d arrayz out of boundsr)r?r@rr rCrErrs rGtest_0dzTestRavelUnravelIndex.test_0dsp  Q # #QJ B4Dqc2NNN )9A3 D D D D DrTr7)r5r;rkcNtjtjdtjd|}|jdksJt t 5tjtjdtjd|ddddS#1swxYwYdS)N)r)rrLrrrr6r)r)r)r?rBzerosrZrrrC)rEr7ress rGtest_empty_array_ravelz,TestRavelUnravelIndex.test_empty_array_ravels"HV27333YTKKKyD    : & & K K  HV27333YT K K K K K K K K K K K K K K K K K K K K K Ks6BB!BcRtjtjdtjd}t |dksJt d|DsJt t5tjdgdddddS#1swxYwYdS)NrrLryr)c3,K|]}|jdkVdS)rzN)r).0as rG zATestRavelUnravelIndex.test_empty_array_unravel..s(00q17d?000000rTr)r?r@r{rZlenallrrC)rEr|s rGtest_empty_array_unravelz.TestRavelUnravelIndex.test_empty_array_unravelsrx9999EE3xx1}}}}00C00000110 : & & - -  aS) , , , - - - - - - - - - - - - - - - - - -s8BB #B N)__name__ __module__ __qualname__rHrSr`rirmrtrwpytestmark parametrizer}rrKrTrGrrs5I5I5In"""(III0OOO6 L L L### DDD [V%>%>%>??KK@?K-----rTrceZdZdZdZdZdZej dddgdZ d Z d Z d Z d S) TestGridctddd}tddd}t|jdkt|jdkt|ddkt|ddt|ddkt|d|dz ddt|d|dd zdt|d|dz d ddS) Nr$r$@r')rJ)r ggfffff?gqq?)r rrr)rErbs rGrHzTestGrid.test_basics "Qs(O "Qs(O5 !!!5 !!!! AbE1%%%! AaD1IsB///AbE1Q4;333AaD1Iw33333rTctjddd\}}t|dt|tdddddS)NrrJT)retstepg9/?yI@r4)r?linspacerrr )rErssts rGtest_linspace_equivalencez"TestGrid.test_linspace_equivalencesN Ar40002B'''!!U1R8_b99999rTcdtddddddf}tddddddf}t|jdkt|jd kt|d d ddft jd d  t|dddd fdt jd d zt |d dddft jd d d t |ddddfdt jd d zd t |d dddf|d d ddfz dt jdd zd t |ddddf|dddd fz dt jdd zd dS)Nr$rrrr'g?)rrJrJ)rrrrrJdrr)r rrrr?onesr)rEcrs rGtest_ndzTestGrid.test_nds "Qs(BqH$ % "Qs(BqH$ %;&''';&'''1Q4111:C(8(8'89991Q41:r"'"c*:*:':;;;!!A$r111u+rwr3/?/?DDD!!A$qqq"u+qS1A1A/A2FFF!!Aq!!!G*qAqqqz"9"%bgb#&6&6"6 < < <!!Aqqq!G*qAAAqz"9"%bgb#&6&6"6 < < < < 22  DAq A      rTzstart, stop, step, expected)NrJr)rJ)irN)ict||||||f}t|||}t|j|dt|j|ddS)Nrr)r rsize)rEstartstopstepexpectedgrid grid_smalls rGtest_mgrid_size_none_handlingz&TestGrid.test_mgrid_size_none_handlingscU4_eDo565d?+ TY ,,,Z_hqk22222rTc*tdddf}ttjdtjdtjdf}t|jtjkt ||tddd}ttjdtjdtjd}t|jtjkt ||dS)Nr'Q?)r r?float32rrMfloat64r)rEgrid64grid32s rGtest_accepts_npfloatingz TestGrid.test_accepts_npfloatingss4|}&rz#rz$'7'7 3GHJ  *+++!&&111s4|$rz#rz$'7'7 3GH  *+++!&&11111rTcBtdddf}ttjdtjdtjdf}t|jtjkt ||tdtjdd}tdtjddf}t|j|jcxkotjknct ||dtddd}ttjdtjdtjd}t|jtjkt ||dS)Nr'rrry333333 @)r r? longdoublerrMrr)rErgrid128 grid128c_a grid128c_bs rGtest_accepts_longdoublez TestGrid.test_accepts_longdoublesds4|}& M#  r}T222=3E3E E F   .///!&'2221R]1--d23 1R]1--d235   J$4EEEE EEEEFFF:z!}555s4|$ M#  r}T222=3E3E E   .///!&'22222rTc ttdddftddtjdfttdddtddtjdtddd}tdddfd}t |j|jcxkotjknct||tddtjd}tddtjdfd}t |j|jcxkotj knct||dS)Nr'g333333?y@yffffff @r) rr r? complex64rrMrr clongdoubler)rEgrid64_agrid64_b grid128_a grid128_bs rGtest_accepts_npcomplexfloatingz'TestGrid.test_accepts_npcomplexfloatingsf! #c"*+ s3r|B/?/?'?'@!B   " #c"* uSR\"-=-=%=>   S&S(+(.>>>>BJ>>>>???8X...#c"."6"667 #c"."6"6679!<  9?CCCCbmCCCCDDD8X.....rTN)rrrrHrrrrrrrrrrrKrTrGrrs 4 4 4::: < < < [:"#=    3 3    3 2 2 2333,/////rTrc2eZdZdZdZdZdZdZdZdS)TestConcatenatorcttdtjgdtjd}t|dd|f}t|gddS)N)rrr)r%r+r*r+r) rrrrrrrrrrrr)rrr?rDr)rErrs rGtest_1dzTestConcatenator.test_1d/sd2./:L:L:L1M1MNNN GAJJ q!QzN1BBBCCCCCrTcXtdddf}t|jdkdS)Ng333333$@rrJf8)rrrMrEgs rGtest_mixed_typez TestConcatenator.test_mixed_type5s- tQrTzN4     rTctdtjdgtjgddf}t|jdkdS)Ng333333$r)rr)r%g$@r)rr?rDrrMrs rGtest_more_mixed_typez%TestConcatenator.test_more_mixed_type9sG ubhsmmRXiii%8%8$> ?4     rTctddd}t|jdktddtjd}t|jdkdS)NrrYyY@)d)rrrr?rrs rGtest_complex_stepz"TestConcatenator.test_complex_step=s^ qDyM6!""" qBL&&& '6!"""""rTctjdd}tjdd}td||f}t |jdkt |ddddf|t |ddddf|t||f}t |jdkt |ddddf|t |ddddf|dS)Nr+1)r+rJ)rJr+)r?randomrandrrrr)rErrrs rGtest_2dzTestConcatenator.test_2dFs INN1a  INN1a  sAqyM7"###1QQQU8Q'''1QQQU8Q''' q!tH7"###1RaRU8Q'''1QRRU8Q'''''rTc6ttdtjddfgdttgdtjdfgdtttjdgdfgddS)Nrrrrrrr))rrrr)ro)rrr?rDrls rGrwzTestConcatenator.test_0dRsR28A;;)*III666R 28A;;./>>>R YYY./>>>>>rTN) rrrrrrrrrwrKrTrGrr.sqDDD !!!!!!### ( ( (?????rTrceZdZdZdS)TestNdenumeratectjddgddgg}ttt |gddS)Nrrr)r%)))rrr))rrr)rr)))rrr%)r?rDrlistr rErs rGrHzTestNdenumerate.test_basicYsX Hq!fq!f% & &T+a..))III K K K K KrTN)rrrrHrKrTrGrrXs(KKKKKrTrceZdZdZdZdS)TestIndexExpressionctjd}t|dd|tddt|dd|tdddS)Nrr$)r?arangerrrrs rGtest_regression_1z%TestIndexExpression.test_regression_1`s^ IaLLQssVQr#2#wZ(((QssVQy"~./////rTc ,tjddd}t|ddddddgf|tddddddgft|ddddddgf|t ddddddgfdS)Nr%r+r*r)rr)r?rrrrrrs rG test_simple_1z!TestIndexExpression.test_simple_1fs INN1a # #Qqqq"1"q!f}%q111bqb1a&=)A'BCCCQqqq"1"q!f}%qAAArrAq6M):';<<<<z0TestIx_.test_shape_and_dtype..~s!7772dd2hh777rTc36K|]\}}|k |dkVdS)rNrK)rjshks rGrz/TestIx_.test_shape_and_dtype..s/NN2qAvvB!GvvvvNNrT) rr?rr enumeraterrrrr issubdtyperMinteger)rEsizesarraysrrrrs @@rGtest_shape_and_dtypezTestIx_.test_shape_and_dtypezsBI& < rJrr<)r4r)rrr)r%)r+rr,r=)r>rJrr<)r4r%r&rrr))rrrrr)rr?rDrr{rP)dird3s rGtest_diag_indicesr*s aB ,,,,,!//"""$ % %A AbE 28^^^#^^%%%&&&( ) )    a  B C  A AbE 28q!f!f!f!f       rTc eZdZdZdZdZdS)TestDiagIndicesFromctjd}t|\}}t|tjdt|tjddS)Nr:r%)r?rrrr)rErrrrs rGtest_diag_indices_fromz*TestDiagIndicesFrom.test_diag_indices_froms^ I  V $ $ ##11bill+++1bill+++++rTctjd}ttd5t |ddddS#1swxYwYdS)Nr,r)r?rr rCrrvs rGtest_error_small_inputz*TestDiagIndicesFrom.test_error_small_inputs GAJJ ^ < < ! ! a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sAA A ctjdt}ttd5t |ddddS#1swxYwYdS)N)r)r)rr)r")r?r{rPr rCrrvs rGtest_error_shape_mismatchz-TestDiagIndicesFrom.test_error_shape_mismatch s H\3 ' ' ^ < < ! ! a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !sA  AAN)rrrr/r1r3rKrTrGr,r,sA,,, !!! !!!!!rTr,cttddd}dttjdD}t ||ttd}t ||ttd}t |ttdtt}t |dgttd}t |dgttdg}t |gdS) Nrrr)cg|]\}}|SrKrK)rixes rGrz test_ndindex..sAAAur1AAArTro)r)rKr)rrr r?r{rr)rrrs rG test_ndindexr8s WQ1  AAA BHY,?,? @ @AAAHq(### WY    Aq(### WT]]Aq$wqzz**+++ WYYARD WR[[ARD Wqc]ABrT)"rnumpyr? numpy.testingrrrrrrr numpy.lib.index_tricksr r r r rrrrrrrrrrrrrr r r*r,r8rKrTrGr<s  k-k-k-k-k-k-k-k-Zp/p/p/p/p/p/p/p/f'?'?'?'?'?'?'?'?TKKKKKKKK = = = = = = = ='3'3'3'3'3'3'3'3T000 D D D D D D D D N   8!!!!!!!!&rT