ó 2ÄÈ[c@`seddlmZmZmZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZmZmZmZmZddlmZdefd„ƒYZdefd „ƒYZd efd „ƒYZd efd „ƒYZ defd„ƒYZ!defd„ƒYZ"d„Z#defd„ƒYZ$dS(i(tdivisiontabsolute_importtprint_functionN( tarraytaranget atleast_1dt atleast_2dt atleast_3dtblocktvstackthstacktnewaxist concatenatetstack(tassert_t assert_raisestassert_array_equalt assert_equaltassert_raises_regextassert_almost_equal(tlongt TestAtleast1dcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`s_tdƒ}tdƒ}t|ƒt|ƒg}tdgƒtdgƒg}t||ƒdS(Nii(RRR(tselftatbtrestdesired((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_0D_arrays   cC`sqtddgƒ}tddgƒ}t|ƒt|ƒg}tddgƒtddgƒg}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_1D_arrays $cC`sqtddgddggƒ}tddgddggƒ}t|ƒt|ƒg}||g}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_2D_arrays  cC`s•tddgddggƒ}tddgddggƒ}t||gƒ}t||gƒ}t|ƒt|ƒg}||g}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_3D_array&s cC`s™ttdƒjd kƒttdƒjd kƒtttdƒƒjd kƒttdƒjd kƒttddgddggƒjd kƒdS(sB Test to make sure equivalent Travis O's r1array function iiy@g@iiiN(i(i(i(i(ii(RRtshapeR(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_r1array/s (t__name__t __module__RRRRR (((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRs     t TestAtleast2dcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`setdƒ}tdƒ}t|ƒt|ƒg}tdggƒtdggƒg}t||ƒdS(Nii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR:s   $cC`swtddgƒ}tddgƒ}t|ƒt|ƒg}tddggƒtddggƒg}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRAs *cC`sqtddgddggƒ}tddgddggƒ}t|ƒt|ƒg}||g}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRHs  cC`s•tddgddggƒ}tddgddggƒ}t||gƒ}t||gƒ}t|ƒt|ƒg}||g}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyROs cC`sttdƒjdkƒttddgƒjd kƒttddgddggddgddgggƒjd kƒdS( sB Test to make sure equivalent Travis O's r2array function iiy@iiiN(ii(ii(iii(RRR(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_r2arrayXs(R!R"RRRRR$(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR#9s     t TestAtleast3dcB`s,eZd„Zd„Zd„Zd„ZRS(cC`sktdƒ}tdƒ}t|ƒt|ƒg}tdgggƒtdgggƒg}t||ƒdS(Nii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRas   *cC`sƒtddgƒ}tddgƒ}t|ƒt|ƒg}tdgdgggƒtdgdgggƒg}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRhs 6cC`s£tddgddggƒ}tddgddggƒ}t|ƒt|ƒg}|dd…dd…tf|dd…dd…tfg}t||ƒdS(Niii(RRR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRos >cC`s•tddgddggƒ}tddgddggƒ}t||gƒ}t||gƒ}t|ƒt|ƒg}||g}t||ƒdS(Niii(RRR(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRvs (R!R"RRRR(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR%`s   t TestHstackcB`s5eZd„Zd„Zd„Zd„Zd„ZRS(cC`stttdƒdS(Ni(Rt TypeErrorR (R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_non_iterablescC`stttdƒdS(N((Rt ValueErrorR (R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_empty_input„scC`sMtdƒ}tdƒ}t||gƒ}tddgƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR‡s   cC`sStdgƒ}tdgƒ}t||gƒ}tddgƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRŽs cC`sqtdgdggƒ}tdgdggƒ}t||gƒ}tddgddggƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR•s (R!R"R(R*RRR(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR&€s     t TestVstackcB`s>eZd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`stttdƒdS(Ni(RR'R (R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR(žscC`stttdƒdS(N((RR)R (R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR*¡scC`sStdƒ}tdƒ}t||gƒ}tdgdggƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR¤s   cC`sYtdgƒ}tdgƒ}t||gƒ}tdgdggƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR«s cC`swtdgdggƒ}tdgdggƒ}t||gƒ}tdgdgdgdggƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR²s $cC`setddgƒ}tddgƒ}t||gƒ}tddgddggƒ}t||ƒdS(Nii(RR R(RRRRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_2D_array2¹s (R!R"R(R*RRRR,(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR+s      tTestConcatenatecB`s>eZd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`s¹xŒdddgD]{}tjd|ƒ}tj||fddƒttjtj||fd|ƒttjtj||fd|d ƒqWtttdƒttttjdƒfƒtjd ƒ}tjd ƒ}ttdƒƒ}x·tdƒD]©}tj||fd|dƒtttj||fd|dƒtttj||fd|dƒtj |ddƒ}tj |ddƒ}|j |j dƒƒqøWtttd ƒdS( Niiitaxisiiÿÿÿÿ(i(i(iii(iii(( tnptonesR Rt AxisErrorR)Rtlisttrangetmoveaxistappendtpop(RtndimRRR.ti((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_exceptionsÂs$"+##c C`stjddtjƒjdƒ}ttdƒƒ}dg}tj||fddƒ}t|j |j ƒt|j dƒtj||fddƒ}t|j |j t |ƒƒt|j |j ƒtj|||fddƒ}t dd d d d d ddgƒ}t||ƒtj|j t |ƒƒ}tj||fddƒ}tj||fddd|ƒ}t||kƒt||ƒdS(NitdtypeiitxR.is0.0s1.0s2.0s3.0t0t1t2tout(ii(R/Rtfloat64treshapeR2R3R tNoneRR:R7tsizetlenRRtzerosR(RRRtctrtdR?trout((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_concatenate_axis_NoneÝs$!  !cC`sTtjddƒ}tj|dƒ}t||ƒtj|dƒ}t||ƒdS(Niid(R/RR RBR(RR;RG((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_large_concatenate_axis_Noneòs  c C`s*ttdƒƒ}tt|fƒ|ƒttt|ƒfƒ|ƒttt|ƒfƒ|ƒttdƒƒ}tt||fƒ||ƒttt|ƒ|fƒ||ƒttt|ƒ|fƒ||ƒtt||fdƒ||ƒtt||fdƒ||ƒtdddgdd d ggƒ}tdd d ggƒ}tdddgdd d gdd d ggƒ}tt||fƒ|ƒtt||fdƒ|ƒtt|j|jfd ƒ|jƒtt|j|jfdƒ|jƒttt|j|jfdƒt dƒj dƒ}|ddd…f}|ddd…f}|ddd…f}tt|||fd ƒ|ƒtt|||fdƒ|ƒtt|j|j|jfdƒ|jƒ|j ƒ} t|||fd d| ƒ} t | | kƒt || ƒdS(Niiiiÿÿÿÿi i i i iiiii.iR?ii*(iii(R2R3RR ttupleRtTRR)RRAtcopyRR( Rtr4tr3ta23ta13Rta0ta1ta2R?RI((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_concatenateýs:##  $0%%+ cC`s¶tddgƒ}tddgƒ}ttt||fdtjdƒƒttt||fdtjdƒƒttt||fdtjdƒƒt||fdtjdƒƒdS( NiiiiR?i(ii(ii(RRR)R R/tempty(RRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_bad_out_shape's %%%cC`s÷tjdtjƒ}ttddgƒtddgƒfd|ƒ}t||kƒtjdtjƒ}ttddgƒtdd gƒfd|ƒ}t||kƒtjdtjƒ}tt ttddgƒtdd gƒfd|ƒdS( NiiiiR?gš™™™™™¹?gš™™™™™É?g333333Ó?gš™™™™™Ù?( R/RWtfloat32R RRt complex64tint32RR'(RR?R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_out_dtype0s00 (R!R"R9RJRKRVRXR\(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyR-Ás    * c C`sítttdƒx{dtjdƒtjdƒtjdƒgtjdƒtjdƒtjdƒggD]"}tt|ƒdddgƒqhWtjdddgƒ}tjdddgƒ}tdddgdddggƒ}ttj||fƒ|ƒttj||fddƒ|jƒttjt||gƒƒ|ƒttjt||gƒƒ|ƒgt dƒD]}tj j dƒ^qt}d dd d g}ddddg}x9t ||ƒD](\}} t tj||ƒj| ƒqÆWttjd t|ddƒttjd t|dd ƒgt dƒD]}tj j ddƒ^q7}d ddd d d g}ddddddg}x9t ||ƒD](\}} t tj||ƒj| ƒq˜WttggggƒjdkƒttggggddƒjdkƒttdtgƒttdtdtjdƒgƒttdttjdƒdgƒttdttjdƒdgddƒttdttjdƒtjdƒgddƒttdttjdƒtjdƒgƒdS(NiiiiiiR.i iiÿÿÿÿiþÿÿÿs out of boundsiýÿÿÿsneed at least one arraysmust have the same shape(iii(i i(ii (ii (i i(i ii(ii i(iii (iii (ii i(i ii(ii(ii(ii(RR'R R/R[RRRMR2R3trandomtrandntzipRRRR1RR)RRE( tinput_RRtr1t_tarraystaxestexpected_shapesR.texpected_shape((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_stack?sL'1 $%""+ .  "(    ( t TestBlockcB`s˜eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„ZRS(cC`sitjdƒ}d|}tjddddgddddggƒ}t||gƒ}t||ƒdS(Nii(ii(R/R0RRR(Rta_2dtb_2dRtresult((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_block_simple_row_wisers  cC`sutjdƒ}d|}tjddgddgddgddggƒ}t|g|ggƒ}t||ƒdS(Nii(ii(R/R0RRR(RRiRjtexpectedRk((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_block_simple_column_wisezs   cC`sttjdddgƒ}tjdddgƒ}tjddddddgƒ}t||gƒ}t||ƒdS(Niiii(R/RRR(RRRRmRk((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt"test_block_with_1d_arrays_row_wise„s !cC`s˜tjdddgƒ}tjdddgƒ}tjddddddgddddddggƒ}t||g||ggƒ}t||ƒdS(Niiii(R/RRR(RRRRmRk((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt'test_block_with_1d_arrays_multiple_rowsŒs cC`s€tjdddgƒ}tjdddgƒ}tjdddgdddggƒ}t|g|ggƒ}t||ƒdS(Niiii(R/RRR(Rta_1dtb_1dRmRk((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt%test_block_with_1d_arrays_column_wise”s cC`swtjdƒ}tjddgƒ}t|g|ggƒ}tjddgddgddggƒ}t||ƒdS(Nii(ii(R/R0RRR(RRiRrRkRm((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_block_mixed_1d_and_2ds c C`s…tjdddggƒ}tjdddggƒ}tjddddddggƒ}tjddddddgƒ}tjdƒ}tjdddddgƒ}tjdƒ}tjddddddgddddddgddddddgddddddgddddddgddddddggƒ}t||g|g|g||g|ggƒ} t| |ƒdS( Niiiiiii(ii(R/RRERR( Rtone_2dttwo_2dtthree_2dtfour_1dtfive_0dtsix_1dtzero_2dRmRk((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_block_complicated¦s&$!  c C`s—tjdddgƒ}tjdddgdddgdddggƒ}tjdddgƒ}tjdddgƒ}tjdƒ}tjdddddgƒ}tjdƒ}tjtj|g|g|ggƒ|g||g|ggƒ}tjddddddgddddddgddddddgddddddgddddddgddddddggƒ} t|| ƒdS( Niiiiiii(ii(R/RRERR( RtonettwotthreetfourtfivetsixtzeroRkRm((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_nested¾s,3  c C`s×tjd tƒd}tjd tƒd}tjd tƒd}tjd tƒd}tjd tƒd}tjdtƒd}tjdtƒd}tjdtƒd}tj||g||gg||g||gggƒ} tdddddgdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddgdddddggdddddgdddddgdddddgdddddgdddddgggƒ} t| | ƒdS(Niiiiiiii(iii(iii(iii(iii(iii(iii(iii(iii(R/R0tintRRR( Rta000ta100ta010ta001ta011ta101ta110ta111RkRm((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_3dÜsN   cC`sZtjddgƒ}tjdƒ}tttj||gƒtttj||gƒdS(Nii(R/RteyeRR)R(RRR((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_block_with_mismatched_shapescC`sKttjdƒtjdƒƒttjtjdƒƒtjdƒƒdS(Nii(RR/RRR(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_no_listsscC`s·d}tt|tjddggƒtt|tjdggƒtt|tjdgdgƒtt|tjgdgƒtt|tjdgdggddggdggƒdS(Nsdepths are mismatchediiiii(RR)R/R(Rtmsg((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_invalid_nestings cC`sRttdtjgƒttdtjggƒttdtjdgggƒdS(NRWi(RR)R/R(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_empty_lists)scC`sHttdtjddgddgfƒttdtjddgƒdS(NRLiiii(ii(ii(RR'R/R(R((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyt test_tuple.s(cC`s|d}dtjdƒ}dtjdƒ}tj|||gƒ}tjddddddgggƒ}t||ƒdS( Ngð?iiig@g@(ii(iii(R/R0RRR(RRRRFRkRm((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_different_ndims2s 'cC`s‘d}dtjdƒ}dtjdƒ}tj||g|ggƒ}tjdddgdddgdddgggƒ}t||ƒdS( Ngð?iiig@g@(ii(iii(R/R0RRR(RRRRFRkRm((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyttest_different_ndims_depths<s (R!R"RlRnRoRpRsRtR|R„RŽRR‘R“R”R•R–R—(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyRhqs       7     (%t __future__RRRtwarningstnumpyR/t numpy.coreRRRRRRR R R R R t numpy.testingRRRRRRt numpy.compatRtobjectRR#R%R&R+R-RgRh(((s?/tmp/pip-build-fiC0ax/numpy/numpy/core/tests/test_shape_base.pyts  L.)' $~ 2