def test_load_and_save_lcurve(self):
lcurve = Lightcurve(np.linspace(0, 10, 15), np.random.poisson(30, 15),
mjdref=54385.3254923845,
gti = np.longdouble([[-0.5, 3.5]]))
save_to_intermediate_file(lcurve, self.dum)
ds = load_dataset_from_intermediate_file(self.dum)
assert ds
def test_load_and_save_events(self):
events = EventList([0, 2, 3.], pi=[1, 2, 3], mjdref=54385.3254923845,
gti = np.longdouble([[-0.5, 3.5]]))
events.energy = np.array([3., 4., 5.])
save_to_intermediate_file(events, self.dum)
ds = load_dataset_from_intermediate_file(self.dum)
assert ds
def test_load_and_save_lcurve(self):
lcurve = Lightcurve(np.linspace(0, 10, 15),
np.random.poisson(30, 15),
mjdref=54385.3254923845,
gti=np.longdouble([[-0.5, 3.5]]))
save_to_intermediate_file(lcurve, self.dum)
lcurve2 = load_from_intermediate_file(self.dum)
assert np.allclose(lcurve.time, lcurve2.time)
assert np.allclose(lcurve.counts, lcurve2.counts)
assert np.allclose(lcurve.mjdref, lcurve2.mjdref)
assert np.allclose(lcurve.gti, lcurve2.gti)
assert lcurve.err_dist == lcurve2.err_dist
def test_load_and_save_pds(self):
pds = Powerspectrum()
pds.freq = np.linspace(0, 10, 15)
pds.power = np.random.poisson(30, 15)
pds.mjdref = 54385.3254923845
pds.gti = np.longdouble([[-0.5, 3.5]])
save_to_intermediate_file(pds, self.dum)
pds2 = load_from_intermediate_file(self.dum)
assert np.allclose(pds.gti, pds2.gti)
assert np.allclose(pds.mjdref, pds2.mjdref)
assert np.allclose(pds.gti, pds2.gti)
assert pds.m == pds2.m
def test_load_and_save_events(self):
events = EventList([0, 2, 3.],
pi=[1, 2, 3],
mjdref=54385.3254923845,
gti=np.longdouble([[-0.5, 3.5]]))
events.energy = np.array([3., 4., 5.])
save_to_intermediate_file(events, self.dum)
events2 = load_from_intermediate_file(self.dum)
assert np.allclose(events.time, events2.time)
assert np.allclose(events.pi, events2.pi)
assert np.allclose(events.mjdref, events2.mjdref)
assert np.allclose(events.gti, events2.gti)
assert np.allclose(events.energy, events2.energy)
def get_intermediate_file(filepath, target):
try:
stingray_object = DaveReader.get_stingray_object(filepath)
if stingray_object:
filename = FileUtils.get_intermediate_filename(
target, filepath, HEN_FILE_EXTENSION)
if DaveReader.save_to_intermediate_file(stingray_object, filename):
return filename
except:
logging.error(ExHelper.getException('get_intermediate_file'))
return None