
    bi?                        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y)    )annotationsN)	assert_eq))A)f4      B)r   r   )C)f8r   ))r   )i4r   r
   )r   )S4r   )paramsc                @    t        j                  | j                        S N)npdtypeparamrequests    ]/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/tests/test_numpy_compat.pyr   r   
   s     88GMM""    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        ||       y )N))ar   )br   )cr   )   r   )r   r   )r   chunksr   r   )r   onesdar   )r   xdxresultexpecteds        r   
test_basicr(      sM    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 y )Nr      r	   )r*   r	   )r!   )r*   r*   )r#   
from_arrayr   arrayintminmaxround)images    r   test_min_max_round_funcsr2   #   s{    MM"((QFQF#34VDE rvve}"""rvve}"""88E?4 A%%%r   )
__future__r   numpyr   pytest
dask.arrayr,   r#   dask.array.utilsr   fixturer   r   r(   r2    r   r   <module>r:      st    "    & CC## Sz?;< = &r   