def test_uniform_distribution(
workspace,
presampled_big_ensemble,
stages_config,
uniform_parameters_config,
):
ert3.engine.sample_record(
workspace,
uniform_parameters_config,
"uniform_coefficients",
"uniform_coefficients0",
1000,
)
coefficients = ert3.storage.get_ensemble_record(
workspace=workspace, record_name="uniform_coefficients0")
assert 1000 == coefficients.ensemble_size
assert_distribution(
presampled_big_ensemble,
stages_config,
uniform_parameters_config,
"uniform",
coefficients,
)
def test_uniform_distribution(
presampled_big_ensemble,
stages_config,
uniform_parameters_config,
):
orig_method = ert3.stats.Uniform.sample
returned_samples = []
def wrapper(self):
retval = orig_method(self)
returned_samples.append(retval)
return retval
with patch(
"ert3.stats.Uniform.sample", side_effect=wrapper, autospec=True
) as sample_calls:
coefficients = ert3.engine.sample_record(
uniform_parameters_config,
"uniform_coefficients",
1000,
)
assert 1000 == len(coefficients)
assert 1000 == len(sample_calls.call_args_list)
assert 1000 == len(returned_samples)
assert_distribution(
presampled_big_ensemble,
stages_config,
uniform_parameters_config,
"uniform",
coefficients,
returned_samples,
)
def test_gaussian_distribution(
workspace,
big_ensemble,
stages_config,
gaussian_parameters_config,
):
ert3.engine.sample_record(workspace, gaussian_parameters_config,
"coefficients", "coefficients0", 1000)
coefficients = ert3.storage.get_ensemble_record(
workspace=workspace, record_name="coefficients0")
assert 1000 == coefficients.ensemble_size
assert_distribution(
big_ensemble,
stages_config,
gaussian_parameters_config,
"gaussian",
coefficients,
)