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, )