ó àÆ÷Xc@`s°ddlmZmZmZddlZddlZddlmZesiddl m Z e dƒ‚nddlm Z m Z ddl Z d d „Zdd „Zd „ZdS(i(tabsolute_importtprint_functiontdivisionN(tgnumpy_available(tSkipTests.gnumpy not installed. Skip test related to it.(tgarray_to_cudandarraytcudandarray_to_garrayiiic C`s tjjjj}|tjjdƒƒ}tj|g||dƒƒ}tj |Œ}t |ƒ}|j |j ks|t ‚||ƒ}t |ƒ}|j |j ks¬t ‚|djƒ}|jƒ} tj|ƒ} || kjƒsït ‚|| kjƒst ‚dS(sf Make sure that the gnumpy conversion is exact from garray to CudaNdarray back to garray. tUiN(ttheanotsandboxtcudat basic_opst gpu_from_hostttensortftensor3tfunctiontgnumpytrandRtshapetAssertionErrorRtasarraytnptarraytall( RtgpuRtiitAtA_cndtB_cndtBtutvtw((sC/tmp/pip-build-X4mzal/theano/theano/misc/tests/test_gnumpy_utils.pyttests    cC`stjjjj}|tjjdƒƒ}tj|g||dƒƒtj j |Œj dƒ}tjjj |ƒ}t |ƒ}|j|jks›t‚t|ƒ}|j|jks¿t‚|j|jks×t‚|j|jksït‚tj|ƒ}||kjƒst‚dS(sk Make sure that the gnumpy conversion is exact from CudaNdarray to garray back to CudaNdarray. Ritfloat32N(RR R R R R RRRtrandomRtastypet CudaNdarrayRRRRt_stridestgpudataRR(RRRRRtA_garRR((sC/tmp/pip-build-X4mzal/theano/theano/misc/tests/test_gnumpy_utils.pyttest2*s  cC`s@tdƒtdƒtdƒtdƒtd ƒtd ƒdS( sd Test with some dimensions being 1. CudaNdarray use 0 for strides for those dimensions. iiiN(iii(iii(iii(iii(iii(iii(R!R)(((sC/tmp/pip-build-X4mzal/theano/theano/misc/tests/test_gnumpy_utils.pyttest_broadcast_dimsDs      (iii(iii(t __future__RRRtnumpyRRttheano.misc.gnumpy_utilsRtnose.plugins.skipRRRRR!R)R*(((sC/tmp/pip-build-X4mzal/theano/theano/misc/tests/test_gnumpy_utils.pyts