def test_jumpproposal(dmx_psrs,caplog):
m2a=models.model_2a(dmx_psrs,noisedict=noise_dict)
jp=sampler.JumpProposal(m2a)
assert jp.draw_from_prior.__name__ == 'draw_from_prior'
assert jp.draw_from_signal_prior.__name__ == 'draw_from_signal_prior'
assert (jp.draw_from_par_prior('J1713+0747').__name__ ==
'draw_from_J1713+0747_prior')
assert (jp.draw_from_par_log_uniform({'gw':(-20,-10)}).__name__ ==
'draw_from_gw_log_uniform')
assert (jp.draw_from_signal('red noise').__name__ ==
'draw_from_red noise_signal')
groups = sampler.get_parameter_groups(pta)
Sampler = ptmcmc(ndim, pta.get_lnlikelihood, pta.get_lnprior,
cov, groups=groups, outDir=args.outdir, resume=True)
np.save(args.outdir + 'pars.npy', pta.param_names)
np.save(args.outdir + 'par_model.npy', np.array(pta.params).astype(str))
np.save(args.outdir + 'signals.npy', list(pta.signals.keys()))
try:
achrom_freqs = get_freqs(pta, signal_id='gw')
np.savetxt(args.outdir + 'achrom_rn_freqs.txt', achrom_freqs, fmt='%.18e')
except:
pass
jp = sampler.JumpProposal(pta)
Sampler.addProposalToCycle(jp.draw_from_prior, 15)
if args.rednoise:
Sampler.addProposalToCycle(jp.draw_from_red_prior, 15)
Sampler.addProposalToCycle(jp.draw_from_gwb_log_uniform_distribution, 15)
if args.bayes_ephem:
Sampler.addProposalToCycle(jp.draw_from_ephem_prior, 15)
N = int(args.niter)
Sampler.sample(x0, Niter=N, SCAMweight=30, AMweight=15,
DEweight=50, burn=100000)
save_core(args.corepath, args.outdir)