B
    OK\I                 @   s   d dl mZmZmZ d dlZd dlmZ d dlmZ yd dlm	Z	 W n e
k
r\   dZ	Y nX ejdedd feje	d	d ejje	 d
ddgdd Zdd ZdS )    )print_functiondivisionabsolute_importN)Counter)loop)DigestzCD,sizec             C   s   t |  S )N)sumvalues)d r   =lib/python3.7/site-packages/distributed/tests/test_counter.py<lambda>   s    r   c             C   s   |   S )N)size)xr   r   r   r      s    zno crick library)reason)Zmarksc                s  || d}| d | d  |jd dks4t|  d |jd   k rZdk s`n td |jd   k r~dk sn tt fdd|jD dkstx>tt|jd D ](} |j|  |j|d  kstqW | d t fdd|jD | kstd S )	N)r         r   c             3   s   | ]} |V  qd S )Nr   ).0r
   )r   r   r   	<genexpr>   s    ztest_digest.<locals>.<genexpr>   c             3   s   | ]} |V  qd S )Nr   )r   r
   )r   r   r   r   #   s    )add
componentsAssertionErrorshiftr   rangelenr   )r   ZCDr   cir   )r   r   test_digest   s    


$$ (
r   c             C   sV   t | d}|d x<tdD ]0}|  ttdd |jD d dk stqW d S )N)r   r      c             s   s   | ]}|d  V  qdS )r   Nr   )r   Zccr   r   r   r   ,   s    ztest_counter.<locals>.<genexpr>gvIh%<=)r   r   r   r   absr   r   r   )r   r   r   r   r   r   test_counter&   s
    

r!   )Z
__future__r   r   r   ZpytestZdistributed.counterr   Zdistributed.utils_testr   r   ImportErrorZmarkZparametrizeZparamZskipifr   r!   r   r   r   r   <module>   s   

"