
    IR-e                     x   d dl Z d dlZd dlmZ er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 Zej                            e d          d	             Zej                            e d          d
             Zej                            e d          d             Zej                            e d          d             Zej                            e d          d             Zej                            e d          d             Zej                            e d          d             ZdS )    N)HAS_PLT)units)Angle)quantity_supportc                 .    t          j        d           d S )Nall)pltclose)functions    Flib/python3.11/site-packages/astropy/visualization/tests/test_units.pyteardown_functionr      s    Ie    zrequires matplotlib.pyplot)reasonc                  2   t          j                     t                      5  t          j                    } t          j        g dt          j        z  g dt          j        z  d           t          j        g dt          j	        z  g dt          j
        z             t          j                     t          j        ddgt          j        z  dd	gt          j        z  d
dgt          j
        z             t          j        | d           t          j                    j                                        t          j        k    sJ t          j                    j                                        t          j        k    sJ 	 d d d            d S # 1 swxY w Y   d S )N         r         label)r   i      i;    i    r   r   r   r   r   svg)format)r	   figurer   ioBytesIOplotumkgcmglegendfill_betweensavefiggcaxaxis	get_unitsyaxis)buffs    r   
test_unitsr2      s   JLLL			 3 3z||QS)))ad"2'BBBB14');););ac)ABBB
 	!Q!#1v}tTlQS6HIIID''''wyy((**ac1111wyy((**ad222223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   EFFFc                  
   t          j        d           t          j                     t	                      5  g dt
          j        z  } g dt
          j        z  }g dt
          j        z  }t          j	                    \  }}|
                    | ||           |j                                        t
          j        k    sJ |j                                        t
          j        k    sJ 	 d d d            d S # 1 swxY w Y   d S )N
matplotlibr   )r   r   r   )yerr)pytestimportorskipr	   r!   r   r%   sr&   r(   subplotserrorbarr.   r/   r0   )xyr5   figaxs        r   test_units_errbarrr?   *   s(   
%%%JLLL			 	+ 	+IIOIIOyy14,..R
Aqt$$$x!!##qs****x!!##qs*****	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+s   B4C88C<?C<c                     	 ddl m}  | }n# t          $ r t          j        }Y nw xY wt          j                     t                      5  t          j        g dt          j	        z             t          j        |          5  t          j        g dt          j        z             d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )Nr   )ConversionErrorr   r   )matplotlib.unitsrA   ImportErrorr%   UnitConversionErrorr	   r!   r   r$   r&   r6   raisesr'   )rA   err_types     r   test_incompatible_unitsrG   ;   sb   #444444 #  ) ) )()
 JLLL			 - -QS!!!]8$$ 	- 	-H___qt+,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	-- - - - - - - - - - - - - - - - - -sB    $$8C $B0$C0B4	4C7B4	8CCCc                     t          j                     t                      5  t          j        t	          g dt
          j                  g dt
          j        z             t          j        g dt
          j        z  g dt
          j	        z             t          j
        t	          g dt
          j                  g dt
          j	        z             t          j                    j                                        t
          j        k    sJ t          j                    j                                        t
          j        k    sJ 	 ddd           dS # 1 swxY w Y   dS )zlCheck that subclasses are recognized.

    Also see https://github.com/matplotlib/matplotlib/pull/13536
    r   r   r   r   N)r	   r!   r   scatterr   r%   degr'   arcsecr)   r$   r-   r.   r/   r0    r   r   test_quantity_subclassrM   N   sX    JLLL			 3 3E)))QU++YYY-=>>>OOOah.0B0B0BQS0HIII113E3E3E3KLLLwyy((**ae3333wyy((**ad222223 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s   D$EEEc                     t                      5  t                      5  t          j                    } |                     ddd          }|                    t          g dt          j                  g dt          j        z             |j	        
                                t          j        k    sJ |j        
                                t          j        k    sJ 	 d d d            n# 1 swxY w Y   t          j                    } |                     ddd          }|                    t          g dt          j                  g dt          j        z             |j	        
                                t          j        k    sJ |j        
                                t          j        k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r   r   )r   r	   r!   add_subplotrI   r   r%   rJ   r'   r.   r/   r0   rK   pcr=   r>   s     r   test_nestedrR   _   s   			 , , 	0 	0*,,CAq))BJJuYYY..			AD0@AAA8%%''1500008%%''14/////	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 jll__Q1%%


5AH--yyy14/?@@@x!!##qx////x!!##qt+++++, , , , , , , , , , , , , , , , , ,s6   F;B<C'F;'C+	+F;.C+	/B>F;;F?F?c                  <   t                      5  t          j                    } |                     ddd          }|                    g dt
          j        z  d           |                    g t
          j        z  d           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   r   r   d   )bins)r   r	   r!   rO   histr%   mmagrQ   s     r   test_empty_histrX   r   s    			 ' 'jll__Q1%%
qv%C000 	QV#&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's   A5BBBc                  v   t                      5  t          j                    \  } }|                    g dg dt          j        z  t          j        z             | j        	                                 d |j
                                        D             }|g dk    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   c                 6    g | ]}|                                 S rL   )get_text).0tls     r   
<listcomp>z)test_radian_formatter.<locals>.<listcomp>   s     DDDB"++--DDDr   )u   π/2u   πu   3π/2u   2πu   5π/2u   3πu   7π/2)r   r	   r9   r$   r%   radnppicanvasdrawr0   get_ticklabels)r=   r>   labelss      r   test_radian_formatterrf   }   s   			 Q Q,..R
			999qu,ru4555
DD"(*A*A*C*CDDDPPPPPPPPQ Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Qs   BB..B25B2)r"   r6   "astropy.utils.compat.optional_depsr   matplotlib.pyplotpyplotr	   numpyr`   astropyr   r%   astropy.coordinatesr   astropy.visualization.unitsr   r   markskipifr2   r?   rG   rM   rR   rX   rf   rL   r   r   <module>rp      s   
			  6 6 6 6 6 6
 $######           % % % % % % 8 8 8 8 8 8   K(DEE3 3 FE3$ K(DEE+ + FE+  K(DEE- - FE-$ K(DEE3 3 FE3  K(DEE, , FE,$ K(DEE' ' FE' K(DEEQ Q FEQ Q Qr   