U
    øÕ/e
  ã                   @   sl   d dl Zd dlZd dlmZ d dlmZ dd„ Zdd„ Z	dd„ Z
d	d
„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )é    N)Ú
draw_sizesc                 C   s&   |   dd¡}tjj |¡d k	s"t‚d S )Nz&rarr;Ú )ÚreplaceÚxmlÚetreeÚElementTreeÚ
fromstringÚAssertionError)ÚtextÚcleaned© r   ú=/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_svg.pyÚparses	   s    r   c                   C   s‚   t t d¡ ¡ ƒ t t d¡ ¡ ƒ t t d¡ ¡ ƒ t t d¡ ¡ ƒ t t d¡ ¡ ƒ t t d¡ ¡ ƒ t t d¡ ¡ ƒ d S )Né
   ©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   r   r   )r   ÚdaÚonesÚto_svgr   r   r   r   Ú
test_basic   s    r   c                   C   s‚   t  d¡ t g ¡ ¡ st‚t d¡d d…  ¡ s6t‚t d¡ ¡ sHt‚t d¡ ¡ sZt‚t d¡ ¡ slt‚t d¡ ¡ s~t‚d S )NÚjinja2r   r   r   r   r   )ÚpytestÚimportorskipr   r   Ú_repr_html_r	   r   r   r   r   Útest_repr_html   s    
r   c               	   C   sô   t  t¡} t g ¡ ¡  W 5 Q R X dt| jƒks6t‚t  t¡} t 	d¡ ¡  W 5 Q R X dt| jƒkslt‚t  t¡} t d¡d d…  ¡  W 5 Q R X dt| jƒksªt‚t  t¡$} t d¡}||dk }| ¡  W 5 Q R X dt| jƒksðt‚d S )Nz0 dimensionsé   r   r   z0-length dimensionsé   zunknown chunk sizes)
r   ZraisesÚNotImplementedErrorr   r   r   ÚstrÚvaluer	   Zasarray)ZexcptÚxr   r   r   Útest_errors"   s     
r#   c                  C   sˆ   t  d¡ t d¡} tjddd} |  ¡ }d|ks>d|ks>t‚t| jƒ|ksPt‚t| jƒ|ksbt‚t	|ƒ tjdd	d} t	|  ¡ ƒ d S )
Nr   )é'  iˆ  )é¸  r$   )éè  r&   ©ÚchunksZMBZMiB)r%   r$   é2   )r&   r&   r   )
r   r   r   r   r   r	   r    ÚshapeZdtyper   )r"   r
   r   r   r   Útest_repr_html_size_units:   s    

r+   c                  C   s†   t ddddkst‚t ddddks(t‚t ddddks<t‚t d	ƒ\} }}| |ksVt‚||ksbt‚| |d
 k srt‚||d
 k s‚t‚d S )Nr   éd   )Úsize)r,   r,   éÈ   )r.   r.   )r   r   )r,   r)   )r&   r,   r   r   )r   r	   )ÚaÚbÚcr   r   r   Útest_draw_sizesJ   s    r2   c                  C   s8   t jddd} |  ¡ }d|ks"t‚| d¡dk s4t‚d S )N)i€>  é`	  é  )r   r3   r4   r'   Z8B4903Ú
i,  )r   r   r   r	   Úcount)Údatar
   r   r   r   Ú&test_too_many_lines_fills_sides_darkerV   s    r8   c                  C   s$   t  d¡ ¡ } |  d¡dks t‚d S )Nr   z<svgr   )r   r   r   r6   r	   )r
   r   r   r   Útest_3d]   s    r9   )Úxml.etree.ElementTreer   r   Z
dask.arrayÚarrayr   Zdask.array.svgr   r   r   r   r#   r+   r2   r8   r9   r   r   r   r   Ú<module>   s   

