U
    e»{j  ã                   @   st   d dl m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 ddlmZ G dd	„ d	e
ƒZG d
d„ deƒZdS )é    N)Úfigure_test)ÚWCSAxes)Ú	BaseFrame)ÚWCSé   )ÚBaseImageTestsc                   @   s   e Zd ZdZdd„ ZdS )ÚHexagonalFrameZabcdefc                 C   sð   | j  ¡ \}}| j  ¡ \}}d||  }|| d }|| d d }t ||g||gf¡| d _t ||g||gf¡| d _t ||g||gf¡| d _t ||g||gf¡| d _t ||g||gf¡| d _t ||g||gf¡| d	 _d S )
Ng      à?g      @g      @ÚaÚbÚcÚdÚeÚf)Zparent_axesZget_xlimZget_ylimÚnpÚarrayÚdata)ÚselfZxminZxmaxZyminZymaxZymidZxmid1Zxmid2© r   úm/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyÚupdate_spines   s    zHexagonalFrame.update_spinesN)Ú__name__Ú
__module__Ú__qualname__Zspine_namesr   r   r   r   r   r      s   r   c                   @   sD   e Zd Zedd„ ƒZedd„ ƒZedd„ ƒZedd„ ƒZd	d
„ ZdS )Ú	TestFramec                 C   s,  t | jƒ}tjdd�}t|ddddg|td�}| |¡ |jjdd� |j	t
 d¡d	d
dtjjd�}i }d |d< |d< d |d<  |d<  |d< |d< |jd jd|d� |jd  d¡ |jd jd|d� |jd  d¡ |jd  d¡ |jd  d¡ | dd¡ | dd¡ | |jjj¡ |S )N)é   r   )Zfigsizeg333333Ã?gffffffæ?)ÚwcsÚframe_classÚwhite)Úcolor)é•   r   ç        ç       @Úlower)ZvminZvmaxÚoriginZcmapr   r	   r   g      @r
   r   r   r   ZglonZ	Longitude)ÚminpadÚadZglatZLatitudeZbcefg      @g      Y@g      [@)r   Z
msx_headerÚpltÚfigurer   r   Úadd_axesÚcoordsÚgridÚimshowr   ZonesÚcmZ	gist_heatZset_axislabelZset_axislabel_positionZset_ticklabel_positionÚset_xlimÚset_ylimZset_clip_pathÚframeÚpatch)r   r   ÚfigÚaxZimr$   r   r   r   Útest_custom_frame"   s2    

û zTestFrame.test_custom_framec                 C   sš   t  ¡ }t|ddddgdd�}| |¡ | dd¡ | dd¡ | |d ¡ | t 	d¡¡ | d	d
¡ | d	d¡ |j
d  d¡ |j
d  d¡ |S ©Nçš™™™™™¹?çš™™™™™é?Úequal)Úaspectr    r!   Únothing©é   r   ç      à¿ç      @ç      '@r   Fr   )r&   r'   r   r(   r-   r.   Úsavefigr+   r   Úzerosr)   Úset_auto_axislabel©r   Ztmp_pathr1   r2   r   r   r   Ú!test_update_clip_path_rectangularK   s    
z+TestFrame.test_update_clip_path_rectangularc                 C   s|   t  ¡ }t|ddddgdtd�}| |¡ | dd¡ | dd¡ | |d ¡ | t	 
d¡¡ | d	d
¡ | d	d¡ |S )Nr5   r6   r7   )r8   r   r    r!   r9   r:   r<   r=   r>   )r&   r'   r   r   r(   r-   r.   r?   r+   r   r@   rB   r   r   r   Ú$test_update_clip_path_nonrectangularb   s     
  ÿ
z.TestFrame.test_update_clip_path_nonrectangularc                 C   s¢   t  ¡ }t|ddddgdd�}| |¡ | dd¡ | dd¡ | |d ¡ | ¡  | t	 
d¡¡ | d	d
¡ | d	d¡ |jd  d¡ |jd  d¡ |S r4   )r&   r'   r   r(   r-   r.   r?   Ú	reset_wcsr+   r   r@   r)   rA   rB   r   r   r   Ú test_update_clip_path_change_wcsx   s    
z*TestFrame.test_update_clip_path_change_wcsc                 C   st   t  ¡ }t|ddddgƒ}| |¡ |jj d¡ |jj d¡ | ¡  |jj 	¡ dks\t
‚|jj ¡ dkspt
‚d S )Nr5   r6   é   Úpurple)r&   r'   r   r(   r)   r/   Zset_linewidthZ	set_colorrE   Zget_linewidthÚAssertionErrorÚ	get_color)r   r1   r2   r   r   r   Ú%test_copy_frame_properties_change_wcs”   s    
z/TestFrame.test_copy_frame_properties_change_wcsN)	r   r   r   r   r3   rC   rD   rF   rK   r   r   r   r   r   !   s   
(


r   )Zmatplotlib.pyplotZpyplotr&   Únumpyr   Zastropy.tests.figuresr   Zastropy.visualization.wcsaxesr   Z#astropy.visualization.wcsaxes.framer   Zastropy.wcsr   Ztest_imagesr   r   r   r   r   r   r   Ú<module>   s   