U
    /e
                     @   sZ   d dl Z e jjZd dlmZ d dlmZ e 	dZ
dd Ze jdddd	gd
d ZdS )    N)	assert_eqcupyc                  C   sz  dd } t jjtjjd}| |jddddd | |jd	d
ddd | |jdddd | |jdddd | |jddddd | |j	ddddd | |j
dddd | |jdddd | |jdddddd | |jddd | |jddd | |jddd | |jd
ddd | |jdd
ddd | |jddddd | |jdddddd | |jddddd | |jdddd | |jddd | |jdddd | |jddd | |jddd | |jdddddd | |jddd | |jddddd | |jddddd | |jdddd | |jdddd | |j ddd | |j!ddd | |j"dddd | |j#ddd | |j$dddd d S )Nc                 _   s,   | ||}t |jtjkstt|| d S )N)type_metar   ndarrayAssertionErrorr   )funcargskwargsa r   E/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_cupy_random.pyrnd_test   s    
z!test_random_all.<locals>.rnd_testRandomState            sizechunks
   g      ?)%darandomr   r   betaZbinomialZ	chisquareZexponentialfgammaZ	geometricZgumbelZhypergeometricZlaplaceZlogisticZ	lognormalZ	logseriesZnegative_binomialZnoncentral_chisquareZnoncentral_fnormalZparetopoissonpowerZrayleighZrandom_sample
triangularuniformZvonmisesZwaldZweibullZzipfZstandard_cauchyZstandard_exponentialZstandard_gammaZstandard_normalZ
standard_t)r   rsr   r   r   test_random_all   sF    r$   shape)r   r   )r   r      )r   r   r&   r   c                 C   sh   t jjtjjd}|j| dd}t|jtjks4tt	|| |jj
dt|  ksVt|j
| ksdtd S )Nr   r   r   )r   )r   r   r   r   r   r   r   r   r   r   r%   len)r%   r#   xr   r   r   test_random_shapes=   s    
r)   )ZpytestmarkZgpuZ
pytestmarkZ
dask.arrayarrayr   Zdask.array.utilsr   Zimportorskipr   r$   Zparametrizer)   r   r   r   r   <module>   s   
2