
    >ieQ                    2   d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
Z
d dlZd dlZd dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlm Z  d Z!d Z"d	 Z#d
 Z$d Z%d Z&d Z'd Z(d Z)e	d             Z*e
j+        ,                    ej-        dk    d          e
j+        .                    dddg          d                         Z/d Z0d Z1d Z2d Z3d Z4d Z5d Z6d Z7d  Z8d! Z9d" Z:d# Z;e
j+        .                    d$d%d&g          d'             Z<d( Z=d) Z>d* Z?e
j+        .                    d+d,d-d.d/d0gd/d1gfd2d/d0hd/d1hfd3d0id3d1ifd4g d5d6d4g d7d6fg          d8             Z@d9 ZAd: ZBe
j+        .                    d;d<d=g          d>             ZCd? ZDd@ ZEdA ZFdB ZGdC ZHe
j+        .                    dDdEdFdGifdFdGifi fdHdGifi fg          dI             ZIdJ ZJe
j+        .                    d;dHdFg          dK             ZKdL ZLdM ZMdN ZNdO ZOdP ZPdS )Q    )annotationsN)OrderedDict)contextmanager)
_get_pathscanonical_namecollectcollect_envcollect_yamlconfigdeserializeensure_fileexpand_environment_variablesgetmergepoprefreshrename	serializeupdateupdate_defaults)tmpfilec                    ddd} t          d|           dk    sJ t          d|           dk    sJ t          d|           dk    sJ t          d|           dk    sJ t          d|           dk    sJ t          d	|           d	k    sJ d S )
N      )foo-bar	fizz_buzzr   foo_barz	fizz-buzzr   znew-keynew_key)r   )cs    6lib/python3.11/site-packages/dask/tests/test_config.pytest_canonical_namer!   $   s    A&&A)Q''94444)Q''94444+q))[8888+q))[8888)Q''94444)Q''9444444    c                     dddid} ddt          ddi          d}t          ||            |dddddd	k    sJ dddid} dddddd}t          || d
           |dddddd	k    sJ d S )Nr   axyr      br&   zr'   r$   r)   r&   r'   r+   oldpriorityr   r   r,   s     r    test_updater2   .   s    sAhAak3(3344A
1aLLLaAA..Q777777sAhAaAA..//A
1a%    aAA..Q77777777r"   c                 R   dddddd} ddddddddid}ddt          ddd          dddid}t          ||d| 	          ddddddddidk    sJ t          ||d|	          t          ||d
          k    sJ t          ||dd 	          t          ||d          k    sJ d S )Nr   r,   r-   r   d)r&   r'   r+   r   c2r(   znew-defaults)r0   defaultsnewr/   r.   r1   )r4   ons      r    test_update_new_defaultsr:   :   sC   aAA..//AaAA..QsAhGGAak*:*:;;!CQR8TTA!Q!<<<1AhA A     !Q!<<<	1uA A A     !Q$???6	1uD D D      r"   c                     dddiddddidg} dddddd	}dddddd
}t          |||            | dddiddddiddddddd
gk    sJ |dddddddk    sJ d S )Nr   r   r,   r   r4   r(   )r   r4   r   )r$   r)   extra)r$   r)   	new-extra)r6   )r$   r)   r<   r=   )r   )r6   currentr7   s      r    test_update_defaultsr?   M   s    sAhsAhH !!,,q99Gaa((q
9
9CC84444sAhsAhAA&&Q77    
 A!!$4$4qqQQQQQQQQr"   c                 ^    ddddi} ddgdd}t          ||            |dddddk    sJ d S Nr&   r   r   r'   r+   r(   )r&   wr   r,   s     r    test_update_list_to_dictrE   ^   sY    	AAA!f1A
1aLLLAA&&Q////////r"   c                 \    dddgi} ddddd}t          ||            |ddgddk    sJ d S rA   rD   r,   s     r    test_update_dict_to_listrG   e   sT    	q!fAAQ''A
1aLLLq!f1%%%%%%%%r"   c                 f    dddid} ddddid}dddddd	}t          | |          }||k    sJ d S )
Nr   r$   r%   r   r(   r)   r*   r,   r-   )r   )r$   r)   expectedr   s       r    
test_mergerJ   l   s[    sAhAasAh''A11--A66HaA======r"   c                    dddid} ddddid}dddddd	}t          d
          5 }t          d
          5 }t          |d          5 }t          j        | |           d d d            n# 1 swxY w Y   t          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   t	          t          ||g           }||k    sJ 	 d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr   r$   r%   r   r(   r)   r*   r,   r-   yaml	extensionrC   paths)r   openrL   dumpr   r
   )r$   r)   rI   fn1fn2fr   s          r    test_collect_yaml_pathsrV   v   s   sAhAasAh''A11--A66H	6	"	"	" &cv&&& 	&#c3  1	!Q                             c3  1	!Q                              LSz:::;FX%%%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&& & & & & & & & & & & & & & & & & &s|   C;C#A-!C#-A11C#4A15C#B*C#*B..C#1B.2$C#C;#C'	'C;*C'	+C;;C?C?c                 ^   dddid} ddddid}dddddd	}t                      5 }t          j        |           t          t          j                            |d
          d          5 }t          j        | |           d d d            n# 1 swxY w Y   t          t          j                            |d          d          5 }t          j        ||           d d d            n# 1 swxY w Y   t          t          |g           }||k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r$   r%   r   r(   r)   r*   r,   r-   a.yamlrC   modeb.yamlrO   )
r   osmkdirrQ   pathjoinrL   rR   r   r
   )r$   r)   rI   dirnamerU   r   s         r    test_collect_yaml_dirra      s   sAhAasAh''A11--A66H	 "g
"',,w11<<< 	IaOOO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	"',,w11<<< 	IaOOO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 G95556!!!!!" " " " " " " " " " " " " " " " " "sZ   AD"+BD"B	D"B	2D"C)D")C-	-D"0C-	1#D""D&)D&c              #    K   t          j        t          j         |           j                  }|t           j        z  }	 t          j        | |           d V  t          j        | |           d S # t          j        | |           w xY wN)statS_IMODEr\   st_modeS_IREADchmod)r^   	perm_origperm_news      r    no_read_permissionsrk      s|      RWT]]233I4<'H"
x   
y!!!!!y!!!!s   A. .Bwin32z$Can't make writeonly file on windows)reasonkind	directoryfilec                `   ddd}ddd}t          |           }t          j                            |d          }t          j                            |d          }t	          |d	
          5 }t          j        ||           d d d            n# 1 swxY w Y   t	          |d	
          5 }t          j        ||           d d d            n# 1 swxY w Y   |dk    r|}i }	n|}|}	t          |          5  t          t          |g           }
|
|	k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r   r%   r(      rB   rX   r[   rC   rY   ro   rO   )
strr\   r^   r_   rQ   rL   rR   rk   r   r
   )tmpdirrn   r$   r)   dir_patha_pathb_pathrU   	cant_readrI   r   s              r    #test_collect_yaml_permission_errorsry      s   
 aAaA6{{HW\\(H--FW\\(H--F	f3			 1	!Q              	f3			 1	!Q               {			Y	'	' " "H:6667!!!!!" " " " " " " " " " " " " " " " " "s6   +BBB)CCC4!D##D'*D'c                   t          |           }t          j                            |d          }t	          |d          5 }|                    d           d d d            n# 1 swxY w Y   t          j        t                    5 }t          |g           d d d            n# 1 swxY w Y   t          |          t          |j                  v sJ dt          |j                  v sJ dt          |j                  v sJ d S )NrX   wbrY      {rO   is malformedzoriginal error messagers   r\   r^   r_   rQ   writepytestraises
ValueErrorr
   reprvaluert   ru   fil_pathrU   recs        r     test_collect_yaml_malformed_filer      sd   6{{Hw||Hh//H	hT	"	"	" a	               
z	"	" 'cH:&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '>>S^^++++S^^++++#s39~~555555$   A##A'*A'B%%B),B)c                   t          |           }t          j                            |d          }t	          |d          5 }|                    d           d d d            n# 1 swxY w Y   t          j        t                    5 }t          |g           d d d            n# 1 swxY w Y   t          |          t          |j                  v sJ dt          |j                  v sJ dt          |j                  v sJ d S )NrX   r{   rY   s   [1234]rO   r}   zmust have a dictr~   r   s        r    #test_collect_yaml_no_top_level_dictr      sg   6{{Hw||Hh//H	hT	"	"	" a		               
z	"	" 'cH:&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '>>S^^++++S^^++++SY//////r   c            	     f    ddddddddd} d	d
dd	ddg ddd}t          |           }||k    sJ d S )N123Truehello456z[1, 2, "3"]z/not/parsable/as/literalznot included)DASK_A_BDASK_CDASK_D	DASK_E__X	DASK_E__YDASK_FDASK_GFOO{   T  r%   )r   r   3)a_br   r4   erU   gr	   envrI   ress      r    test_envr      sr    ,	 	C S!![[' H c

C(??????r"   c                 X    ddddddd} d d d d d d d}t          |           }||k    sJ d S )	NNoneNONEnoneNullNULLnull)DASK_ADASK_Br   r   DASK_Er   )r$   r)   r   r4   r   rU   r   r   s      r    test_env_none_valuesr      s`     C  H c

C(??????r"   c                    dddid} ddddid}dd	i}d
dddddd}t          d          5 }t          d          5 }t          |d          5 }t          j        | |           d d d            n# 1 swxY w Y   t          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   t	          ||g|          }||k    sJ 	 d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr   r$   r%   r   r(   r)   r*   DASK_W4rr   r,   )rC   r&   r'   r+   rL   rM   rC   )r   )r   rQ   rL   rR   r   )r$   r)   r   rI   rS   rT   rU   r   s           r    test_collectr     s'   sAhAasAh''AS/CQ11%5%5A>>H	6	"	"	" &cv&&& 	&#c3  1	!Q                             c3  1	!Q                              c3ZS111FX%%%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&& & & & & & & & & & & & & & & & & &s|   C:C"A2&C"2A66C"9A6:C"B/#C"/B33C"6B37C"C:"C&	&C:)C&	*C::C>C>c                |    |                      dd           t          g           } |j        d          dk    sJ d S )NDASK_FOObarfoo)setenvr   r   )monkeypatchr   s     r    test_collect_env_noner     sF    z5)))R[[F6:e%%%%%%r"   c                 &   dddid} t          d|           dk    sJ t          d|           dk    sJ t          dd	|           d	k    sJ t          j        t                    5  t          d|            d d d            d S # 1 swxY w Y   d S )
Nr   r$   r   r%   r&   r   zy.azy.br   )r   r   r   KeyErrorr4   s    r    test_getr   $  s    sAhAs1""""uQ1$$$$uc!$$$++++	x	 	   E!                 s   'BB
B
c                   dddid}ddi}t           j                            t          |           d          }t           j                            t          |           d          }t           j                            |d          }t	          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   t          ||d	
           t	          |          5 }t          j        |          }d d d            n# 1 swxY w Y   ||k    sJ t	          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   t          ||d	
           t	          |          5 }t          j        |          }d d d            n# 1 swxY w Y   ||k    sJ t          j	        |           t          ||d
           t	          |          5 }|
                                }d d d            n# 1 swxY w Y   d|v sJ t	          |          5 }t          j        |          }d d d            n# 1 swxY w Y   |rJ d S )Nr   r$   r%   r&   r   source.yamldestrC   F)sourcedestinationcommentTr   )r\   r^   r_   rs   rQ   rL   rR   r   	safe_loadremoveread)	rt   r$   r)   r   r   r   rU   resulttexts	            r    test_ensure_filer   .  s   sAhA	c
AW\\#f++}55F7<<FV,,D',,t]33K	fc		 a	!Q               v4????	k		 #a""# # # # # # # # # # # # # # #Q;;;; 
fc		 a	!Q               v4????	k		 #a""# # # # # # # # # # # # # # #Q;;;;Ik v4>>>>	k		 avvxx              D====	k		 #a""# # # # # # # # # # # # # # #sl   B88B<?B<$DD	D	(E

EE6FFFH  HH IIIc                    t           j                            d          5  t          d         dk    sJ t           j                            d          5  t          d         dk    sJ 	 d d d            n# 1 swxY w Y   t          d         dk    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ t           j                            ddi          5  t          d         dk    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ t           j                            dddd          5  t          d         ddd	did
k    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ i } t           j                            ddi|            | d         d         dk    sJ ddd} t           j                            ddddd|            | dddddk    sJ d S )Nr   )abcr   r   r   r   r(   )abc.xabc.yzabc.z.ar$   r-   r   r   r&   r   )e_fg-hrr      )r   c-de-fg_h)r   r   r   r   daskr   setr   s    r    test_setr   W  s   	S	!	! $ $e}####[___%% 	( 	(%=C'''''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	(e}#####	$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ 	%	&	& $ $e}#####$ $ $ $ $ $ $ $ $ $ $ $ $ $ $	1qQ??	@	@ @ @e}aasAh ? ??????@ @ @ @ @ @ @ @ @ @ @ @ @ @ @
AKOOWcN1O---U8C=C !AKOOAa!<<QOGGG!Aa88888888sY   4B!A6*B!6A:	:B!=A:	>B!!B%(B%C99C= C=2EE Ec                 @   t           j                            dd          5  t          d         dddk    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ t           j                            ddddd	          5  t          d         dddd
k    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ t           j                            ddddidd	          5  t          d         dddd
k    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ d S )Nr   r   )foo__barfoo__bazr   )r   baz)foo.barzfoo.bazr(   rr   )	foo__buzzr   )r   r   buzzr    r"   r    test_set_kwargsr   r  s%   	!a	0	0 5 5e}! 4 4444445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 
Q155Q	O	O @ @e}!Q ? ??????@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ 
%1!5!56!a	P	P @ @e}!Q ? ??????@ @ @ @ @ @ @ @ @ @ @ @ @ @ @s5   AA
A
B&&B*-B*#DDDc                    t           j                            dddii          5  t          d         ddik    sJ t           j                            ddi          5  t          d         dddk    sJ 	 d d d            n# 1 swxY w Y   t          d         ddik    sJ 	 d d d            n# 1 swxY w Y   dt          vsJ d S )Nr   r&   r   r   r   r%   r   r   r"   r    test_set_nestedr     sO   	%#s,	-	- + +e}c
****[__gs^,, 	9 	9%=#C$8$888888	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9e}c
*****	+ + + + + + + + + + + + + + +
 s5   7B,A?3B,?B	B,B	B,,B03B0c                    dd l } t          j                            |                                           5  t          j                            d          5  	 d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr   )r&   r   )r'   )	threadingr   r   r   Lock)r   s    r    test_set_hard_to_copyablesr     s    	9>>++	,	,  [__q_!! 	 		 	 	 	 	 	 	 	 	 	 	 	 	 	 	                 s5   !A>A&A>&A*	*A>-A*	.A>>BBr]   TFc                P   dddid}t           j                            t          |          d          }t           j                            t          |          d          }t	          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   | rt          j        |           t          ||           t           j        	                    |          sJ t           j        
                    t           j                            |d                    sJ d S )Nr   r$   r%   r   r   rC   )r   r   )r\   r^   r_   rs   rQ   rL   rR   r]   r   isdirexists)r]   rt   r$   r   r   rU   s         r    test_ensure_file_directoryr     s.   sAhAW\\#f++}55F7<<FV,,D	fc		 a	!Q                
v400007==7>>"',,t];;<<<<<<<s   2BBBc                   dddid}t           j                            t          |           d          }t	          |d          5 }t          j        ||           d d d            n# 1 swxY w Y   t           j                            t          |           d          }t          j        j	        }	 |t          j        _	        t          |           |t          j        _	        n# |t          j        _	        w xY wt           j                            |          sJ t          j        |          \  }t           j                            |          d         t           j                            |          d         k    sJ d S )Nr   r$   r%   r   rC   r   )r   )r\   r^   r_   rs   rQ   rL   rR   r   r   PATHr   r   listdirsplit)rt   r$   r   rU   r   r   fns          r    2test_ensure_file_defaults_to_DASK_CONFIG_directoryr     sq   sAhAW\\#f++}55F	fc		 a	!Q               ',,s6{{F33K;D &6""""47==%%%%%:k""DR7==Q27==#8#8#;;;;;;;s   A''A+.A+1!C$ $C7c                 >   ddddd} t          j        t                    5  t          d|            d d d            n# 1 swxY w Y   t          dd| 	          dk    sJ t          d
|           dk    sJ t          d|           dk    sJ | dddiik    sJ d S )Nr   r   )zba-rr   r(   )r   asdr&   r   rr   )defaultr   zfoo.ba_rr   r   r   )r   r   r   r   r   s    r    test_popr     s   **155F	x	 	     C                             sAf---2222z&)))Q....uV$$$))))eeQZ(((((((s   A  AAc                     g } i }t          ddi||            |ddik    sJ t          g ddi||            |dddk    sJ t          g d	d
i||            |dddk    sJ d S )Nr$   r   )r   r6   r   2)rP   r   r   r6   r   r,   r   r   r(   r$   r   )r   r   )r6   r   s     r    test_refreshr     s    HFS!HVh????c1X"8S/&8LLLL11%%%%%%"8S/&8LLLL11%%%%%%%%r"   zinp,out)1r   )r   r   )$FOOr   r   r   r   ))r   r   )r   r   r$   A)r   r   r   r,   )r   r   r   c                \    |                      dd           t          |          |k    sJ d S )Nr   r   )r   r   )r   inpouts      r    !test_expand_environment_variablesr     s9     ue$$$',,333333r"   c                    d}|                      dt          |                     i }t          j                            |           t          d|          |k    sJ t          d|          |k    sJ d S )Nr(   r   r   r   a-b)r   rs   r   r   r   r   )r   r   r4   s      r    test_env_var_canonical_namer    s    Ez3u::...
AKq!!!uQ5((((uQ5((((((r"   c                 j   dddii} g d}|D ])}t           j                            ||           dk    sJ *t           j                            dddii|           5  |D ])}t           j                            ||           dk    sJ *	 d d d            n# 1 swxY w Y   t           j                            dd	d
iddi|           5  t           j                            d|           d	d
ik    sJ t           j                            d|           dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nzx-yr   r   )zx_y.a_bzx-y.a-bx_y.a-br   x_yr  r   c_dr   r   )r  r   r  zx_y.e_f)r   r   r   r   )r   keysks      r    test_get_set_canonical_namer
    s   	A,,,D 3 3{q++s22222	%%.q	9	9 7 7 	7 	7A;??1Q?//366666	77 7 7 7 7 7 7 7 7 7 7 7 7 7 7
 
%%A!>!>?	J	J 9 9{y33qzAAAA{y33q888889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s%   -BBB	AD((D,/D,key
custom_key
custom-keyc                   d}t           j                            | |i          5  t           j                            d          |k    sJ t           j                            d          |k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r  r  r   r   r   r   )r  r   s     r    test_get_set_roundtripr    s    E	#u	&	& 6 6{|,,5555{|,,555556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   AA==BBc                 j    t           j                            d dddddii          ddiddk    sJ d S )Nr   r   r$   r)   r   )r   r   r   r   r"   r    test_merge_None_to_dictr    s[    ;4a003a/BB1XG G      r"   c                     dt           j        j        v sJ dt           j        j        v sJ dt           j                            d          v sJ d S )Nztemporary-directory	dataframecompressionzdataframe.shuffle)r   r   r   r   r"   r    test_core_filer    sT     DK$66666$+,,,,,DKOO,?@@@@@@@@r"   c                 R   t          j        d          } t          j                            t          j                            t                    dd          }t          j                            t          j                            t                    dd          }t          |          5 }t          j	        |          }d d d            n# 1 swxY w Y   t          |          5 }t          j	        |          }d d d            n# 1 swxY w Y   | 
                    ||           d S )N
jsonschema..	dask.yamldask-schema.yaml)r   importorskipr\   r^   r_   r`   __file__rQ   rL   r   validate)r  	config_fn	schema_fnrU   r   schemas         r    test_schemar"    sW   $\22JRW__X66kJJIRW__X66>PQQI	i #A""# # # # # # # # # # # # # # # 
i #A""# # # # # # # # # # # # # # # '''''s$   *CCC%DD
D
c                 "   t           j                            t           j                            t                    dd          } t           j                            t           j                            t                    dd          }t          |           5 }t          j        |          }d d d            n# 1 swxY w Y   t          |          5 }t          j        |          }d d d            n# 1 swxY w Y   fd ||           d S )Nr  r  r  c           
     t   |                                  D ]\  }}t          |           t          |d                   k    rCt          d                    t	          |           t	          |d                                       t          |t                    r | |         |d         |                    d S )N
propertiesa  
The dask.yaml and dask-schema.yaml files are not in sync.
This usually happens when we add a new configuration value,
but don't add the schema of that value to the dask-schema.yaml file
Please modify these files to include the missing values: 

    dask.yaml:        {}
    dask-schema.yaml: {}

Examples in these files should be a good start, 
even if you are not familiar with the jsonschema spec)itemslistr   formatsorted
isinstancedict)r   sr	  vtest_matchess       r    r.  z-test_schema_is_complete.<locals>.test_matches+  s    GGII 	7 	7DAqAww$q//// L MSFq		6!L/#:#:M M   !T"" 7QqT1\?1#5666	7 	7r"   )r\   r^   r_   r`   r  rQ   rL   r   )r  r   rU   r   r!  r.  s        @r    test_schema_is_completer/  !  s]   RW__X66kJJIRW__X66>PQQI	i #A""# # # # # # # # # # # # # # # 
i #A""# # # # # # # # # # # # # # #7 7 7 7 7$ L     s$   B88B<?B<C33C7:C7c                    dddddd} ddd	d
dd iddid}t          j        t                    5 }t          | |           d d d            n# 1 swxY w Y   d |j        D             g dk    sJ |ddddd	ii d ddk    sJ d S )Nr   zfoo.yabz	not.foundsuper)r   zx.yza.bz	not-foundz	super.oldr   r   r(   rB   r)   r.   rr   )r   r&   r$   r2  r   c                6    g | ]}t          |j                  S r   )rs   message).0wis     r    
<listcomp>ztest_rename.<locals>.<listcomp>P  s     ---C
OO---r"   )zRDask configuration key 'foo-bar' has been deprecated; please use 'foo.bar' insteadzLDask configuration key 'x.y' has been deprecated; please use 'foo.y' insteadzIDask configuration key 'a.b' has been deprecated; please use 'ab' insteadzRDask configuration key 'super.old' has been deprecated; please use 'super' instead)r   r'   r+   )r   r&   r$   r1  r2  )r   warnsFutureWarningr   r'  )aliasesr   rC   s      r    test_renamer;  @  s<     G 14[	 F 
m	$	$ 'wv&&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '--af--- 2 2 2     q!!1X       s   AAAzargs,kwargsr   fuse_ave_widthr   zfuse-ave-widthc                L   t          j        t                    5 }t          j        j        | i |5  t          j                            d          dk    sJ 	 d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   dt          |d         j                  v sJ d S )Noptimization.fuse.ave-widthr   r   )	r   r8  r9  r   r   r   r   rs   r4  )argskwargsinfos      r    test_deprecations_on_setrB  _  s=    
m	$	$ I[_d-f-- 	I 	I;??#@AASHHHHH	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	I 	II I I I I I I I I I I I I I I )CQ,@,@@@@@@@s4   A;&A$A;$A(	(A;+A(	,A;;A?A?c                B   i }|                      dd           t          j        t                    5 }t          j                            |           d d d            n# 1 swxY w Y   dt          |d         j                  v sJ t          d|          dk    sJ d S )NDASK_FUSE_AVE_WIDTHr   r   r>  r   r   )
r   r   r8  r9  r   r   r   rs   r4  r   )r   r4   rA  s      r    "test_deprecations_on_env_variablesrE  n  s    
A,e444	m	$	$ &1%%%& & & & & & & & & & & & & & &(CQ,@,@@@@@,Q7773>>>>>>s   !AA#&A#c                   i }t          | dz  d          5 }t          j        |di|           d d d            n# 1 swxY w Y   t          j        t
                    5 }t          j                            || g           d d d            n# 1 swxY w Y   dt          |d         j
                  v sJ t          d|          dk    sJ d S )Nr  rC   r   )r   rP   r>  r   r   )rQ   rL   rR   r   r8  r9  r   r   r   rs   r4  r   )tmp_pathr  r4   fhrA  s        r    test_deprecations_on_yamlrI  w  sI   
A	h$c	*	* "b	3*b!!!" " " " " " " " " " " " " " " 
m	$	$ 81XJ7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8(CQ,@,@@@@@,Q7773>>>>>>s   :>>#BBBc                    t           j                            ddi          5  t           j                            d          dk    sJ t           j                            dd           dk    sJ t           j                            dd          dk    sJ t           j                            dd          du sJ t           j                            dd          du sJ t           j                            dd          dk    sJ t           j                            ddd	i          dd	ik    sJ t           j                            dd
g          d
gk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r   )override_withr   FTr   r   worldoner  r   r"   r    test_get_override_withrN    s   	%	(	( H H{u%%....{uD99UBBBB {uE::eCCCC{uE::eCCCC{uD99TAAAA{uC88C????{uWg4FGGWL
 
 
 
 
 {uUG<<GGGGGH H H H H H H H H H H H H H H H H Hs   D9E))E-0E-c                    t           j                            dt           j                            d          i          5  t	          ddddiii          } t          |           }t           j                            t           j        j        |           t           j                            d          dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Narray.svg.sizearraysvgsize   )r   r   r   r   r   r   r   global_config)
serializedr   s     r    test_config_serializationrW    s    	*DKOO<L,M,MN	O	O 8 8%&#)?@AA
Z((4;4f==={/00C777778 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s   A9CC
C
c            	         t          dt          ddddiii          i          } t          j                            d|           dk    sJ d S )NDASK_INTERNAL_INHERIT_CONFIGrQ  rR  rS  rT  rP  r   )r	   r   r   r   r   r   s    r    test_config_inheritancerZ    s[    	'Gefc]=S3T)U)UV F ;??+F?;;sBBBBBBr"   c                   |                      dd           |                      dd           |                     t          dg            dt          j                            t          j        dd          t          j                            t          j                            d	          d
d          g}t                      }||k    sJ t          |          t          t          |                    k    sJ |                                 5 }|                    dd           t                      }||dgz   k    sJ t          |          t          t          |                    k    sJ 	 d d d            n# 1 swxY w Y   |                                 5 }|                    dd           t                      }|dg|dd          z   k    sJ t          |          t          t          |                    k    sJ 	 d d d            n# 1 swxY w Y   |                                 5 }t          j                            ddd          }|                    t          dt          j        |gz              t                      }t          j                            |dd          |v sJ t          |          t          t          |                    k    sJ 	 d d d            d S # 1 swxY w Y   d S )NDASK_CONFIGF)raisingDASK_ROOT_CONFIGPREFIXESz	/etc/dasketcr   ~z.configr   r   includethisr^   )delenvsetattrsiter\   r^   r_   sysprefix
expanduserr   lenr   contextr   r_  )r   rI   rP   mrh  s        r    test__get_pathsrm    s\    }e444)5999j"--- 	
SZ//
RW'',,i@@H
 LLEHu::SZZ((((					 -!		***I;.....5zzSU__,,,,,	- - - - - - - - - - - - - - - 
				 -!	#Y///hqrrl222225zzSU__,,,,,	- - - - - - - - - - - - - - - 
				 -!i88			$
DMVH$<===w||FE622e;;;;5zzSU__,,,,,- - - - - - - - - - - - - - - - - -s9   A E;;E?E?A(HHH.B.K**K.1K.c                 L    t           j        j        t                      k    sJ d S rc   )r   r   rP   r   r   r"   r    test_default_search_pathsro    s"    ;
,,,,,,r"   )Q
__future__r   r\   rf  rd   rg  collectionsr   
contextlibr   r   rL   dask.configr   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   
dask.utilsr   r!   r2   r:   r?   rE   rG   rJ   rV   ra   rk   markskipifplatformparametrizery   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r
  r  r  r  r"  r/  r;  rB  rE  rI  rN  rW  rZ  rm  ro  r   r"   r    <module>ry     s   " " " " " " 				   



 # # # # # # % % % % % %                                           &      5 5 5	8 	8 	8  &R R R"0 0 0& & &  & & &"" " "" " " " LG$J    +v!677" " 87 "26 6 60 0 0  2  .& & &$& & &  & & &R9 9 96        4-00= = 10=$< < <&) ) )& & & 
Vq%j!!
Vq%j!
ve%)))	*	*#OOO,L,LM	 4 4 4
) ) )9 9 9" | <==6 6 >=6  A A A( ( (! ! !>  > 	$%S
!	#R(S
!	#R( A A A? ? ? !13C DEE? ? FE?H H H"8 8 8C C C"- "- "-J- - - - -r"   