def test_kepler_astropy(): filename = get_pkg_data_filename('timeseries/kplr010666592-2009131110544_slc.fits') timeseries = TimeSeries.read(filename, format='kepler.fits') assert timeseries["time"].format == 'isot' assert timeseries["time"].scale == 'tdb' assert timeseries["sap_flux"].unit.to_string() == 'electron / s' assert len(timeseries) == 14280 assert len(timeseries.columns) == 20
def test_tess_astropy(): filename = get_pkg_data_filename('timeseries/hlsp_tess-data-alerts_tess_phot_00025155310-s01_tess_v1_lc.fits') with pytest.warns(UserWarning, match='Ignoring 815 rows with NaN times'): timeseries = TimeSeries.read(filename, format='tess.fits') assert timeseries["time"].format == 'isot' assert timeseries["time"].scale == 'tdb' assert timeseries["sap_flux"].unit.to_string() == 'electron / s' assert len(timeseries) == 19261 assert len(timeseries.columns) == 20
def test_read(): timeseries = TimeSeries.read(CSV_FILE, time_column='Date', format='csv') assert timeseries.colnames == ['time', 'A', 'B', 'C', 'D', 'E', 'F', 'G'] assert len(timeseries) == 11 assert timeseries['time'].format == 'iso' assert timeseries['A'].sum() == 266.5
def test_read_time_wrong(): with pytest.raises(ValueError) as exc: TimeSeries.read(CSV_FILE, time_column='abc', format='csv') assert exc.value.args[0] == "Time column 'abc' not found in the input data."
def test_read_time_missing(): with pytest.raises(ValueError) as exc: TimeSeries.read(CSV_FILE, format='csv') assert exc.value.args[0] == '``time_column`` should be provided since the default Table readers are being used.'