B
    T\                  @   sz   d dl Z d dlmZ d dlmZ d dlmZ dddZdd	iZd
diZ	dd Z
dd Ze jjdd Zdd Zdd ZdS )    N)merge)Mapping)	ShareDict      )xyz   wc           	   C   s|   t  } t| tst| t | t | d dks8ttt	 | d  W d Q R X tt
tf d| d< W d Q R X d S )Nr   r   abc{   )r   
isinstancer   AssertionErrorupdateabpytestZraisesKeyErrorNotImplementedError	TypeError)s r   8lib/python3.7/site-packages/dask/tests/test_sharedict.py	test_core   s    

r   c                  sH   t    t  t  t t fddtttgD sDtd S )Nc             3   s*   | ]" t  fd dj D V  qdS )c             3   s   | ]}| kV  qd S )Nr   ).0d)r   r   r   	<genexpr>"   s    z+test_structure.<locals>.<genexpr>.<genexpr>N)anydictsvalues)r   )r   )r   r   r   "   s   z!test_structure.<locals>.<genexpr>)r   r   r   r   callr   r   r   )r   r   test_structure   s    


r#   c              C   sj   t  } | jtdd | jtdd | jtdd | jdddgksDt| jtdd | jdddgksftd S )Nr   )keyr   r!   )r   update_with_keyr   r   r!   orderr   )r   r   r   r   test_structure_2&   s    r'   c              C   s   t  } | jtdd | jtdd | jtdd tttt}x&tttgD ]}|| ||ksHt	qHW x:dd dd dd gD ] }t|| t||ks|t	q|W d S )	Nr   )r$   r   r!   c             S   s   |   S )N)r    )r   r   r   r   <lambda>?   s    z!test_keys_items.<locals>.<lambda>c             S   s   |   S )N)keys)r   r   r   r   r(   ?   s    c             S   s   |   S )N)items)r   r   r   r   r(   ?   s    )
r   r%   r   r   r!   r   dictsetlenr   )r   r   fnr   r   r   test_keys_items4   s    r/   c              C   s   t  } | jtdd | jtdd | jtdd ddi}t  }|jtdd |j|dd | | | jd | jd ks|td S )Nr   )r$   r   r!   r	      r   )r   r%   r   r   r!   r   r   r   )r   r   s2r   r   r   test_update_with_sharedictC   s    
r2   )r   Ztoolzr   Zdask.compatibilityr   Zdask.sharedictr   r   r   r!   r   r#   Zmarkskipr'   r/   r2   r   r   r   r   <module>   s   

