def test_pickle(self, tmpdir):
fn = str(tmpdir.join("{}.pickle".format(self.name)))
with open(fn, "wb") as f:
pickle.dump(self.potential, f)
with open(fn, "rb") as f:
p = pickle.load(f)
p.value(self.w0[:self.w0.size//2])
def test_pickle(self, tmpdir):
fn = str(tmpdir.join("{}.pickle".format(self.name)))
with open(fn, "wb") as f:
pickle.dump(self.potential, f)
with open(fn, "rb") as f:
p = pickle.load(f)
p.energy(self.w0[:self.w0.size // 2])
def read_pickle_file(self, filename):
"""Read the TOAs from the pickle file specified in filename. Note
the filename should include any pickle-specific extensions (ie
".pickle.gz" or similar), these will not be added automatically."""
log.info("Reading pickled TOAs from '%s'..." % filename)
if os.path.splitext(filename)[1] == '.gz':
infile = gzip.open(filename, 'rb')
else:
infile = open(filename, 'rb')
tmp = pickle.load(infile)
self.filename = tmp.filename
if hasattr(tmp, 'toas'):
self.toas = tmp.toas
if hasattr(tmp, 'table'):
self.table = tmp.table.group_by("obs")
self.commands = tmp.commands
def read_pickle_file(self, filename):
"""Read the TOAs from the pickle file specified in filename. Note
the filename should include any pickle-specific extensions (ie
".pickle.gz" or similar), these will not be added automatically."""
log.info("Reading pickled TOAs from '%s'..." % filename)
if os.path.splitext(filename)[1] == '.gz':
infile = gzip.open(filename,'rb')
else:
infile = open(filename,'rb')
tmp = pickle.load(infile)
self.filename = tmp.filename
if hasattr(tmp, 'toas'):
self.toas = tmp.toas
if hasattr(tmp, 'table'):
self.table = tmp.table.group_by("obs")
self.commands = tmp.commands
