ó 2ÄÈ[c@`sjddlmZmZmZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZdefd„ƒYZdefd„ƒYZd efd „ƒYZd efd „ƒYZd efd„ƒYZdefd„ƒYZ d„Z!defd„ƒYZ"d„Z#d„Z$d„Z%dS(i(tdivisiontabsolute_importtprint_functionN(tassert_t assert_equaltassert_array_equaltassert_almost_equaltassert_array_almost_equalt assert_raisestassert_raises_regex( tmgridtogridt ndenumeratet fill_diagonalt diag_indicestdiag_indices_fromt index_exptndindextr_ts_tix_tTestRavelUnravelIndexcB`s>eZd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`sçttjdd"ƒd#ƒttjd$d%ƒdƒttjdd&ƒd'ƒttjd(d)ƒdƒtttjdd*ƒtttjd d+ƒtttjd d,ƒtttjd-d.ƒtttjd/d0ƒtttjd1d2ƒtttjd3d4ƒtttjd5d6ƒttjd:d;ƒddd gƒttjddd gd<ƒd@ƒtjdddgd ddggƒ}ttj|dAƒdddgƒttj|dBddƒdddgƒttj|dCddƒdddgƒttj|dDddEƒdddgƒttjdFdGƒd!ƒttjtjdddgƒdHƒdddgd ddggƒttjtjdddgƒdIddƒdddgd ddggƒttjd!dJƒddd dgƒdS(KNiiiiþii^iBiÿÿÿÿgà?iiýÿÿÿgš™™™™™¹?giiiiii)i%tordertFii tmodetclipiitwrapi ii iU(ii(ii(ii(ii(ii^(iiB(iiB(ii^(ii(ii(ii(iýÿÿÿi(ii(ii(ii(iiýÿÿÿ(ii(ii(ii(gš™™™™™¹?g(iiiii*i.(iii(iiiiii*i.(ii(ii(ii(ii(RR(iiii(iiii (ii(ii(iiii (Rtnpt unravel_indextravel_multi_indexRt ValueErrort TypeErrortarray(tselftarr((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_basics<"'"%%$'c C`s8tjtjkryddgddgddgddgdd gddgd dgf}ttj|dƒddgƒnd gd gf}tjtjƒjd}ttj||dfƒd gƒtttj||ddfƒttj||dfddƒd gƒtttj||ddfddƒdS(NiiiiiuiiiK iii)iixi$i’ iiI¼úpOI 7MRR(i)iixi$i’ ii( Rtintptint64RRtiinfotmaxRR(R!R"t dummy_arrthalf_max((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_big_indices2s $!%c C`s(x!tjtjtjtjtjtjgD]õ}tjddddddgddddddggd|ƒ}d}d |d|d}ttj ||ƒ|ƒt|tj ||ƒƒ|dd|d}ttj ||d d ƒ|ƒt|tj ||d d ƒƒtjddddddgddddddgddddd dggd|ƒ}d}d d |d|d|d}ttj ||ƒ|ƒt|tj ||ƒƒ|dd|dd |d}ttj ||d d ƒ|ƒt|tj ||d d ƒƒq+WdS(NiiiiiitdtypeiiRRi i (ii(iii ( Rtint16tuint16tint32tuint32R%tuint64R RRR(R!R+tcoordstshapetuncoords((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_dtypesGs.9E ""c C`s¬ttjddddgdd d ƒtjddd dgdƒƒttjddddgdd dƒtjddddgdƒƒtttjddddgdƒdS(Niiiÿÿÿÿiiiii RRitraiseRi(iiii (iiii (iiii (RR5RR5(iiii (iiii (RRRRR(R!((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_clipmodesbs!cC`sEtjdddgdƒ\}}t|jjƒt|jjƒdS(Niiiii(ii(RRRtflagst writeable(R!txty((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_writeabilityns!cC`s[tjddƒ}t|dƒttdtjdgdƒttdtjdgdƒdS( Nis0d arrays out of boundsi(((((RRRR R(R!R9((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_0dus  (t__name__t __module__R#R*R4R6R;R<(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyRs  #   tTestGridcB`s,eZd„Zd„Zd„Zd„ZRS(cC`sÝtddd…}tddd…}t|jd kƒt|jd kƒt|ddkƒt|ddƒt|ddkƒt|d|dddƒt|d|dddƒt|d|dddƒdS(Niÿÿÿÿiy$@gš™™™™™¹?i iii ig@g"@(i (iggfffffþ?gÇqÇqÌ?(R RR2R(R!tatb((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR#€scC`sLtjddddƒ\}}t|d ƒt|tddd…dƒdS( Nii tretstepiig€H@yI@i g9/§àåÄ?(RtlinspaceRRR (R!R:tst((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_linspace_equivalenceŒs cC`sÅtddd…ddd…f}tddd…ddd…f}t|jd kƒt|jdkƒt|d d dd…ftjdd ƒ ƒt|ddd…d fdtjdd ƒƒt|d ddd…ftjdd ƒd ƒt|ddd…dfdtjdd ƒd ƒt|d ddd…f|d d dd…fdtjd d ƒd ƒt|ddd…df|ddd…d fdtjd d ƒd ƒdS(Niÿÿÿÿiy$@iþÿÿÿigš™™™™™¹?gš™™™™™É?i iitdi (ii i (iii(R RR2RRtonesR(R!tcRF((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_nd‘s"".10400cC`s„tddd…ddd…f}tddd…ddd…f}tj|Œ}x*t||ƒD]\}}t||ƒqcWdS(Niÿÿÿÿiy$@iþÿÿÿi(R R Rtbroadcast_arraystzipR(R!t grid_fullt grid_sparsetgrid_broadcasttfRA((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_sparseŸs ""(R=R>R#RERIRP(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR?s  tTestConcatenatorcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`s†ttdtjddddddgƒƒtjdƒ}t|dd|f}t|ddddddddddddg ƒdS( Niiiiiii(iiiiii(RRRR RG(R!RARH((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_1dªs,cC`s-tddd…f}t|jdkƒdS(Ng333333$@ii tf8(RRR+(R!tg((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_mixed_type°scC`sKtdtjdgƒtjdddgƒdf}t|jdkƒdS(Ng333333$Àiiiig$@RS(RRR RR+(R!RT((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_more_mixed_type´s4cC`stjjddƒ}tjjddƒ}td||f}t|jdkƒt|dd…dd…f|ƒt|dd…dd…f|ƒt||f}t|jdkƒt|dd…dd…f|ƒt|dd…dd…f|ƒdS(Nit1i (ii (i i(RtrandomtrandRRR2R(R!RARHRF((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_2d¸s###cC`sšttdtjdƒdfdddgƒttdddgtjdƒfddddgƒtttjdƒdddgfddddgƒdS(Niiii(RRRR (R!((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR<Äs,5(R=R>RRRURVRZR<(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyRQ©s     tTestNdenumeratecB`seZd„ZRS(cC`sJtjddgddggƒ}ttt|ƒƒdd d d gƒdS(Niiiii(ii((iii(ii((iii(ii((iii(ii((iii(RR RtlistR (R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR#Ës!(R=R>R#(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR[ÊstTestIndexExpressioncB`seZd„Zd„ZRS(cC`sEtjdƒ}t|d |td ƒt|d |td ƒdS(Niiÿÿÿÿ(RtarangeRRR(R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_regression_1ÒscC`sºtjjdddƒ}t|dd…dd…ddgf|tdd…dd…ddgfƒt|dd…dd…ddgf|tdd…dd…ddgfƒdS(Niiiiii(RRXRYRRR(R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_simple_1ØsO(R=R>R_R`(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR]Ñs tTestIx_cB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`sIxBttjfD]1}tj|dƒƒ\}t|jtjƒqWdS(Ni(trangeRR^RRR+R$(R!tfuncR@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR_àsc`sÑd}xÄttjfD]³}tjg|D]}||ƒ^q)Œ}x‚tt||ƒƒD]k\‰\}}t|jˆ|ƒtt ‡fd†t|jƒDƒƒƒttj |j tj ƒƒqZWqWdS(Niiiic3`s-|]#\}}|ˆkr|dkVqdS(iN((t.0tjtsh(tk(s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pys îs(iiii( RbRR^Rt enumerateRKRR2Rtallt issubdtypeR+tinteger(R!tsizesRctsztarraysR@((Rgs@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_shape_and_dtypeçs(()cC`sBttttg}tj|ƒ\}ttj|ƒd|ƒdS(Ni(tTruetFalseRtnonzeroRR(R!tbool_atint_a((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_boolñscC`s5dddgdddgg}tttj|ƒdS(Niiiiii(RRRR(R!tidx2d((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyt test_1d_onlyöscC`sod}tj|ƒ}t||ƒ}t|dj|dfƒt|djd|fƒt|j|fƒdS(Niii(RR^RRR2(R!tlength_of_vectorR9tout((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_repeated_inputús (R=R>R_RoRuRwRz(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyRaßs    c C`sotjtjdddggƒddtjdddggƒf}t|ddddddddggƒdS(Niiiiiii(Rtc_R R(R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_c_sCtTestFillDiagonalcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`s]tjdtƒ}t|dƒt|tjdddgdddgdddggƒƒdS(Niii(ii(RtzerostintR RR (R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR# s   cC`s±tjdtƒ}t|dƒt|tjdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg ƒƒdS(Ni iii(i i(RR~RR RR (R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_tall_matrixs         cC`s´tjdtƒ}t|dtƒt|tjdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgg ƒƒdS(Ni iii(i i(RR~RR RpRR (R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_tall_matrix_wrap#s        cC`sœtjdtƒ}t|dƒt|tjddddddddddg ddddddddddg ddddddddddg gƒƒdS(Nii ii(ii (RR~RR RR (R!R@((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_wide_matrix3s  *!cC`sctjdtƒ}t|dƒtjdddgƒ}ttj|dkƒ||||fƒdS(Niiiii(iiii(RR~RR R Rtwhere(R!R@ti((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyttest_operate_4d_array<s (R=R>R#R€RR‚R…(((s@/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_index_tricks.pyR} s    c C`s"tdƒ}tjddddgddddgd d d d gd dddggƒ}d||s 4Lq*!% :