def test_from_timeseries_bad(self): from astropy.time import TimeDelta times = TimeDelta(np.arange(10) * u.s) ts = TimeSeries(time=times) with pytest.raises(ValueError) as excinfo: Lightcurve.from_astropy_timeseries(ts) assert "Input timeseries must contain at least" in str(excinfo.value)
def test_timeseries_roundtrip(self): """Test that io methods raise Key Error when wrong format is provided. """ N = len(self.times) lc = Lightcurve(self.times, self.counts, mission="BUBU", instr="BABA", mjdref=53467.) ts = lc.to_astropy_timeseries() new_lc = lc.from_astropy_timeseries(ts) for attr in ['time', 'gti', 'counts']: assert np.allclose(getattr(lc, attr), getattr(new_lc, attr)) for attr in ['mission', 'instr', 'mjdref']: assert getattr(lc, attr) == getattr(new_lc, attr)
def test_timeseries_roundtrip_ctrate(self): """Test that io methods raise Key Error when wrong format is provided. """ N = len(self.times) dt = 0.5 mean_counts = 2.0 times = np.arange(0 + dt / 2, 5 - dt / 2, dt) countrate = np.zeros_like(times) + mean_counts lc = Lightcurve(times, countrate, mission="BUBU", instr="BABA", mjdref=53467., input_counts=False) ts = lc.to_astropy_timeseries() new_lc = lc.from_astropy_timeseries(ts) for attr in ['time', 'gti', 'countrate']: assert np.allclose(getattr(lc, attr), getattr(new_lc, attr)) assert np.allclose(new_lc.counts, lc.countrate * lc.dt) for attr in ['mission', 'instr', 'mjdref']: assert getattr(lc, attr) == getattr(new_lc, attr)