def evaluator(make_ee_config):
ensemble = TestEnsemble(0, 2, 1, 2)
ee = EnsembleEvaluator(
ensemble,
make_ee_config(),
0,
ee_id="ee-0",
)
yield ee
ee.stop()
def startSimulations(self, arg):
with open(arg["config_file"], "r") as f:
config = yaml.safe_load(f)
ensemble = PrefectEnsemble(config)
ee_config = arg["ee_config"]
ee_id = str(uuid.uuid1()).split("-")[0]
ee = EnsembleEvaluator(ensemble, ee_config, ee_id=ee_id)
ee.run_and_get_successful_realizations()
self._stop_time = time.time()
self._queue_running = False
ee.stop()
def evaluator(ee_config):
ensemble = (create_ensemble_builder().add_realization(
real=create_realization_builder().active(True).set_iens(0).add_stage(
stage=create_stage_builder().add_step(
step=create_step_builder().set_id(0).add_job(
job=create_legacy_job_builder().set_id(0).set_name(
"cat").set_ext_job(Mock())).add_job(
job=create_legacy_job_builder().set_id(1).set_name(
"cat2").set_ext_job(Mock())).set_dummy_io()).
set_id(0).set_status("Unknown"))).set_ensemble_size(2).build())
ee = EnsembleEvaluator(ensemble=ensemble, config=ee_config)
yield ee
ee.stop()
def test_verify_dispatch_failing_job(make_ee_config):
ee_config = make_ee_config(use_token=False, generate_cert=False)
mock_ensemble = MagicMock()
mock_ensemble.snapshot.to_dict.return_value = {}
ee = EnsembleEvaluator(
mock_ensemble,
ee_config,
0,
ee_id="0",
)
ee.run()
dispatch_failing_job.verify(ee_config.client_uri, on_connect=lambda: None)
ee.stop()
def evaluator(ee_config):
ensemble = (create_ensemble_builder().add_realization(
real=create_realization_builder().active(True).set_iens(0).add_step(
step=create_step_builder().set_id("0").set_name("cats").add_job(
job=create_legacy_job_builder().set_id(0).set_name("cat").
set_ext_job(Mock())).add_job(job=create_legacy_job_builder(
).set_id(1).set_name("cat2").set_ext_job(
Mock())).set_dummy_io())).set_ensemble_size(2).build())
ee = EnsembleEvaluator(
ensemble,
ee_config,
0,
ee_id="ee-0",
)
yield ee
ee.stop()
