a
    ìÝEb4  ã                   @   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   sF  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  }g d¢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   úa/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/mpl_toolkits/tests/test_axes_grid.pyÚtest_imagegrid_cbar_mode_edge   s2    
$4ÿû
r3   c                  C   sZ   t  ¡ } t| ddd�}|d }|jddggtj ¡ d�}|j |¡}t	|j
tjƒsVJ ‚d S )Néo   )é   r5   )r   r   r5   r   )Znorm)r   r!   r   r#   ÚmplÚcolorsZLogNormr'   r(   Ú
isinstanceÚlocatorÚmtickerZ
LogLocator)r+   r.   r/   Zimr0   r1   r1   r2   Útest_imagegrid2   s    r;   )Únumpyr   Z
matplotlibr6   Zmatplotlib.tickerZtickerr:   Zmatplotlib.testing.decoratorsr   Zmatplotlib.pyplotZpyplotr   Zmpl_toolkits.axes_grid1r   r3   r;   r1   r1   r1   r2   Ú<module>   s   ÿ
#