def test_run_legacy_ensemble(tmpdir, unused_tcp_port, make_ensemble_builder):
num_reals = 2
with tmpdir.as_cwd():
ensemble = make_ensemble_builder(tmpdir, num_reals, 2).build()
config = EvaluatorServerConfig(unused_tcp_port)
evaluator = EnsembleEvaluator(ensemble, config, ee_id="1")
monitor = evaluator.run()
for e in monitor.track():
if (e["type"] in (
identifiers.EVTYPE_EE_SNAPSHOT_UPDATE,
identifiers.EVTYPE_EE_SNAPSHOT,
) and e.data.get("status") in ["Failed", "Stopped"]):
monitor.signal_done()
assert evaluator._snapshot.get_status() == "Stopped"
assert evaluator.get_successful_realizations() == num_reals
def test_run_legacy_ensemble(tmpdir, unused_tcp_port, make_ensemble_builder):
num_reals = 2
with tmpdir.as_cwd():
ensemble = make_ensemble_builder(tmpdir, num_reals, 2).build()
config = EvaluatorServerConfig(unused_tcp_port)
evaluator = EnsembleEvaluator(ensemble, config, 0, ee_id="1")
with evaluator.run() as monitor:
for e in monitor.track():
if (e["type"] in (
identifiers.EVTYPE_EE_SNAPSHOT_UPDATE,
identifiers.EVTYPE_EE_SNAPSHOT,
) and e.data.get(identifiers.STATUS) in [
state.ENSEMBLE_STATE_FAILED,
state.ENSEMBLE_STATE_STOPPED
]):
monitor.signal_done()
assert evaluator._snapshot.get_status() == state.ENSEMBLE_STATE_STOPPED
assert evaluator.get_successful_realizations() == num_reals
def test_run_legacy_ensemble(tmpdir, unused_tcp_port, make_ensemble_builder):
num_reals = 2
conf_file = Path(tmpdir / CONFIG_FILE)
with tmpdir.as_cwd():
with open(conf_file, "w") as f:
f.write(f'port: "{unused_tcp_port}"\n')
ensemble = make_ensemble_builder(tmpdir, num_reals, 2).build()
config = load_config(conf_file)
evaluator = EnsembleEvaluator(ensemble, config, ee_id="1")
monitor = evaluator.run()
for e in monitor.track():
if (e["type"] in (
identifiers.EVTYPE_EE_SNAPSHOT_UPDATE,
identifiers.EVTYPE_EE_SNAPSHOT,
) and e.data.get("status") == "Stopped"):
monitor.signal_done()
assert evaluator.get_successful_realizations() == num_reals
