dbartlett/borg_velocity
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Switch to BORG velocity model

Browse source
This commit is contained in:
Deaglan Bartlett 2024-04-24 11:58:34 +02:00
parent 8728be9b3a
commit f483a62dda
12 changed files with 147 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_likelihood.py
View file

@ -8,10 +8,10 @@ import borg_velocity.forwards as forwards
import borg_velocity.utils as utils
ini_file = '../conf/basic_ini.ini'
test_scaling = False
test_sigma8 = False
test_omegam = False
test_alpha = False
test_scaling = True
test_sigma8 = True
test_omegam = True
test_alpha = True
test_muA = True
# Input box
@ -27,7 +27,8 @@ model = likelihood.build_gravity_model(None, box_in, ini_file=ini_file)
cosmo = utils.get_cosmopar(ini_file)
model.setCosmoParams(cosmo)
fwd_param = forwards.NullForward(box_in)
mylike = likelihood.VelocityBORGLikelihood(model, fwd_param, ini_file)
fwd_vel = likelihood.fwd_vel
mylike = likelihood.VelocityBORGLikelihood(model, fwd_param, fwd_vel, ini_file)
# Create mock data
state = borg.likelihood.MarkovState()
@ -111,7 +112,7 @@ if test_omegam:
# Test bias model
if test_alpha:
all_alpha = np.linspace(-1.0, 5.0, 50)
all_alpha = np.linspace(-2.0, 5.0, 50)
all_lkl = np.empty(all_alpha.shape)
for i, alpha in enumerate(all_alpha):
mylike.fwd_param.setModelParams({'alpha0':alpha})
@ -135,12 +136,12 @@ if test_alpha:
# Test bias model
if test_muA:
all_muA = np.linspace(0.5, 1.5, 50)
all_muA = np.linspace(0.95, 1.05, 50)
all_lkl = np.empty(all_muA.shape)
for i, muA in enumerate(all_muA):
mylike.fwd_param.setModelParams({'muA0':muA})
mylike.fwd_param.setModelParams({'mua0':muA})
all_lkl[i] = mylike.logLikelihoodComplex(s_hat, None)
mylike.fwd_param.setModelParams({'muA0':mylike.muA[0]})
mylike.fwd_param.setModelParams({'mua0':mylike.muA[0]})
fid_lkl = mylike.logLikelihoodComplex(s_hat, None)
all_lkl -= fid_lkl
all_lkl = np.exp(-all_lkl)