U
    /en
                     @   s   d dl Z e d d dlZd dlmZ d dlmZ d dl	m
Z
 dd Zdd Zd	d
 Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZdS )    Nnumpy)	assert_eqonesc                  C   sD   t dddd} t| }|t ddk s0t| jds@td S )N
   r   i4   r
   dtypechunksz
ones_like-)r   nparrayallAssertionErrorname
startswithax r   >/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_wrap.py	test_ones   s    
r   c                  C   s:   t ddgddd} t| }|tj dddk s6td S )Nr   r   r	   r   r   r   r   r   r   r   r   r   r   r   r   test_size_as_list   s    
r   c                  C   s6   t dddd} t| }|tj dddk s2td S Nr   r   r
   r   r   r   r   r   r   r   test_singleton_size   s    
r   c                  C   s6   t dddd} t| }|tj dddk s2td S r   r   r   r   r   r   test_kwargs    s    
r    c                  C   s`   t jddddd} |  dk s&t| j| jddj  krFdksLn t| jds\td S )	N   r"   d      r%   i8r   r   sync)Z	schedulerz
full_like-)dafullZcomputer   r   r   r   r   r   r   r   r   	test_full&   s    &r,   c                	   C   s4   t jtdd tjdddgddd W 5 Q R X d S )Nfill_value must be scalarmatchr!   r#   r$   r&   r'   )pytestraises
ValueErrorr)   r*   r   r   r   r   $test_full_error_nonscalar_fill_value/   s    r3   c               	   C   sj   t d} tjtdd8}t jd| d}|j| jks8tt|t	jddd W 5 Q R X t
|dksftd S )Nr#   znot implemented by Dask arrayr.   r!   )shape
fill_value   )r)   Z
from_arrayr0   ZwarnsFutureWarningr*   r   r   r   r   len)r   recordr   r   r   r   test_full_detects_da_dtype4   s    
r:   c                  C   s*   t jddd d} t| tjddd d d S )Nr!   r#   )r4   r5   r   )r)   r*   r   r   r+   r   r   r   test_full_none_dtype>   s    r;   c               	   C   sD   t jdddd} tjtdd tj| ddgddd	 W 5 Q R X d S )
Nr!   r6   r&   r   r-   r.   r#   r$   r'   )r   r*   r0   r1   r2   r)   Z	full_like)r   r   r   r   )test_full_like_error_nonscalar_fill_valueC   s    r<   c                   C   s   t ddd t ddd d S )N)@B r=   )順 r>   r   )r4   r   r   r   r   r   r   &test_can_make_really_big_array_of_onesI   s    r@   c                   C   s   t tddddjt tddddjks,tt tddddjt tdddjksVtt tjdddd	d
jt tjdddd	d
jkstt tjddddd
jt tjddddjkstd S )Nr   r   r   r   r?   r!   r#   r$   Zf8r'   i2)sortedr   Zdaskr   r)   r*   r   r   r   r   test_wrap_consistent_namesN   s    


rC   )r0   Zimportorskipr   r   Z
dask.arrayr   r)   Zdask.array.utilsr   Zdask.array.wrapr   r   r   r   r    r,   r3   r:   r;   r<   r@   rC   r   r   r   r   <module>   s    
	
