def test_J0613_H4(self):
log = logging.getLogger("TestJ0613.fit_tests")
f = ff.GLSFitter(self.toasJ0613, self.modelJ0613)
f.fit_toas()
f.set_fitparams('H3', 'H4')
for pn, p in (f.get_fitparams()).items():
op = getattr(f.model_init, pn)
diff = np.abs(p.value - op.value)
sigma = diff / op.uncertainty_value
# Fit th
assert sigma < 0.7, "refit %s is %lf sigma different from original value" % (
pn, sigma)
def test_dmxparse():
"""Test for dmxparse function."""
m = tm.get_model(os.path.join(datadir, "B1855+09_NANOGrav_9yv1.gls.par"))
t = toa.get_TOAs(os.path.join(datadir, "B1855+09_NANOGrav_9yv1.tim"))
f = fitter.GLSFitter(toas=t, model=m)
f.fit_toas()
dmx = dmxparse(f, save=False)
# Check exception handling
m = tm.get_model(os.path.join(datadir, "B1855+09_NANOGrav_dfg+12_DMX.par"))
t = toa.get_TOAs(os.path.join(datadir, "B1855+09_NANOGrav_dfg+12.tim"))
f = fitter.WLSFitter(toas=t, model=m)
f.fit_toas()
dmx = dmxparse(f, save=False)
def test_J0613_H4(self):
log = logging.getLogger("TestJ0613.fit_tests")
f = ff.GLSFitter(self.toasJ0613, self.modelJ0613)
f.fit_toas()
f.model.free_params = ("H3", "H4")
for pn, p in f.model.get_params_dict("free", "quantity").items():
op = getattr(f.model_init, pn)
diff = np.abs(p.value - op.value)
sigma = diff / op.uncertainty_value
# Fit th
assert (
sigma < 0.7
), "refit %s is %lf sigma different from original value" % (pn,
sigma)
def test_J0613_STIG(toasJ0613, modelJ0613_STIG):
log = logging.getLogger("TestJ0613.fit_tests_stig")
f = ff.GLSFitter(toasJ0613, modelJ0613_STIG)
f.fit_toas()
def test_J0613_fit_with_H3_H4(toasJ0613, modelJ0613):
log = logging.getLogger("TestJ0613.fit_tests")
f = ff.GLSFitter(toasJ0613, modelJ0613)
f.fit_toas()
import glob
log.setLevel('CRITICAL')
import pta_sim
import pta_sim.bayes
import pta_sim.parse_sim as parse_sim
args = parse_sim.arguments()
if args.pickle == 'no_pickle':
if args.use_pint:
parfile = args.pardir + args.psr + '*.gls.par'
timfile = args.timdir + args.psr + '*.tim'
t = toa.get_TOAs(timfile)
m = models.get_model(parfile)
f = fit.GLSFitter(t, m)
f.fit_toas(maxiter=2)
psr = Pulsar(t, f.model, ephem='DE436')
else:
if args.psr == 'J1713+0747':
parfile = glob.glob(args.pardir + args.psr + '*.gls.t2.par')[0]
else:
parfile = glob.glob(args.pardir + args.psr + '*.gls.par')[0]
timfile = glob.glob(args.timdir + args.psr + '*.tim')[0]
psr = Pulsar(parfile, timfile, ephem='DE436')
else:
with open(args.pickle, 'rb') as fin:
psrs = pickle.load(fin)
psr = [p for p in psrs if p.name == args.psr][0]