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]