ó 2ÄÈ[c@`sWddlmZmZmZddlZddlmZddlm Z m Z m Z m Z ddl mZmZmZmZd„Zd„Zeed„Zd „Zd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!dej"fd„ƒYZ#de#fd„ƒYZ$d„Z%d„Z&d„Z'dS(i(tdivisiontabsolute_importtprint_functionN(trational(t assert_equaltassert_array_equalt assert_raisestassert_(t as_stridedtbroadcast_arrayst_broadcast_shapet broadcast_tocC`sng|D]}tj|ƒ^q}t|Œ}g|D]}|j^q5}|gt|ƒ}t||ƒdS(N(tnptzerosR tshapetlenR(t input_shapestexpected_shapetstinarrayst outarraystat outshapestexpected((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pytassert_shapes_correct s " cC`s6g|D]}tj|ƒ^q}ttt|ŒdS(N(R R Rt ValueErrorR (RRR((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt assert_incompatible_shapes_raises"c C`s¿tj|dtƒ}ttjj|ƒƒ}tj|ƒj|ƒ}|r`|j}|j}n|r|ddd…}|ddd…}n||}t||ƒ\}} t || ƒdS(Ntdtypeiÿÿÿÿ( R R tinttmultiplytreducetarangetreshapetTR R( tshape0tshape1t transposedtflippedtx0tntx1tytb0tb1((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pytassert_same_as_ufuncs   cC`sQtjdƒ}tjdƒ}t||ƒ\}}t||ƒt||ƒdS(Ni (R RR R(txR)tbxtby((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt test_same5s  cC`s«tjdddggƒ}tjdgdgdggƒ}t||ƒ\}}tjdddgdddgdddggƒ}|j}t||ƒt||ƒdS(Niii(R tarrayR R!R(R-R)R.R/tbx0tby0((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt test_one_off=s!3  c C`s„dddddd d d d d g }xY|D]Q}|g}t||ƒ||g}t||ƒ|||g}t||ƒq+WdS(Niii((i(i(ii(ii(ii(ii(ii(ii(ii(R(tdataRRt input_shapes2t input_shapes3((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyttest_same_input_shapesGs"      cC`s ddgdgdd gd gd d gd gddgdgddgdgddgdgddgdgddgdgddgdgd d!gd"gd#d$gd%gd&d'gd(gd)d*gd+gg }x;|D]3\}}t||ƒt|ddd…|ƒqÐWdS(,Niiiiÿÿÿÿ(i(i(i(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(R(R5RR((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt(test_two_compatible_by_ones_input_shapesbs  cC`stddgdgdd gd gd d gd gddgdgddgdgddgdgddgdgddgdgddgdgd d!gd"gd#d$gd%gd&d'gd(gd)d*gd+gd,d-gd.gd/d0gd1gd2d3gd4gd5d6gd7gd8d9gd:gd;d<gd=gd>d?gd@gg}x;|D]3\}}t||ƒt|ddd…|ƒq9WdS(ANiiiiÿÿÿÿ((i(i(i(ii(ii(i(ii(ii(i(ii(ii((ii(ii(ii(i(ii(i(ii(ii(i(ii(ii((ii(ii((ii(ii((i(i(i(ii(ii(i(ii(ii(i(ii(ii((ii(ii(ii(i(ii(i(ii(ii(i(ii(ii((ii(ii((ii(ii(R(R5RR((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt3test_two_compatible_by_prepending_ones_input_shapes{s. cC`scddgdd gd d d gd dgg}x/|D]'}t|ƒt|ddd…ƒq4WdS(Niiiiiÿÿÿÿ(i(i(ii(i(i(i(i(iii(iii(R(R5R((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyt)test_incompatible_shapes_raise_valueerror›s     c#C`s¬ddgdgdd gd gd d gd gddgdgddgdgddgdgddgdgddgdgddgdgd d!gd"gd#d$gd%gd&d'gd(gd)d*gd+gd,d-gd.gd/d0gd1gd2d3gd4gd5d6gd7gd8d9gd:gd;d<gd=gd>d?gd@gdAdBgdCgdDdEgdFgdGdHgdIgdJdKgdLgdMdNgdOgdPdQgdRgdSdTgdUgdVdWgdXgdYdZgd[gd\d]gd^gd_d`gdagdbdcgddgdedfgdggg!}x°|D]¨\}}t|d|dd|d|dfƒt|d|dƒt|d|dtƒdh|krüt|d|dttƒt|d|dttƒqüqüWdS(iNiiis Shapes: %s %s(i(i(i(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii((i(i(i(ii(ii(i(ii(ii(i(ii(ii((ii(ii(ii(i(ii(i(ii(ii(i(ii(ii((ii(ii((ii(ii((i(i(i(ii(ii(i(ii(ii(i(ii(ii((ii(ii(ii(i(ii(i(ii(ii(i(ii(ii((ii(ii((ii(ii((R,tTruetFalse(R5RR((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyttest_same_as_ufuncªsR cC`s&tjdƒdtjdƒgtjdƒdtjdƒgtjdƒdtjdƒgtjdƒd tjdƒgtjdƒd tjdƒgtjdƒd tjd ƒgtjdƒd tjdƒgtjdƒdtjdƒjddƒgtjdƒdtjdddgdddggƒgtjdƒdtjdƒgtjdƒdtjdƒgtjdƒdtjdƒgtjdƒdtjdƒgtjdƒdtjdƒgtjdƒdtjdƒgg}x3|D]+\}}}t||ƒ}t||ƒqóWdS(Niiiiiÿÿÿÿ(i(i(i(i(i(iii(iii(i(ii(ii(i(ii(ii(ii(ii(ii(ii(R R1R tonesRR R R(R5t input_arrayRRtactual((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyttest_broadcast_to_succeedsÞs$*6$c `s¬dd gd d gd d gddgddgddgddgddgddgddgddgg }x<|D]4\}‰tj|ƒ‰tt‡‡fd†ƒqpWdS(Niiiiiiÿÿÿÿc`s tˆˆƒS(N(R ((tarrt target_shape(sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyts(i((i((i((i(i(i(i(i(i(ii(ii(ii(i(i(i(iÿÿÿÿ(ii(iÿÿÿÿi(R R RR(R5t orig_shape((RCRDsA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyttest_broadcast_to_raisesøs          c`s!ttƒdƒttddgƒd ƒtttjd ƒƒd ƒtttjd ƒtjd ƒƒdƒtttjdƒgdŒdƒtttjdƒgdŒdƒtttjdƒgddgŒdƒtjdƒgdtjdƒgd‰tt‡fd†ƒdS(Niiiii idc`s tˆŒS(N(R ((tbad_args(sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyREs((i(ii(ii(ii(ii(ii(ii(ii(ii(ii(i(RR R R?RR(((RHsA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyttest_broadcast_shape s(##**cC`s,tjdgƒ}t|ƒ}tjdgƒ}t|tjdgƒƒtjddddgƒ}t|dddd|jfƒ}tjddgƒ}t||ƒtjddddgƒ}t|ddddd|jfƒ}tjddddgddddgddddggƒ}t||ƒtjddgƒ}tjdd |ƒ}tjdd ƒ|dRBRGRIR]RbRrRcRkRuR€R‚(((sA/tmp/pip-build-fiC0ax/numpy/numpy/lib/tests/test_stride_tricks.pyts0 ""      4    -  %