idhDddlZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZmZmZmZmZejdkZdZGddZGdd ZGd d ZGd d Z GddZ!GddZ"GddZ#GddZ$GddZ%GddZ&GddZ'GddZ(GddZ)Gd d!Z*Gd"d#Z+Gd$d%Z,d&Z-dS)'N)apply_along_axisapply_over_axes array_splitsplithsplitdsplitvsplitdstack column_stackkrontile expand_dimstake_along_axisput_along_axis)assert_ assert_equalassert_array_equal assert_raises assert_warnscFtjfd}|S)z: hack in keepdims behavior into a function taking an axis cN|fd|i|}|d}tj||S)Naxisrr)npr)arkwargsresfuncs T/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/lib/tests/test_shape_base.pywrappedz_add_keepdims..wrappeds>d1**4*6** <D~c----) functoolswraps)rr!s` r _add_keepdimsr%s8_T..... Nr"c&eZdZdZdZdZdZdS)TestTakeAlongAxisc ddlm}|ddd}tjtjt ft tjt tjt ft tj t tj t ftj tj t dfg}|D]d\}}}tt|jdgzD]8}||fd |i|}||fd |i|} t!|t#|| | 9edS) z- Test it translates from arg to rrand)kthNrr) numpy.randomr*rsortargsortdictr%minargminmaxargmax partition argpartitionlistrangendimrr) selfr*rfuncsrargfuncrra_funcai_funcs r test_argequivalentz$TestTakeAlongAxis.test_argequivalentsH%%%%%% DAqMMWbj$&& ) 26 " "M")$<$.double|s 7Nr"ceZdZdS).MyNDArrayNr`rarbrcr"r MyNDArrayr Dr"rrrGr.r+r,)rndarrayrKviewrr isinstancer)r=rrmexpectedresults r test_preserve_subclassz)TestApplyAlongAxis.test_preserve_subclass{s              Hq!fq!f% & & + +I 6 68aVaV,--229==!&!Q// 69--...68,,,!&!Q// 69--...68,,,,,r"cGddtj}d}tjd|}t t |d|tjgddS)NceZdZdZdS)9TestApplyAlongAxis.test_subclass..MinimalSubclassrGN)r`rarbdatarcr"r MinimalSubclassrsDDDr"rc|jSN)r)rKs r minimal_functionz:TestApplyAlongAxis.test_subclass..minimal_functions : r"rr+r)rGrGrG)rrzerosrrrrK)r=rrrs r test_subclassz TestApplyAlongAxis.test_subclasss     bj       HV   ! !/ 2 2 -q! 4 4bhyyy6I6I     r"c2tjd|}ttjd|}t t ||t|tjgd|dS)Nrrrrr) rrHrrrrrrrK)r=clsrrs r test_scalar_arrayz$TestApplyAlongAxis.test_scalar_arraysz GFOO  % %rvq!,, 3$$%%%3 3 3 8 8 = =>>>>>r"cd}tjd|}t|d|}t t ||t |tjgd|t|d|}t t ||t |tjgd|dS)Nc|t|jdtjtj|dS)z/ Sum x, returning a 0d array of the same class rGT)keepdims)rr<rsqueezerxs r sum_to_0dz3TestApplyAlongAxis.test_0d_array..sum_to_0ds3  # # #:bfQ66677 7r"rrrrG)r+r+r+r+r+r+)rrHrrrrrrK)r=rrrrs r test_0d_arrayz TestApplyAlongAxis.test_0d_arrays 8 8 8 GFOO  % %y!Q// 3$$%%%3 3 3 8 8 = =>>>y!Q// 3$$%%%3););); < < A A# F FGGGGGr"cJfdtjddtd}tjfdt jdDd}tt|t|t||td}tjfd t jdDd}tt|t|t||tjd d td}tjfd t jd Dd}tt|t|t||dS)Nct|jd|ddd|dddfzS)/produces an asymmetric non-square matrix from xrGN)rr<r)rrs r f1to2z5TestApplyAlongAxis.test_axis_insertion..f1to2sD  # # #dddGa4j(..s33 3r"rrc:g|]}dd|fSrrc.0ia2drs r z:TestApplyAlongAxis.test_axis_insertion..s:   !EE#aaac(OO   r"rGrrc:g|]}|ddfSrrcrs r rz:TestApplyAlongAxis.test_axis_insertion..s:   !EE#ac(OO   r"Z)rr-r+c g|]>tjfdtjdDd?S)c<g|]}|ddfSrrc)rra3drjs r rzETestApplyAlongAxis.test_axis_insertion...s>&'c!AAAa%j!!r"rr)rstackr;rV)rrrrs @r rz:TestApplyAlongAxis.test_axis_insertion..s~    H+01+>+>      r"r.) rrrrsrrr;rVrrtype)r=rrXrrrrs ` @@@r test_axis_insertionz&TestApplyAlongAxis.test_axis_insertionsH 4 4 4 4 4 inn$$V,,"%C008     %*39Q<%8%8   DII  T&\\4>>222VX&&&"%C008     %*39Q<%8%8   499  T&\\4>>222VX&&&i&&y11!%C008     39Q<((      DII  T&\\4>>222VX&&&&&r"cGddtj}||||||dS)NceZdZdS)FTestApplyAlongAxis.test_subclass_preservation..MinimalSubclassNrrcr"r rrrr"r)rrrrr)r=rs r test_subclass_preservationz-TestApplyAlongAxis.test_subclass_preservationsp     bj    /// ?+++   11111r"cZd}tjdd}t|d|}t t |tjjt|j dt|dddddfj ||dddfj t|dddddfj ||dddfj t|dddddfj ||dddfj dS)Nct|jd|ddd|dddfz}tj|dzdk|S)rrGNrr-r)rr<rma masked_where)rrs r rz8TestApplyAlongAxis.test_axis_insertion_ma..f1to2sT  # # #DDbD'AabbgJ&C5%%c!eQh44 4r"rrrr+rGr.) rrrrsrrrr masked_arrayrr<rmask)r=rrrs r test_axis_insertion_maz)TestApplyAlongAxis.test_axis_insertion_mas% 5 5 5 IcNN " "6 * *ua++ 3 233444SXq!!!3qqq1u:?EE!AAAaC&MM,>???3qqq1u:?EE!AAAaC&MM,>???3qqq1u:?EE!AAAaC&MM,>?????r"c d}tj|dtjddgddgg}t|tjddgddggdS)Nc"|d|dfS)NrGrrcrs r sample_1dz7TestApplyAlongAxis.test_tuple_func1d..sample_1dsQ41: r"rGr.r+r,)rrrKr)r=rrs r test_tuple_func1dz$TestApplyAlongAxis.test_tuple_func1dsi   !)Q1a&1a&9I0J0JKK31a&1a&)9 : :;;;;;r"cd}tjd}tttj|d|tttj|d|d}tjd}tj|d|}t |tjdtttj|d|dS)Nc$tddS)NF)rrs r never_callz1TestApplyAlongAxis.test_empty..never_calls ENNNNNr")rrrrGcFtt|dkdS)NrrG)rrzrs r empty_to_1z1TestApplyAlongAxis.test_empty..empty_to_1s CFFaK 1r")rDrrD)remptyrrJrrrH)r=rrrrXs r rYzTestApplyAlongAxis.test_emptys    HV  j""5z1aHHHj""5z1aHHH    HW  $ZA66VRWR[[)))j""5z1aHHHHHr"ctjddhddhddhgddhd d hd d hgg}tjd d|}tjhdhdhdg}t||tj|jD]8}tt ||t ||9dS)NrG r.r+rr,,r-7rBctj|Sr)setunion)rs r z>TestApplyAlongAxis.test_with_iterable_object..s sy!}r"r>rGr,rr>r.r-rr>r+rrr)rrKrrndindexrVr)r=rxrXrrs r test_with_iterable_objectz,TestApplyAlongAxis.test_with_iterable_objects HWq"g2w 'Wq"g2w '   $%<%>>>>>r"ctjdd}tj||dzdk}tj|d}t t|tjjt|j dt|j j ddS) NrD)r.r-r+rrrGr)r.rGr-) rrrrsrrKrrr MaskedArrayrrVr)r=rexpandeds r test_subclasseszTestExpandDims.test_subclasses6s IbMM ! !& ) ) EKK!qK ) )>!!,,, 8RU%677888X^Y///X]()44444r"N)r`rarbrrrrrrcr"r rrsb...OOO<<<???55555r"rceZdZdZdZdZdZdZej e ddZ d Z d Zd Zd S) TestArraySplitcftjd}ttt|ddS)NrDr)rrrrrJrr{s r test_integer_0_splitz#TestArraySplit.test_integer_0_splitAs) IbMMj+q!44444r"c R tjd}t|d}tjdg}t||t|d}tjdtjddg}t||t|d}tjdtjddtjddg}t||t|d}tjdtjddtjdd tjd dg}t||t|d}tjdtjddtjddtjdd tjd dg}t||t|d}tjdtjddtjddtjdd tjd d tjd dg}t||t|d}tjdtjddtjddtjddtjdd tjd d tjd dg}t||t|d }tjdtjddtjddtjddtjddtjdd tjd d tjd dg}t||t|d }tjdtjddtjddtjddtjddtjddtjdd tjd d tjd dg }t||t|d}tjdtjddtjddtjddtjddtjddtjddtjdd tjd d tjd dg }t||t|d }tjdtjddtjddtjddtjddtjddtjddtjdd tjd d tjd dtjgg }t||dS) NrDrGr.r-r+r,r r)rrrrcompare_resultsrKr=rrdesireds r test_integer_splitz!TestArraySplit.test_integer_splitEs IbMM!Q9R==/W%%%!Q9Q<<1b!1!12W%%%!Q9Q<<1a")Ar2B2BCW%%%!Q9Q<<1a")Aq//9Q##%W%%%!Q9Q<<1a")Aq//9Q??BIa$4$46W%%%!Q9Q<<1a")Aq//9Q??BIaOORYq"5E5EGW%%%!Q9Q<<1a")Aq//9Q??BIaOORYq!__9Q##% W%%%!Q9Q<<1a")Aq//9Q??BIaOORYq!__9Q??BIa$4$46 W%%%!Q9Q<<1a")Aq//9Q??BIaOORYq!__9Q??BIaOORYq"5E5EG W%%%!R  9Q<<1a")Aq//9Q??BIaOORYq!__9Q??BIaOORYq!__9Q##% W%%%!R  9Q<<1a")Aq//9Q??BIaOORYq!__9Q??BIaOORYq!__9Q##RXb\\3 W%%%%%r"c6tjtjdtjdg}t|dd}tjtjdgtjtjdgtjdg}t ||t |jj|djjut|ddgd}tjdtjtjdgtjtjdgg}t ||t |jj|djjudS)NrDr+rrrrDrrG rrKrrrrrrrFrr=rrtgts r test_integer_split_2D_rowsz)TestArraySplit.test_integer_split_2D_rowss< HbimmRYr]]3 4 4!QQ'''x2(("(BIbMM?*C*C8G$$&S!!! B 22333!aV!,,,x  "(BIbMM?";";x2((*S!!! B 2233333r"c  tjtjdtjdg}t|dd}tjtjdtjdgtjtjddtjddgtjtjddtjddgg}t ||dS)NrDr+rrr,r)rrKrrrrrs r test_integer_split_2D_colsz)TestArraySplit.test_integer_split_2D_colss HbimmRYr]]3 4 4!QR(((8RYq\\29Q<<8998RYq!__bi1oo>??8RYq"--ryB/?/?@AAC W%%%%%r"ctjtjdtjdg}t|d}tjtjdgtjtjdgtjdg}t ||t |jj|djjudS)z2 This will fail if we change default axis rDr+rrNrrs r test_integer_split_2D_defaultz,TestArraySplit.test_integer_split_2D_defaults HbimmRYr]]3 4 4!Qx2(("(BIbMM?*C*C8G$$&S!!! B 2233333r"zNeeds 64bit platform)reasonctjdgd}t|d}tjdgd}|gdz}tt |D](}t ||j||j)dS)Nr)rr.r,)i@r.)r broadcast_torr;rzrrV)r=rrchunkrrs r ,test_integer_split_2D_rows_greater_max_int32z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32s OQC . .!Q\22gks3xx 5 5A Qs1v| 4 4 4 4 5 5r"ctjd}gd}t||d}tjddtjddtjddtjddg}t||dS) NrD)rGr-rrrrrGr-r)rrrrrr=rindicesrrs r test_index_split_simplez&TestArraySplit.test_index_split_simples} IbMM))!W2...9Q??BIaOORYq!__9Q##%W%%%%%r"ctjd}gd}t||d}tjgtjddtjddtjddg}t ||dS)NrD)rr-rrrrr-rrrrrrKrr%s r test_index_split_low_boundz)TestArraySplit.test_index_split_low_bounds{ IbMM))!W2...8B<<1a")Aq//9Q##%W%%%%%r"chtjd}gd}t||d}tjgtjddtjddtjddtjgtjgg}t ||dS)NrD)rr-rrD rrrr-rr)r%s r test_index_split_high_boundz*TestArraySplit.test_index_split_high_bounds IbMM###!W2...8B<<1a")Aq//9Q##RXb\\28B<<AW%%%%%r"N)r`rarbr rrrrpytestmarkskipifIS_64BITr#r'r*r-rcr"r r r @s555;&;&;&z 4 4 4&&&444 [H -CDD55ED5&&&&&&&&&&&r"r ceZdZdZdZdS) TestSplitctjd}t|d}tjdtjddg}t||dS)NrDr.r-)rrrrrrs r test_equal_splitzTestSplit.test_equal_splitsP IbMMAqkk9Q<<1b!1!12W%%%%%r"cftjd}ttt|ddS)NrDr+)rrrrrJrr{s r test_unequal_splitzTestSplit.test_unequal_splits) IbMMj%A.....r"N)r`rarbr5r7rcr"r r3r3s2 &&& /////r"r3c&eZdZdZdZdZdZdS)TestColumnStackc<tttddSNrG)r TypeErrorr r=s r test_non_iterablez!TestColumnStack.test_non_iterablesiq11111r"ctjd}tjd}tjddgddgddgg}tj||f}t||dS)NrGr.r+)r.r+r,rGr.r+r,rrKr rr=rrrrXs r test_1D_arrayszTestColumnStack.test_1D_arrayssu HY   HY  8aVVV%&&!Q((VX&&&&&r"ctjdgdgdgg}tjdgdgdgg}tjddgddgddgg}tj||f}t||dS)NrGr.r+r,rArBs r test_2D_arrayszTestColumnStack.test_2D_arrayss HqcA3_ % % HqcA3_ % %8aVVV%&&!Q((VX&&&&&r"ctjtd5tdt dDddddS#1swxYwYdS)Narrays to stack must bematchc3>K|]}tjdVdSr+Nrrrr_s r z1TestColumnStack.test_generator..s*991")A,,999999r"r.)r.raisesr<r r;r=s r test_generatorzTestColumnStack.test_generators ]9,E F F F ; ; 99a999 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;'AAAN)r`rarbr>rCrErQrcr"r r9r9sP222'''''';;;;;r"r9c2eZdZdZdZdZdZdZdZdS) TestDstackc<tttddSr;)rr<r r=s r r>zTestDstack.test_non_iterablesi+++++r"ctjd}tjd}t||g}tjddggg}t||dSNrGr.rrKr rr=rrrrs r test_0D_arrayzTestDstack.test_0D_arraysX HQKK HQKKaVnn(aVH:&&3(((((r"ctjdg}tjdg}t||g}tjddggg}t||dSrWrXrYs r test_1D_arrayzTestDstack.test_1D_arrays\ HaSMM HaSMMaVnn(aVH:&&3(((((r"ctjdgdgg}tjdgdgg}t||g}tjddggddggg}t||dSrWrXrYs r test_2D_arrayzTestDstack.test_2D_arraysu HqcA3Z  HqcA3Z aVnn(aVH1xj1223(((((r"ctjddg}tjddg}t||g}tjddgddggg}t||dSrWrXrYs r test_2D_array2zTestDstack.test_2D_array2 sl HaV   HaV  aVnn(aVaV,-..3(((((r"ctjtd5tdt dDddddS#1swxYwYdS)NrGrHc3>K|]}tjdVdSrKrLrMs r rOz,TestDstack.test_generator..s*33QBIaLL333333r"r.)r.rPr<r r;r=s r rQzTestDstack.test_generators ]9,E F F F 5 5 33%((333 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5rRN) r`rarbr>rZr\r^r`rQrcr"r rTrTsn,,,))))))))))))55555r"rTc*eZdZdZdZdZdZdZdS) TestHsplit&Only testing for integer splits. c>tttdddSr;)rrJrr=s r r>zTestHsplit.test_non_iterablej&!Q/////r"ctjd} t|dtddS#t$rYdSwxYw)NrGr.r)rrKrrrJr{s r rZzTestHsplit.test_0D_array!sR HQKK  1aLLL AJJJJJ    DD s7 AActjgd}t|d}tjddgtjddgg}t||dSNrGr.r+r,r.rGr+r,rrKrrrs r r\zTestHsplit.test_1D_array)s^ H\\\ " "Qll8QF##RXq!f%5%56W%%%%%r"ctjgdgdg}t|d}tjddgddggtjddgddggg}t||dSrjrlrs r r^zTestHsplit.test_2D_array/s Hlll,,  ! !Qll8aVaV,--rx!Q!Q8H/I/IJW%%%%%r"Nr`rarb__doc__r>rZr\r^rcr"r rdrdsZ000&&& &&&&&r"rdc*eZdZdZdZdZdZdZdS) TestVsplitrec>tttdddSr;)rrJr r=s r r>zTestVsplit.test_non_iterable;rgr"cftjd}ttt|ddSrW)rrKrrJr r{s r rZzTestVsplit.test_0D_array>) HQKKj&!Q/////r"ctjgd} t|dtddS#t$rYdSwxYwNrkr.r)rrKr rrJr{s r r\zTestVsplit.test_1D_arrayBsX H\\\ " "  1aLLL AJJJJJ    DD s9 AActjgdgdg}t|d}tjgdgtjgdgg}t||dSNrkr.)rrKr rrs r r^zTestVsplit.test_2D_arrayJsr Hlll,,  ! !Qll8\\\N++RX|||n-E-EFW%%%%%r"Nrnrcr"r rqrq7sZ000000&&&&&r"rqc,eZdZdZdZdZdZdZdS) TestDsplitc>tttdddSr;)rrJrr=s r r>zTestDsplit.test_non_iterableTrgr"cftjd}ttt|ddSrWrrKrrJrr{s r rZzTestDsplit.test_0D_arrayWrtr"cjtjgd}ttt|ddSrxr}r{s r r\zTestDsplit.test_1D_array[s/ H\\\ " "j&!Q/////r"ctjgdgdg} t|dtddS#t$rYdSwxYwrv)rrKrrrJr{s r r^zTestDsplit.test_2D_array_sh Hlll,,  ! !  1aLLL AJJJJJ    DD s= A  A ctjgdgdggdgdgg}t|d}tjddgddggddgddgggtjddgddggddgddgggg}t||dSrj)rrKrrrs r test_3D_arrayzTestDsplit.test_3D_arrayhs H|||<<!<<<<!" # #Qll8q!fq!f-AA/?@AA8q!fq!f-AA/?@AACW%%%%%r"N)r`rarbr>rZr\r^rrcr"r rzrzRs_000000000&&&&&r"rzceZdZdZdS) TestSqueezecrddlm}|ddddd}|ddddd}|dddd}ttj|tj|dttj|tj|dttj|tj|dd ggg}tj|}t |d t |jdt t|tj dS) Nrr)rhrDrG)rhrDrD)rhrDrhrwg?) r0r*rrrrsrr<rr)r=r*rrcrs r test_basiczTestSqueeze.test_basicts%%%%%% DRQ " " DQAr " " DAr2  2:a=="*Q *E*EFFF2:a=="*Q *E*EFFF2:a=="*Q*A*ABBBeWIjmmS#SXq!!!T#YY +++++r"Nr`rarbrrcr"r rrss#,,,,,r"rceZdZdZdZejdej ej gdZ dZ ejdgddZ d S) TestKronctjd}tjddgddgg}tjddgddgg}ttj|||tjddgddgg}tjd}ttj|||tjdg}tjddgddgg}tjddgddgg}ttj|||tjddgddgg}tjdg}ttj|||tjdggdggg}tjddgddgg}tjddgddggddgddggg}ttj|||tjddgddgg}tjdggdggg}tjddgddggddgddggg}ttj|||dS) NrGr.r+r,rrr,r)rrKrr )r=rrks r rzTestKron.test_basicsR HQKK Hq!fq!f% & & Hq!fq!f% & &271a==!,,, Hq!fq!f% & & HQKK271a==!,,, HaSMM Hq!fq!f% & & Hq!fq"g& ' '271a==!,,, Hq!fq!f% & & HaSMM271a==!,,, HseqcU^ $ $ Hq!fq!f% & & H1v1v&!Q!Q(89 : :271a==!,,, Hq!fq!f% & & HseqcU^ $ $ H1v1v&!Q!Q(89 : :271a==!,,,,,r"cGddtj}tjddg}||j|j|j}t tt||tjt tt|||t tt|||t tt|||dS)NceZdZdZdS)*TestKron.test_return_type..myarrayg?N)r`rarb__array_priority__rcr"r myarrayrs!$   r"rr.) rrrHrVrFrrrr )r=rrrs r test_return_typezTestKron.test_return_types % % % % %bj % % % GQFOO WQWagqv . .T$q!**%%rz222T$r2,,''111T$q"++&&000T$r1++&&00000r" array_classc|tjddg}|tjddg}|tjddg}ttj|||dS)Nr+r)rrHrr )r=rrrrs r test_kron_smokezTestKron.test_kron_smokesu KA ( ( KA ( ( KA ( (271a==!,,,,,r"chtjddgddggddgddgg}tjtjgdtjtjdt }t |tj||dS) NrGr.r+r,rr)rGr,r,rE)rrrKdiagidentityrNrr )r=rrs r test_kron_mazTestKron.test_kron_mas EKK!Q!Q(AA/?K @ @ EKK ..hr{1~~T::::  < < 1bgamm,,,,,r"zshape_a,shape_b))rr)r@)r,r-r)r.r.)r.r.r.))rGrr)r.rr.r))r.rrr.rctj|}tj|}dtdt|t|z z|z}dtdt|t|z z|z}tj||}tj||}tj|j|s JddS)N)rGrzUnexpected shape from kron)rrHr6rzmultiplyr array_equalrV) r=shape_ashape_brrnormalised_shape_anormalised_shape_bexpected_shapers r test_kron_shapezTestKron.test_kron_shapes GG   GG  !C3w<<G +D$E$EEO!C3w<<G +D$E$EEO%79KLL GAqMM~)) G G*F G G G G Gr"N)r`rarbrrr.r/ parametrizerasarraymatrrrrcr"r rrs---: 1 1 1 [ BF+------ [      G G   G G Gr"rc&eZdZdZdZdZdZdS)TestTilectjgd}ddgddgg}tt|dgdtt|dgdgdgtt|dgdgtt|dgd gd gtt|d ddgddgddgddggtt|dgd gd gd gd gdS) NrrGr.r+r,)rrGr.rrGr.rrGr.)rGr.rGr.)r+r,r+r,r.rG)rrKrr r=rrs r rzTestTile.test_basics3 HYYY  VaV T!QZZ!3!3!3444T!V__'9'9'9;M;M;M&NOOOT!V__'9'9'9&:;;;T!QZZ,,, !=>>>T!V__1v1v1v1v&FGGGT!V__|||\\\'3||\\\'C D D D D Dr"ctjd}t|d}|dz }t|tjddS)Nr-rGr.)rrrr rrs r (test_tile_one_repetition_on_array_gh4679z1TestTile.test_tile_one_repetition_on_array_gh4679sC IaLL AJJ QQ ! %%%%%r"ctjggg}tjggg}t|dj}t|dj}t |dt |ddS)Nr.r^)r.r)r+r.r)rrKr rVr)r=rrrrxs r rYzTestTile.test_emptysp HrdV   Hb"X   AJJ  I   $QQ """""r"cddlm}gd}gd}|D]_}|dd|}|D]L}tj||j}t ||}t ||} t|| M`dS)Nr)randint))r.rrr)r.r+r.)r+r.))r+)r.r+)r+r,r+)r+r.r+)r,r+r.r,rrD)size)r0rrrHrFr r r) r=rrepsrVrrrrlargeklarges r test_kroncomparezTestTile.test_kroncompares((((((@@@JJJ , ,A2A&&&A , ,GAqw''Q aUF++++  , , ,r"N)r`rarbrrrYrrcr"r rrsS D D D&&& ### , , , , ,r"rceZdZdZdS)TestMayShareMemoryc tjd}tjd}ttj||ttj||dddttj||dddttj||dddddfttj|ddd| ttj|ddd| ttj|dddddf| ttj|dddddf|dS)N)rkri)rgrirrr.rG)rrHrmay_share_memory)r=rxd2s r rzTestMayShareMemory.test_basicse GH   W[ ! !#Aq))***#Aq2w//000#Aq1v..///#AqTTrT{33444B'$$B$444555B'##A#333444B'!""ddd( R888999#Bqrr44R4xL"5566666r"Nrrcr"r rrs# 7 7 7 7 7r"rct|t|krtdt||D]\}}t||dS)zCompare lists of arrays.z Iterables have different lengthsN)rzrJzipr)rrrys r rr sb 3xx3w<<;<<<C!!!!11a    !!r").numpyrr#sysr.numpy.lib.shape_baserrrrrrr r r r r rrr numpy.testingrrrrrmaxsizer1r%r'rerurrr r3r9rTrdrqrzrrrrrrcr"r rs   ; 1.1.1.1.1.1.1.1.h999999992c=c=c=c=c=c=c=c=LCCCCCCCC%5%5%5%5%5%5%5%5PC&C&C&C&C&C&C&C&L / / / / / / / / ;;;;;;;;:"5"5"5"5"5"5"5"5N&&&&&&&&:&&&&&&&&6&&&&&&&&B,,,,,,,,&LGLGLGLGLGLGLGLG^%,%,%,%,%,%,%,%,P 7 7 7 7 7 7 7 7 !!!!!r"