def test_sample_starting_params_from_lhs__with_uniform_priors_and_two_samples(
):
priors = [
{
"param_name": "ice_cream_sales",
"distribution": "uniform",
"distri_params": [1, 5]
},
{
"param_name": "air_temp",
"distribution": "uniform",
"distri_params": [1, 10]
},
]
specify_missing_prior_params(priors)
params = sample_starting_params_from_lhs(priors, n_samples=2)
assert _prepare_params(params) == _prepare_params([{
"ice_cream_sales": 2.0,
"air_temp": 3.25
}, {
"ice_cream_sales": 4.0,
"air_temp": 7.75
}], ) or _prepare_params(params) == _prepare_params([{
"ice_cream_sales": 4.0,
"air_temp": 3.25
}, {
"ice_cream_sales": 2.0,
"air_temp": 7.75
}], )
def test_sample_starting_params_from_lhs__with_uniform_prior_and_one_sample():
priors = [{
"param_name": "ice_cream_sales",
"distribution": "uniform",
"distri_params": [1, 5]
}]
specify_missing_prior_params(priors)
params = sample_starting_params_from_lhs(priors, n_samples=1)
assert _prepare_params(params) == _prepare_params([{
"ice_cream_sales": 3.0
}])
def test_sample_starting_params_from_lhs__with_gamma_prior_and_one_sample():
priors = [{
"param_name": "ice_cream_sales",
"distribution": "gamma",
"distri_mean": 5.0,
"distri_ci": [3.0, 7.0],
}]
specify_missing_prior_params(priors)
params = sample_starting_params_from_lhs(priors, n_samples=1)
assert _prepare_params(params) == _prepare_params([{
"ice_cream_sales":
4.932833078981056
}])
def test_sample_starting_params_from_lhs__with_beta_prior_and_one_sample():
priors = [{
"param_name": "ice_cream_sales",
"distribution": "beta",
"distri_mean": 0.05,
"distri_ci": [0.01, 0.1],
}]
specify_missing_prior_params(priors)
params = sample_starting_params_from_lhs(priors, n_samples=1)
assert _prepare_params(params) == _prepare_params([{
"ice_cream_sales":
0.04680260472064115
}])
def test_sample_starting_params_from_lhs__with_lognormal_prior_and_one_sample(
):
priors = [{
"param_name": "ice_cream_sales",
"distribution": "lognormal",
"distri_params": [-1, 1],
}]
specify_missing_prior_params(priors)
params = sample_starting_params_from_lhs(priors, n_samples=1)
assert _prepare_params(params) == _prepare_params([{
"ice_cream_sales":
0.36787944117144233
}])
