U
    d»{j4  ã                   @   sf   d dl 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dgdddd�d	d
„ ƒZdd„ ZdS )é    N)Úimage_comparison)Ú	ImageGridzimagegrid_cbar_mode.pngTZmpl20g333333Ó?)Zremove_textÚstyleZtolc               
   C   sJ  dt jd< t t ddd¡t ddd¡¡\} }t | ¡t |¡ dt d| ¡t |d ¡   }t jd	d
�}d}dgd dgd  }ddddgd }t|||ƒD ]¢\}}}	t	||d||	ddd�}
|
\}}}}|j
|jdd� |j
|jdd� |j
t |¡dd� |j
t |j|j¡dd� |
D ]"}|j ¡  |j |jd ¡}�q q¢d S )NFzpcolormesh.snapr   é   é   y              ð?é   g       @)é   é	   )Zfigsize)éñ   éò   éó   éô   éõ   éö   é÷   éø   Úrowé   ÚcolumnÚleftÚrightÚtopÚbottomé   )r   r   z20%Úedge)Únrows_ncolsÚ	directionZcbar_locationZ	cbar_sizeZ	cbar_modeZnipy_spectral)ZcmapZhotZjetZhsv)ÚpltZrcParamsÚnpZmeshgridZlinspaceÚsinÚcosÚfigureÚzipr   ÚimshowÚrealÚimagÚabsZarctan2ÚcaxZclaÚcolorbarZimages)ÚXÚYZarrÚfigZ	positionsZ
directionsZcbar_locationsÚpositionr   ÚlocationÚgridZax1Zax2Zax3Zax4ÚaxÚcb© r1   ú`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mpl_toolkits/tests/test_axes_grid.pyÚtest_imagegrid_cbar_mode_edge   s6    
$4  ÿû
r3   c                  C   sZ   t  ¡ } t| ddd�}|d }|jddggtj ¡ d�}|j |¡}t	|j
tjƒsVt‚d S )Néo   )é   r5   )r   r   r5   r   )Znorm)r   r!   r   r#   ÚmplÚcolorsZLogNormr'   r(   Ú
isinstanceZlocatorÚmtickerZ
LogLocatorÚAssertionError)r+   r.   r/   Zimr0   r1   r1   r2   Útest_imagegrid2   s    r;   )Únumpyr   Z
matplotlibr6   Zmatplotlib.tickerZtickerr9   Zmatplotlib.testing.decoratorsr   Zmatplotlib.pyplotZpyplotr   Zmpl_toolkits.axes_grid1r   r3   r;   r1   r1   r1   r2   Ú<module>   s     ÿ
#