U
    /e                     @   sR   d dl Zd dlZejjZd dlmZ d dl	m
Z
 d dlmZ edZdd ZdS )    N)apply_gufunc)	assert_eqcupyc                  C   s\   dd } t jddd}tj|ddd}tj|d	d
}t| d|d	ddidd}t|| d S )Nc                 S   s
   t | S )N)npdiff)x r   E/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_cupy_gufunc.pymydiff   s    z&test_apply_gufunc_axis.<locals>.mydiff            F)chunksZasarray   )axisz(i)->(i)i   T)r   Zoutput_sizesZallow_rechunk)	r   randomZrandndaZ
from_arrayr   r   r   r   )r
   aZda_mdmr   r   r	   test_apply_gufunc_axis   s         r   )Znumpyr   ZpytestmarkZgpuZ
pytestmarkZ
dask.arrayarrayr   Zdask.array.gufuncr   Zdask.array.utilsr   Zimportorskipr   r   r   r   r   r	   <module>   s   
