
    CdM                        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dS )    )annotations)addsleepN)Cache)Callback)get_sync)getcacheyc                @    t                               |            | dz   S )N   )flagappend)xs    5lib/python3.11/site-packages/dask/tests/test_cache.pyincr      s    KKNNNq5L    c                 R   t                               d          } t          |           }|5  t          dt          dfid          dk    sJ 	 d d d            n# 1 swxY w Y   t          dgk    sJ | j        d         dk    sJ |j        rJ |j        rJ t          r t                                           t           t          dft          dft          ddfd}|5  t          |d          dk    sJ 	 d d d            n# 1 swxY w Y   t          dgk    sJ t          j        rJ d S )	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;;;;6#;!}|
 


  aQxsCo
>
>C	 " "3}}!!!!!" " " " " " " " " " " " " " " A3;;;;s#    AA AD  DDc                     t          dd          } t          | j        t          j                   sJ | j        j        dk    sJ | j        j        dk    sJ d S )Nr   r   )limit)r   
isinstancecacher   available_bytesr$   )r   s    r   test_cache_with_numberr(   0   s_    e1Aagv|,,,,,7"e++++7=Ar   c                    t          d          } t          j        d          }ddlm}m} |                     |d          d          }|                     |d          d          }| 5  |                                dk                                    sJ |                                dk                                    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   z
dask.arrayr   )oneszerosr   
   )chunks)	r   pytestimportorskipnumpyr*   r+   
from_arraycomputeall)r   dar*   r+   r   os         r   test_cache_correctnessr6   7   s&   eA		\	*	*B!!!!!!!!
eeAhhr**A
dd1ggb))A	
 ( (		q %%'''''		q %%''''''( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s   .ACCCc                ,    t          |            dg|z  S )Nr   r   )durationsizeargss      r   fr;   D   s    	(OOO3:r   c                    t           ddft           dddfd} t          d          }|5  t          | d           d d d            n# 1 swxY w Y   |j        j        j        d         |j        j        j        d         k     sJ d S )	Ng{Gz?r,   gư>r   r   )r   r   r   r   )r;   r   r	   r&   scorercost)r!   r   s     r   test_prefer_cheap_dependentr?   I   s    D"Q!S$9
:
:CeA	
  c               7>s#agn&9#&>>>>>>>s   AA	A	)
__future__r   operatorr   timer   r.   
dask.cacher   dask.callbacksr   
dask.localr	   dask.threadedr
   r/   r   r   r   r"   r(   r6   r;   r?    r   r   <module>rH      s	   " " " " " "                    # # # # # #            		X	&	& 
  
  0  
( 
( 
(  
? ? ? ? ?r   