
    >ie`                       d dl mZ 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 d dlmZ  ej        d          Zd Zd Zej                            d	d
dddi fd
dddddifd
dddddifddddddifd
dddi fd
dddi fd
ddddd ifddddddifddddi fddddi fddddi fd
dddd d!ifdddd"d d#ifd
ddd$d difd
ddd%i fg          d&             Zej                            d'ee
g          ej                            d(ddd ed)fd*dd ed)fdd*d ed)fddded)fddded)fdddedfd+d,d-ed.fd+d,d ed/fg          d0                         Zd1 ZdS )2    )annotationsN)config)	AxisError)	assert_eqcupyc                    t                               d          } t          j        | dd          }t	          |j                  t           j        k    sJ t          ||           t          t          j        |          t                               |                      | | z   dz   } ||z   dz   }t          j        |          }t                               |           }t	          |j                  t           j        k    sJ t          ||           t          ||           t                               d          	                    d          }t          j        |dd          }t	          |j                  t           j        k    sJ t          ||           t          t          j        |          t                               |                     d S )	N      Fchunksasarray   @   )   r   )r   r   )
r   arangeda
from_arraytype_metandarrayr   diagreshape)vdvdarrcupyarrxdxs         Clib/python3.11/site-packages/dask/array/tests/test_cupy_creation.py	test_diagr!      sm   BA	qu	5	5	5B>>T\))))b"bgbkk499Q<<(((	A	A	b1B72;;DiillG
t|++++dDdGB''A	q	7	7	7B>>T\))))b"bgbkk499Q<<(((((    c            	     0
   t                               d          } t          j        t                    5  t          j        |            d d d            n# 1 swxY w Y   t                               d                              d          } t          j        t                    5  t          j        | dd           d d d            n# 1 swxY w Y   t          j        t                    5  t          j        | d           d d d            n# 1 swxY w Y   t          j        t                    5  t          j        | d           d d d            n# 1 swxY w Y   t                               d	                              d
          } t          j	        | dd          } t          t          j        |           t          j        |                      t          t          j        | d          t          j        | d                     t          t          j        | d          t          j        | d                     t          t          j        |                                           t           j                  sJ t          j        t                    5  t          j        | d           d d d            n# 1 swxY w Y   t          t          j        | d          t          j        | d                     t          t          j        | dd          t          j        | dd                     t                               d                              d          } t          j	        | dd          } t          t          j        |           t          j        |                      t          t          j        | ddd          t          j        | ddd                     t          t          j        | ddd          t          j        | ddd                     t          t          j        | ddd          t          j        | ddd                     t          t          j        | d          t          j        | d                     t          t          j        | d          t          j        | d                     t          t          j        | ddd          t          j        | ddd                     d S )Nr	   r   )   r$   r   )axis1axis2)r%   )r&   x   )r         r$   Fr   
   )offseti   )r,   r%   i  )r$   r   r   r)   r*   )r/   r/   r$   )r/   r$   r/   )r$   r/   r$   )r)   r/   r   )r,   r%   r&   )r   r   pytestraises
ValueErrorr   diagonalr   r   r   r   np
isinstancecomputer   )r   s    r    test_diagonalr8   &   sz   BA	z	"	"  
A               	Av&&A	z	"	" ) )
AQa(((() ) ) ) ) ) ) ) ) ) ) ) ) ) ) 
y	!	! ! !
AR    ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 
y	!	! ! !
AR    ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 	I&&y11A
a5111Abk!nnbk!nn---bk!B'''Qr)B)B)BCCCbk!C((("+a*D*D*DEEEbk!nn,,..=====	z	"	" ! !
AR    ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! bk!2&&&AR(@(@(@AAAbk!AR000"+aQS2T2T2TUUU 	%&&..??A
	;U	 	 	A bk!nnbk!nn---
Aaq222
Aaq222  
 
Ab333
Ab333  
 
Ab333
Ab333  
 bk!QQ!2!2333bk!R  "+a"4"4555bk!Q1%%r{1aA'>'>?????sY   AAA&C

CC.DDD5EEE$KKKz&shape, chunks, pad_width, mode, kwargs)r+   )r   r/   constantr$   constant_valuesr.   r$   r   r.   r-   )r+   r	   )r   r)   ))r/   r   r;   )r<   r$   r/   r   edgelinear_ramp
end_values)r<   )r   r   reflect	symmetricwrapmaximumstat_lengthr0   mean)r   r   r=   minimumemptyc                .   t           j                                                            |           }t          j        t
                              |          |          }t          j        |||fi |}t          j        |||fi |}t          |j	        t
          j
                  sJ t          |                                t
          j
                  sJ |dk    r&t          |||          |||          d           d S t          ||d           d S )Nr   rI   F
check_type)r5   randomdefault_rngr   r   r   arraypadr6   r   r   r7   r   )	shaper   	pad_widthmodekwargsnp_ada_anp_rda_rs	            r    test_padrZ   `   s   B 9  ""))%00D=D))&999D6$	422622D6$	422622Ddj$,/////dllnndl33333wI:%&Y	z-A(Bu	
 	
 	
 	
 	
 	
 	$//////r"   xpzN, M, k, dtype, chunksautor   r*   r   r-   rG   )r   r\   c           	         ||||g}t          j        | }| t          u r|                    |            | j        |dt          j        t                               d                    i}t          ||           d S )Nlike )r   trir   appendr   rP   r   )	r[   NMkdtyper   argscp_axp_as	            r    test_tri_likeri      sx     q!UD8T?D	RxxF264<bmDJJrNN;;<<DdDr"   c                    t          j        ddi          5  t          j        t                              d          d          } t          | j        t          j                  sJ | 	                                }t          |j        t          j                  sJ | 	                    d          }t          |j        t          j                  sJ t          j        ddi          5  | 	                    d          }t          |j        t          j                  sJ | 	                                }t          |j        t          j                  sJ 	 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          | | 	                    d          d           t          | | 	                    d                     d S )	Nzarray.backendnumpyr	   r
   rK   r   FrL   )r   setr   r   r   r   r6   r   r   
to_backendr5   r   )r   x_news     r    test_to_backend_cupyro      s   	_g.	/	/ 9 9M$++b//$777!'4<00000 %+rz22222 V$$%+t|44444 Z&122 		9 		9 LL))Eek2:66666 LLNNEek4<888888		9 		9 		9 		9 		9 		9 		9 		9 		9 		9 		9 		9 		9 		9 		9!9 9 9 9 9 9 9 9 9 9 9 9 9 9 96 ag&&59999af%%&&&&&s7   CE8(A,E!E8!E%	%E8(E%	)E88E<?E<)
__future__r   rk   r5   r1   markgpu
pytestmark
dask.arrayrP   r   daskr   dask.array.numpy_compatr   dask.array.utilsr   importorskipr   r!   r8   parametrizerZ   floatboolintri   ro   r_   r"   r    <module>r}      s   " " " " " "     [_
             - - - - - - & & & & & &v6"") ) ),7@ 7@ 7@t ,	aR(	a&7%<=	v
->,IJ 23	
 
a$	a+	aq(9:-.	
 
6+Y;	6+["=	6+VR8	v	M6+BC	6+VmEU5VW	v	M6+BC	a"%7 @0 0A @0& Bx((	
D!UF#	
D!UF#	
Aq$	
D!S&!	
D"c6"	
D!S!	
Ar3	
Aq#{#	 	 	  )(	' ' ' ' 'r"   