a
    ߙfbY                     @   s@   d dl Z ddlmZ d dlmZ d dlmZ G dd deZdS )    N   )FitsTestCase)fitsinfo)__version__c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestFitsinfoc                 C   sJ   t t}tdg W d    n1 s,0    Y  |jjdksFJ d S )Nz-hr   )pytestraises
SystemExitr   mainvaluecode)selfe r   Blib/python3.9/site-packages/astropy/io/fits/tests/test_fitsinfo.py	test_help   s    *zTestFitsinfo.test_helpc                 C   sh   t t:}tdg | d }|dt ks6J W d    n1 sJ0    Y  |jjdksdJ d S )Nz	--versionr   z	fitsinfo )	r   r   r	   r   r
   
readouterrversionr   r   )r   capsysr   outr   r   r   test_version   s
    0zTestFitsinfo.test_versionc                 C   s^   t | dg | \}}| }t|dks6J |d dsHJ |d dsZJ d S )Narange.fits   r   =No.    Name      Ver    Type      Cards   Dimensions   Format   =  0  PRIMARY       1 PrimaryHDU       7   (11, 10, 7)   int32r   r
   datar   
splitlineslen
startswithr   r   r   errr   r   r   test_onefile   s    zTestFitsinfo.test_onefilec                 C   s   t | d| dg | \}}| }t|dks>J |d dsPJ |d dsbJ |d d	ksrJ |d
 dsJ d S )Nr   z
ascii.fits   r   r   r   r   r       z?  1                1 TableHDU        20   5R x 2C   [E10.4, I5]r   r!   r   r   r   test_multiplefiles"   s     zTestFitsinfo.test_multiplefilesN)__name__
__module____qualname__r   r   r#   r'   r   r   r   r   r   
   s   
r   )	r   r%   r   Zastropy.io.fits.scriptsr   Zastropyr   r   r   r   r   r   r   <module>   s   