def test_get_lightcurve_dataset_from_stingray_Lightcurve(capsys): from stingray.lightcurve import Lightcurve from astropy.io.fits import Header lc = Lightcurve([0, 1], [2, 2]) ds = get_lightcurve_dataset_from_stingray_Lightcurve(lc) out, err = capsys.readouterr() assert err.strip().endswith("Light curve has no header") header = Header() header["Bu"] = "Bu" lc.header = header.tostring() ds = get_lightcurve_dataset_from_stingray_Lightcurve(lc) assert np.allclose(ds.tables["RATE"].columns["TIME"].values, lc.time) assert np.allclose(ds.tables["RATE"].columns["RATE"].values, lc.counts)
def test_get_lightcurve_dataset_from_stingray_Lightcurve(capsys): from stingray.lightcurve import Lightcurve from astropy.io.fits import Header lc = Lightcurve([0, 1], [2, 2]) ds = get_lightcurve_dataset_from_stingray_Lightcurve(lc) out, err = capsys.readouterr() if err: assert err.strip().endswith("Light curve has no header") header = Header() header["Bu"] = "Bu" lc.header = header.tostring() ds = get_lightcurve_dataset_from_stingray_Lightcurve(lc) assert np.allclose(ds.tables["RATE"].columns["TIME"].values, lc.time) assert np.allclose(ds.tables["RATE"].columns["RATE"].values, lc.counts)
def test_get_eventlist_dataset_from_stingray_Eventlist(capsys): from stingray.events import EventList from astropy.io.fits import Header ev = EventList(time=[0, 1], pi=[2, 2], energy=[3., 4.], gti=np.array([[-0.5, 1.5]])) ds = get_eventlist_dataset_from_stingray_Eventlist(ev) out, err = capsys.readouterr() print("Out:", out) print("Err:", err) assert "Event list has no header" in err header = Header() header["Bu"] = "Bu" ev.header = header.tostring() ds = get_eventlist_dataset_from_stingray_Eventlist(ev) assert np.allclose(ds.tables["EVENTS"].columns["TIME"].values, ev.time) assert np.allclose(ds.tables["EVENTS"].columns["ENERGY"].values, ev.energy) assert np.allclose(ds.tables["EVENTS"].columns["PI"].values, ev.pi)
def test_get_eventlist_dataset_from_stingray_Eventlist(capsys): from stingray.events import EventList from astropy.io.fits import Header ev = EventList(time=[0, 1], pi=[2, 2], energy=[3., 4.], gti=np.array([[-0.5, 1.5]])) ds = get_eventlist_dataset_from_stingray_Eventlist(ev) out, err = capsys.readouterr() print("Out:", out) print("Err:", err) if err: assert "Event list has no header" in err header = Header() header["Bu"] = "Bu" ev.header = header.tostring() ds = get_eventlist_dataset_from_stingray_Eventlist(ev) assert np.allclose(ds.tables["EVENTS"].columns["TIME"].values, ev.time) if "ENERGY" in ds.tables["EVENTS"].columns: assert np.allclose(ds.tables["EVENTS"].columns["ENERGY"].values, ev.energy) assert np.allclose(ds.tables["EVENTS"].columns["PI"].values, ev.pi)