
    IR-e                         d dl Zd dlZd dlmZ d dlmZ erd dlmZ	 d dl
mZmZ ej                            e d           G d d                      ZdS )	    N)fits)HAS_MATPLOTLIB)fits2bitmapmainzrequires matplotlib)reasonc                   ~    e Zd Zd Zd Zd Zd Zd Zej	        
                    dddg          d	             Zd
 ZdS )TestFits2Bitmapc                 l    d| _         t          j        d                              d          | _        d S )N	test.fitsi @  )   r   )filenamenparangereshapearray)selfs    Tlib/python3.11/site-packages/astropy/visualization/scripts/tests/test_fits2bitmap.pysetup_classzTestFits2Bitmap.setup_class   s,    #Yu%%--j99


    c                 l    || j         z  }t          j        || j                   t	          |           d S )N)r   r   writetor   r   r   tmp_pathr   s      r   test_functionzTestFits2Bitmap.test_function   s6    dm+Xtz***Hr   c                     t          || j        z            }t          j        || j                   t          |ddg           d S )N-e0)strr   r   r   r   r   r   s      r   test_scriptzTestFits2Bitmap.test_script   sE    x$-/00Xtz***hc"#####r   c                    t          || j        z            }t          j                    }t          j        | j                  }t          j        ||g          }|                    |           t          |ddg           d S )Nr   1)	r   r   r   
PrimaryHDUImageHDUr   HDUListr   r   )r   r   r   hdu1hdu2hdulists         r   test_exten_numzTestFits2Bitmap.test_exten_num   st    x$-/00  }TZ((,d|,,!!!hc"#####r   c                    t          || j        z            }t          j                    }d}t          j        | j                  }||j        d<   t          j        ||g          }|                    |           t          |d|g           d S )NSCIEXTNAMEr   )
r   r   r   r"   r#   r   headerr$   r   r   )r   r   r   r%   extnamer&   r'   s          r   test_exten_namezTestFits2Bitmap.test_exten_name'   s    x$-/00  }TZ((!(I,d|,,!!!hg&'''''r   
file_extenz.gzz.bz2c                     t          |d| z            }t          j        || j                   t	          |ddg           d S )Nr   r   r   )r   r   r   r   r   )r   r   r/   r   s       r   test_compressed_fitsz$TestFits2Bitmap.test_compressed_fits1   sL    x":j":"::;;Xtz***hc"#####r   c                 L   t          || j        z            }d}t          ||z            }t          j        d          }d|ddddf<   t	          j        ||           t          |ddd	|g           t          j        |          }|d
         dk    sJ |d         dk    sJ dS )zQ
        Regression test to check the image vertical orientation/origin.
        zfits2bitmap_test.png)    r3   g      ?r      Nr   r   z-o)r   r   r   )   r5   r      )	r   r   r   zerosr   r   r   mpimgimread)r   r   r   out_filenamedataimgs         r   test_orientationz TestFits2Bitmap.test_orientation7   s    
 x$-/00-8l233x!!QrT111WXt$$$hc46777l<((7|q    9~""""""r   N)__name__
__module____qualname__r   r   r   r(   r.   pytestmarkparametrizer1   r=    r   r   r	   r	      s        : : :  
$ $ $
$ $ $( ( ( [\E6?;;$ $ <;$
# # # # #r   r	   )numpyr   rA   
astropy.ior   "astropy.utils.compat.optional_depsr   matplotlib.imageimager8   )astropy.visualization.scripts.fits2bitmapr   r   rB   skipifr	   rD   r   r   <module>rL      s               = = = = = = L$$$$$$KKKKKKKK &/DEE6# 6# 6# 6# 6# 6# 6# FE6# 6# 6#r   