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)
