
    bin                        d dl mZ d dlmZ d dlmZ d dl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g Zd
 Zd Zd Zd Zd Zd Zy)    )annotations)addsleepN)Cache)Callback)get_sync)getcacheyc                6    t         j                  |        | dz   S )N   )flagappend)xs    P/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/tests/test_cache.pyincr      s    KKNq5L    c                     t         j                  d      } t        |       }|5  t        dt        dfid      dk(  sJ 	 d d d        t        dgk(  sJ | j
                  d   dk(  sJ |j                  rJ |j                  rJ t        rt        j                          t        rt        dft        dft        ddfd}|5  t        |d      dk(  sJ 	 d d d        t        dgk(  sJ t        j                  rJ y # 1 sw Y   xY w# 1 sw Y   4xY w)	N'  r   r      y)r   r   zr      )r   r   r
   r   r   data
starttimes	durationspopr   r   active)cccdsks      r   
test_cacher"      s   UA	qB	 .C#q?C(A---. A3;;66#;!}}||

 aQxsCo
>C	 "3}!!!" A3;;#. ." "s   C8>D8DDc                     t        dd      } t        | j                  t        j                         sJ | j                  j                  dk(  sJ | j                  j
                  dk(  sJ y )Nr   r   )limit)r   
isinstancecacher   available_bytesr$   )r   s    r   test_cache_with_numberr(   0   sT    e1Aaggv||,,,77""e+++77==Ar   c                    t        d      } t        j                  d       t        j                  d      }ddlm}m} |j                   |d      d      }|j                   |d      d      }| 5  |j                         dk(  j                         sJ |j                         dk(  j                         sJ 	 d d d        y # 1 sw Y   y xY w)	Nr   numpyz
dask.arrayr   )oneszerosr   
   )chunks)	r   pytestimportorskipr*   r+   r,   
from_arraycomputeall)r   dar+   r,   r   os         r   test_cache_correctnessr6   7   s    eA
 			\	*B!
eAhr*A
d1gb)A	
 (		q %%'''		q %%'''( ( (s   2ACCc                $    t        |        dg|z  S )Nr   r   )durationsizeargss      r   fr;   E   s    	(O3:r   c                    t         ddft         dddfd} t        d      }|5  t        | d       d d d        |j                  j                  j
                  d   |j                  j                  j
                  d   k  sJ y # 1 sw Y   SxY w)	Ng{Gz?r-   gư>r   r   )r   r   r   r   )r;   r   r	   r&   scorercost)r!   r   s     r   test_prefer_cheap_dependentr?   J   s|    D"Q!S$9
:CeA	
 c 77>>s#aggnn&9&9#&>>>> s   BB
)
__future__r   operatorr   timer   r/   
dask.cacher   dask.callbacksr   
dask.localr	   dask.threadedr
   r0   r   r   r   r"   r(   r6   r;   r?    r   r   <module>rH      sT    "     #  			X	& 

0(
?r   