o
    tBh4                     @   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.pngTmpl20g333333?)remove_textstyletolc               
   C   sD  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 ]P\}}}	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 }qqOd S )NFzpcolormesh.snapr         y              ?   g       @)   	   )figsize)                        row   column)leftrighttopbottom   )r   r   z20%edge)nrows_ncols	directioncbar_location	cbar_size	cbar_modenipy_spectral)cmaphotjethsv)pltrcParamsnpmeshgridlinspacesincosfigurezipr   imshowrealimagabsarctan2caxclacolorbarimages)XYarrfig	positions
directionscbar_locationspositionr    locationgridax1ax2ax3ax4axcb rK   x/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mpl_toolkits/tests/test_axes_grid.pytest_imagegrid_cbar_mode_edge   s6   
$4
rM   c                  C   sZ   t  } t| ddd}|d }|jddggtj d}|j|}t	|j
tjs+J d S )No   )   rO   )r   r   rO   r   )norm)r)   r0   r   r2   mplcolorsLogNormr7   r9   
isinstancelocatormticker
LogLocator)r>   rD   rI   imrJ   rK   rK   rL   test_imagegrid2   s   rY   )numpyr+   
matplotlibrQ   matplotlib.tickertickerrV   matplotlib.testing.decoratorsr   matplotlib.pyplotpyplotr)   mpl_toolkits.axes_grid1r   rM   rY   rK   rK   rK   rL   <module>   s    
#