\ K]c@sddlmZddlZddlmZddlmZddlm Z ddl m Z ej dZ de ejfd YZed krejndS( i(tprint_functionN(tcuda(tunittest_support(tENABLE_CUDASIM(t SerialMixincCs||S(N((tatb((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyt tt TestReductioncBsPeZdZdZdZdZdZdZdZdZ RS(cCsHtj|dtjd}|j}t|}|j||dS(Ntdtypei(tnptarangetfloat64tsumt sum_reducet assertEqual(tselftntAtexpecttgot((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyt _sum_reduce s  cCsjtrddg}n3ddddddddd d d d d ddg}x|D]}|j|qOWdS(Niiiiiiiiiiiiiii(RR(Rt test_sizesR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_sum_reduces  cCsHtjddtjd}|j}t|}|j||dS(NiR i(R R R RRR(RRRR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_empty_array_host!s  cCsWtjddtjd}tj|}|j}t|}|j||dS(NiR i(R R R Rt to_deviceRRR(RRtdARR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_empty_array_device's   cCsitjd}tjddtjd}|j}||dd}|jtj||dS(NcSs||S(N((RR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyR/Ri@R itinit(RtreduceR R R tprodt assertTruetallclose(Rt prod_reduceRRR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_prod_reduce.s  cCs`tjd}tjddtjd}|j}||dd}|j||dS(NcSs t||S(N(tmax(RR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyR6RiR iRi(RtReduceR R R R$R(Rt max_reduceRRR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_max_reduce5s  cCsXd}tjddtjd}|j|}t|d|}|j||dS(Nii R iR(R R R RRR(RRRRR((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_non_identity_init<s cCstjddtjd}tjtjddtj}|j}t|d|}|j||j ||ddS(Ni R itresi( R R R RRtzerosRRt assertIsNoneR(RRRRR)((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyttest_result_on_deviceCs !  ( t__name__t __module__RRRRR#R'R(R,(((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyR s       t__main__(t __future__RtnumpyR tnumbaRRtunittestt numba.configRtnumba.cuda.testingRR%RtTestCaseR R-tmain(((sElib/python2.7/site-packages/numba/cuda/tests/cudapy/test_reduction.pyts @