
    >ie
                        d dl mZ d dlZ ej        d           d dlZd dlmZ d dl	m
Z
 d dlmZ d Zd Zd Zd	 Zd
 Zd Zd Zd Zd Zd Zd ZdS )    )annotationsNnumpy)	assert_eqonesc                     t          ddd          } t          j        |           }|t          j         dd          k                                    sJ | j                            d          sJ d S )N
   r
   i4   r   dtypechunksz
ones_like-)r   nparrayallname
startswithaxs     :lib/python3.11/site-packages/dask/array/tests/test_wrap.py	test_onesr      sn    XT&111A
A4(((--/////6\*******    c                     t          ddgdd          } t          j        |           }|t          j         dd          k                                    sJ d S )Nr
   r   r   r   r	   r   r   r   r   r   r   s     r   test_size_as_listr      sY    b"XT&111A
A....335555555r   c                     t          ddd          } t          j        |           }|t          j         dd          k                                    sJ d S Nr
   r   r   r   r   r   r   s     r   test_singleton_sizer#      U    RtD)))A
A4((((--///////r   c                     t          ddd          } t          j        |           }|t          j         dd          k                                    sJ d S r!   r   r   s     r   test_kwargsr&   "   r$   r   c                 $   t          j        dddd          } |                                 dk                                    sJ | j        |                     d          j        cxk    rdk    sn J | j                            d          sJ d S )	N   r)   d      r,   i8r   r   sync)	schedulerz
full_like-)dafullcomputer   r   r   r   r   s    r   	test_fullr5   (   s    
F$777AIIKK3##%%%%%7aii&i117????4??????6\*******r   c                     t          j        t          d          5  t          j        dddgdd           d d d            d S # 1 swxY w Y   d S )Nfill_value must be scalarmatchr(   r*   r+   r-   r.   )pytestraises
ValueErrorr1   r2    r   r   $test_full_error_nonscalar_fill_valuer>   1   s    	z)D	E	E	E ? ?
c
6>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s   AAAc                 V   t          j        d          } t          j        t          d          5 }t          j        d|           }|j        | j        k    sJ t          |t          j        dd                     d d d            n# 1 swxY w Y   t          |          dk    sJ d S )Nr*   znot implemented by Dask arrayr8   r(   )shape
fill_value   )
r1   
from_arrayr:   warnsFutureWarningr2   r   r   r   len)r   recordr   s      r   test_full_detects_da_dtyperH   6   s    
cA	m+J	K	K	K <vG&Q///w!'!!!!!RW6c:::;;;	< < < < < < < < < < < < < < <
 v;;!s   AB		BBc                 ~    t          j        ddd           } t          | t          j        ddd                      d S )Nr(   r*   )r@   rA   r   )r1   r2   r   r   r4   s    r   test_full_none_dtyperJ   @   s?    
fD999Aav#TBBBCCCCCr   c                     t          j        ddd          } t          j        t          d          5  t          j        | ddgdd	           d d d            d S # 1 swxY w Y   d S )
Nr(   rB   r-   r   r7   r8   r*   r+   r.   )r   r2   r:   r;   r<   r1   	full_like)r   s    r   )test_full_like_error_nonscalar_fill_valuerM   E   s    
&&&A	z)D	E	E	E ? ?
Qc
6>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s   AA"Ac                 J    t          dd           t          dd           d S )N)@B rO   )順 rP   r   )r@   r   r   r=   r   r   &test_can_make_really_big_array_of_onesrR   K   s2    	$45555!*:;;;;;;r   c            	        t          t          ddd          j                  t          t          ddd          j                  k    sJ t          t          ddd          j                  t          t          dd          j                  k    sJ t          t          j        dddd	
          j                  t          t          j        dddd	
          j                  k    sJ t          t          j        dddd
          j                  t          t          j        ddd          j                  k    sJ d S )Nr
   r   r"   r   rQ   r(   r*   r+   f8r.   i2)sortedr   daskr1   r2   r=   r   r   test_wrap_consistent_namesrX   P   sj   $rd333899VRtD))).> >     $rd333899VR"> >     "'&#fDAAAFGG6
F$777<L L     "'&#fDAAAFGG6
F+++0L L      r   )
__future__r   r:   importorskipr   r   
dask.arrayr   r1   dask.array.utilsr   dask.array.wrapr   r   r   r#   r&   r5   r>   rH   rJ   rM   rR   rX   r=   r   r   <module>r^      s'   " " " " " "   G              & & & & & &            + + +6 6 60 0 00 0 0+ + +? ? ?
  D D D
? ? ?< < <
    r   