U
    /e                     @   s   d dl Zd dlZd dlmZ d dlmZ ejdddgdddggdd	d
 Z	ejdgddgdddggddd Z
dd Zdd ZdS )    N)	assert_eq)A)f4      )B)r   r   )C)f8r   )r   )i4r   )r	   )ZS4r   )paramsc                 C   s   t | jS N)npdtypeparamrequest r   F/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_numpy_compat.pyr      s    r   r   r   r	   c                 C   s   | j S r   )r   r   r   r   r   index   s    r   c                  C   sN   dddg} t jd| d}tjd| dd}|dd	g }|dd	g }t|| d S )
N)ar
   )br
   )cr
   )   r   )r   r   )r   chunksr   r   )r   Zonesdar   )r   xZdxresultexpectedr   r   r   
test_basic   s    
r   c                  C   sf   t jtddgddggdd} tt| dks6ttt| dksLtt| d dksbtd S )Nr      r   )r    r   )r   )r    r    )	r   Z
from_arrayr   arrayintminAssertionErrormaxround)imager   r   r   test_min_max_round_funcs!   s     r(   )Znumpyr   ZpytestZ
dask.arrayr!   r   Zdask.array.utilsr   Zfixturer   r   r   r(   r   r   r   r   <module>   s   


