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