
    IR-e                         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	e
          Z G d
 de          ZdS )    N)figure_test)WCSAxes)	BaseFrame)WCS   )BaseImageTestsc                       e Zd ZdZd ZdS )HexagonalFrameabcdefc                 h   | j                                         \  }}| j                                         \  }}d||z   z  }||z   dz  }||z   dz  dz  }t          j        ||g||gf          | d         _        t          j        ||g||gf          | d         _        t          j        ||g||gf          | d         _        t          j        ||g||gf          | d         _        t          j        ||g||gf          | d         _        t          j        ||g||gf          | d	         _        d S )
Ng      ?g      @g      @abcdef)parent_axesget_xlimget_ylimnparraydata)selfxminxmaxyminymaxymidxmid1xmid2s           Nlib/python3.11/site-packages/astropy/visualization/wcsaxes/tests/test_frame.pyupdate_spineszHexagonalFrame.update_spines   s1   %..00
d%..00
ddTk"##c)E4=5$-"@AAS	E4=4,"?@@S	D$<%"?@@S	E4=5$-"@AAS	E4=4,"?@@S	D$<%"?@@S	    N)__name__
__module____qualname__spine_namesr"    r#   r!   r
   r
      s-        KA A A A Ar#   r
   c                   l    e Zd Zed             Zed             Zed             Zed             Zd ZdS )	TestFramec                    t          | j                  }t          j        d          }t	          |g d|t
                    }|                    |           |j                            d           |	                    t          j        d          dd	d
t          j        j                  }i }dx|d<   |d<   dx|d<   x|d<   x|d<   |d<   |j        d                             d|           |j        d                             d           |j        d                             d|           |j        d                             d           |j        d                             d           |j        d                             d           |                    dd           |                    dd           |                    |j        j        j                   |S )N)   r,   )figsize)333333?r.   ffffff?r/   )wcsframe_classwhite)color)   r4                  @lower)vminvmaxorigincmapr   r   r   g      @r   r   r   r   glon	Longitude)minpadadglatLatitudebcefg      @g      Y@g     [@)r   
msx_headerpltfigurer   r
   add_axescoordsgridimshowr   onescm	gist_heatset_axislabelset_axislabel_positionset_ticklabel_positionset_xlimset_ylimset_clip_pathframepatch)r   r0   figaximr>   s         r!   test_custom_framezTestFrame.test_custom_frame"   s   $/""j(((S000c~VVVR
	W%%%YYGJ!  
 
 $%%sfSk@DDsDfSkDF3K&+
	&''F'CCC
	&00666
	&''
6'BBB
	&00888
	&00666
	&00888 	C
C 	.///
r#   c                 ,   t          j                    }t          |g dd          }|                    |           |                    dd           |                    dd           |                    |dz             |                    t          j	        d                     |                    dd	           |                    dd
           |j
        d                             d           |j
        d                             d           |S N皙?r\   皙?r]   equal)aspectr5   r6   nothing   r,               @      '@r   Fr   )rD   rE   r   rF   rP   rQ   savefigrI   r   zerosrG   set_auto_axislabelr   tmp_pathrU   rV   s       r!   !test_update_clip_path_rectangularz+TestFrame.test_update_clip_path_rectangularK   s    jllS...w???R
C
C 	Hy()))
		"(7##$$$
D#
D$
	!''...
	!''...
r#   c                    t          j                    }t          |g ddt                    }|                    |           |                    dd           |                    dd           |                    |dz             |                    t          j
        d                     |                    dd	           |                    dd
           |S )Nr[   r^   )r_   r1   r5   r6   r`   ra   rc   rd   re   )rD   rE   r   r
   rF   rP   rQ   rf   rI   r   rg   ri   s       r!   $test_update_clip_path_nonrectangularz.TestFrame.test_update_clip_path_nonrectangularb   s    jll%%%g>
 
 
 	R
C
C 	Hy()))
		"(7##$$$
D#
D$
r#   c                 T   t          j                    }t          |g dd          }|                    |           |                    dd           |                    dd           |                    |dz             |                                 |                    t          j
        d                     |                    dd	           |                    dd
           |j        d                             d           |j        d                             d           |S rZ   )rD   rE   r   rF   rP   rQ   rf   	reset_wcsrI   r   rg   rG   rh   ri   s       r!    test_update_clip_path_change_wcsz*TestFrame.test_update_clip_path_change_wcsx   s   
 jllS...w???R
C
C 	Hy()))

		"(7##$$$
D#
D$
	!''...
	!''...
r#   c                    t          j                    }t          |g d          }|                    |           |j        j                            d           |j        j                            d           |                                 |j        j        	                                dk    sJ |j        j        
                                dk    sJ d S )Nr[      purple)rD   rE   r   rF   rG   rS   set_linewidth	set_colorro   get_linewidth	get_color)r   rU   rV   s      r!   %test_copy_frame_properties_change_wcsz/TestFrame.test_copy_frame_properties_change_wcs   s     jllS...//R
	%%a(((
	!!(+++
y,,..!3333y((**h666666r#   N)	r$   r%   r&   r   rX   rk   rm   rp   rx   r(   r#   r!   r*   r*   !   s        & & [&P   [,   [*   [67 7 7 7 7r#   r*   )matplotlib.pyplotpyplotrD   numpyr   astropy.tests.figuresr   astropy.visualization.wcsaxesr   #astropy.visualization.wcsaxes.framer   astropy.wcsr   test_imagesr   r
   r*   r(   r#   r!   <module>r      s               - - - - - - 1 1 1 1 1 1 9 9 9 9 9 9       ' ' ' ' ' 'A A A A AY A A A&~7 ~7 ~7 ~7 ~7 ~7 ~7 ~7 ~7 ~7r#   