
    IR-e                     ^   d dl mZ d dlZd dlmZmZmZmZ d dlm	Z	 d dl
mZ d ZddZ ej        d ed                    d             Z ej        d ed                    d             Z ej        d edd                    d             Z ej        d ed                    d             Z ej        d edd                    d             Zej                            d          d             Zej                            d          d             ZdS )     )mockN)BinTableHDUHDUListHeader
PrimaryHDU)kepler_fits_reader)get_pkg_data_filenamec           
      4    t          ddd| || | d          S )NT   r   )SIMPLEBITPIXNAXISEXTVERVERSIONTIMESYSTELESCOP)r   )extverversiontimesystelescops       Glib/python3.11/site-packages/astropy/timeseries/io/tests/test_kepler.pyfake_headerr      s7    !|#	
 	

 
 
          TDBKEPLERc                     t          | |||          }t          t          |          t          |d          g          gS )N)header
LIGHTCURVE)r    name)hdus)r   r   r   r   )r   r   r   r   
new_headers        r   fake_hdulistr%      sT    VWgx@@J*---:LAAA	
 	
 	
 r   zastropy.io.fits.openMadeUp)r   )side_effectc                     t          j        t          d          5  t          d            d d d            d S # 1 swxY w Y   d S )NzPMadeUp is not implemented, only KEPLER or TESS are supported through this readermatchpytestraisesNotImplementedErrorr   	mock_files    r   test_raise_telescop_wrongr1   &   s    	
 
 
 ! ! 	4   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !   9= =)r   c                     t          j        t          d          5  t          d            d d d            d S # 1 swxY w Y   d S )Nz/Support for KEPLER v2 files not yet implementedr)   r+   r/   s    r   test_raise_extversion_keplerr4   2   s    	#U
 
 
 ! ! 	4   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !r2   TESS)r   r   c                     t          j        t          d          5  t          d            d d d            d S # 1 swxY w Y   d S )Nz-Support for TESS v2 files not yet implementedr)   r+   r/   s    r   test_raise_extversion_tessr7   :   s    	#S
 
 
 ! ! 	4   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !r2   TCB)r   c                     t          j        t          d          5  t          d            d d d            d S # 1 swxY w Y   d S )Nz?Support for TCB time scale not yet implemented in KEPLER readerr)   r+   r/   s    r   test_raise_timesys_keplerr:   B   s    	P
 
 
 ! ! 	4   	! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !r2   )r   r   c                     t          j        t          d          5  t          d            d d d            d S # 1 swxY w Y   d S )Nz=Support for TCB time scale not yet implemented in TESS readerr)   r+   r/   s    r   test_raise_timesys_tessr<   K   s     
N
 
 
 ! ! 	4   	! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !r2   astropy)sourcec                     ddl m}  t          d          }t          j        |           5  t          |          }d d d            n# 1 swxY w Y   |d         j        dk    sJ |d         j        dk    sJ |d         j        	                                dk    sJ t          |          d	k    sJ t          |j                  d
k    sJ d S )Nr   )UnitsWarningz/timeseries/kplr010666592-2009131110544_slc.fitstimeisottdbsap_fluxelectron / si7     )astropy.unitsr@   r	   r,   warnsr   formatscaleunit	to_stringlencolumns)r@   filename
timeseriess      r   test_kepler_astropyrQ   V   s&   ******$%VWWH	l	#	# 2 2'11
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 f$....f#u,,,,j!&0022nDDDDz??e####z!""b((((((s   AA
A
c                     t          d          } t          j        t          d          5  t	          |           }d d d            n# 1 swxY w Y   |d         j        dk    sJ |d         j        dk    sJ |d         j                                        dk    sJ t          |          d	k    sJ t          |j
                  d
k    sJ d S )NzJtimeseries/hlsp_tess-data-alerts_tess_phot_00025155310-s01_tess_v1_lc.fitsz Ignoring 815 rows with NaN timesr)   rA   rB   rC   rD   rE   i=K  rF   )r	   r,   rH   UserWarningr   rI   rJ   rK   rL   rM   rN   )rO   rP   s     r   test_tess_astropyrT   f   s   $T H 
k)K	L	L	L 2 2'11
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2f$....f#u,,,,j!&0022nDDDDz??e####z!""b((((((s   AAA)r   r   r   r   )unittestr   r,   astropy.io.fitsr   r   r   r   astropy.timeseries.io.keplerr   astropy.utils.datar	   r   r%   patchr1   r4   r7   r:   r<   markremote_datarQ   rT    r   r   <module>r]      s-          D D D D D D D D D D D D ; ; ; ; ; ; 4 4 4 4 4 4  	 	 	 	 "h0O0O0OPPP! ! QP! "A0F0F0FGGG! ! HG! "APV0W0W0WXXX! ! YX! "U0K0K0KLLL! ! ML! UV(T(T(T  ! ! ! 	**) ) +*) 	**
) 
) +*
) 
) 
)r   