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