U d{j @sddlmZddlZddlmZmZmZmZddlm Z ddl m Z ddZ d(d d Z ejde dddddZejde d ddddZejde d dddddZejde dddddZejde ddddd d!Zejjd"d#d$d%Zejjd"d#d&d'ZdS)))mockN) BinTableHDUHDUListHeader PrimaryHDU)kepler_fits_reader)get_pkg_data_filenamec Cstddd||||dS)NTr)SIMPLEZBITPIXZNAXISZEXTVERVERSIONZTIMESYSZTELESCOP)r)extverversiontimesystelescoprf/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/timeseries/io/tests/test_kepler.py fake_header srTDBKEPLERcCs,t||||}tt|dt|ddgdgS)N)headerZ LIGHTCURVE)rname)Zhdus)rrrr)r rrrZ new_headerrrr fake_hdulists rzastropy.io.fits.openZMadeUp)r)Z side_effectc Cs&tjtddtdW5QRXdS)NzPMadeUp is not implemented, only KEPLER or TESS are supported through this readermatchpytestZraisesNotImplementedErrorrZ mock_filerrrtest_raise_telescop_wrong&s r!)r c Cs&tjtddtdW5QRXdS)Nz/Support for KEPLER v2 files not yet implementedrrr rrrtest_raise_extversion_kepler2s r"ZTESS)r rc Cs&tjtddtdW5QRXdS)Nz-Support for TESS v2 files not yet implementedrrr rrrtest_raise_extversion_tess:s r#ZTCB)rc Cs&tjtddtdW5QRXdS)Nz?Support for TCB time scale not yet implemented in KEPLER readerrrr rrrtest_raise_timesys_keplerBs r$)rrc Cs&tjtddtdW5QRXdS)Nz=Support for TCB time scale not yet implemented in TESS readerrrr rrrtest_raise_timesys_tessKs r%Zastropy)sourcec Csddlm}td}t|t|}W5QRX|djdksDt|djdksVt|dj dksltt |d ks|tt |j d kstdS) Nr) UnitsWarningz/timeseries/kplr010666592-2009131110544_slc.fitstimeisottdbsap_flux electron / si7) Z astropy.unitsr'rrwarnsrformatAssertionErrorscaleunit to_stringlencolumns)r'filename timeseriesrrrtest_kepler_astropyVs  r8c Cstd}tjtddt|}W5QRX|djdkss.