
    >ie?                        d dl mZ d dlZd dlZd dlmZ d dlm	Z	  ej
        g dg dg          d             Z ej
        dgdd	gg d
g          d             Zd Zd ZdS )    )annotationsN)	assert_eq))A)f4      B)r   r   )C)f8r   ))r   )i4r   r
   )r   )S4r   )paramsc                4    t          j        | j                  S N)npdtypeparamrequests    Blib/python3.11/site-packages/dask/array/tests/test_numpy_compat.pyr   r   
   s     8GM"""    r   r   )r   r   r   c                    | j         S r   )r   r   s    r   indexr      s
    =r   c                     g d} t          j        d|           }t          j        d| d          }|ddg         }|ddg         }t          ||           d S )N))ar   )br   )cr   )   r   )r   r   )r   chunksr   r   )r   onesdar   )r   xdxresultexpecteds        r   
test_basicr(      sg    333E
e$$$A	uQ	/	/	/Bc
^F#s}Hfhr   c                 >   t          j        t          j        ddgddgg          d          } t	          t          j        |                     dk    sJ t	          t          j        |                     dk    sJ t          j        |           d         dk    sJ d S )Nr      r	   )r*   r	   )r!   )r*   r*   )r#   
from_arrayr   arrayintminmaxround)images    r   test_min_max_round_funcsr2   #   s    M"(QFQF#344VDDDE rve}}""""rve}}""""8E??4 A%%%%%%r   )
__future__r   numpyr   pytest
dask.arrayr,   r#   dask.array.utilsr   fixturer   r   r(   r2    r   r   <module>r:      s    " " " " " "            & & & & & & CCCCCC  # # # Sz???;<<<  =<     & & & & &r   