From bc6e45876ab60ba1e58ed9c2915b05d7ed172bf8 Mon Sep 17 00:00:00 2001 From: jlnav Date: Wed, 16 Sep 2026 10:34:14 -0500 Subject: [PATCH] Reduce slow integration test workloads --- .../test_uniform_sampling_with_variable_resources.py | 4 +++- libensemble/tests/regression_tests/test_mfkg_branin.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/libensemble/tests/functionality_tests/test_uniform_sampling_with_variable_resources.py b/libensemble/tests/functionality_tests/test_uniform_sampling_with_variable_resources.py index cf2d6f81a..384fe53ce 100644 --- a/libensemble/tests/functionality_tests/test_uniform_sampling_with_variable_resources.py +++ b/libensemble/tests/functionality_tests/test_uniform_sampling_with_variable_resources.py @@ -86,7 +86,7 @@ # This can improve scheduling when tasks may run across multiple nodes libE_specs["scheduler_opts"] = {"match_slots": False} - exit_criteria = {"sim_max": 40, "wallclock_max": 300} + exit_criteria = {"sim_max": 12, "wallclock_max": 300} if libE_specs["comms"] == "local": iterations = 4 @@ -122,4 +122,6 @@ if is_manager: assert flag == 0 + assert np.count_nonzero(H["sim_ended"]) >= exit_criteria["sim_max"] + assert len(np.unique(H["resource_sets"])) > 1 save_libE_output(H, persis_info, __file__, nworkers) diff --git a/libensemble/tests/regression_tests/test_mfkg_branin.py b/libensemble/tests/regression_tests/test_mfkg_branin.py index c25ed21c7..5752618bf 100644 --- a/libensemble/tests/regression_tests/test_mfkg_branin.py +++ b/libensemble/tests/regression_tests/test_mfkg_branin.py @@ -46,7 +46,7 @@ UB = np.array([1.0, 1.0]) N_INIT_SAMPLES = 4 # Each initial point gets a high- and a low-fidelity evaluation Q = 4 # Batch size per MFKG iteration (keeps up to q sim workers busy) -SIM_MAX = 16 # Exit after running this many simulations +SIM_MAX = 12 # Initial sample plus one MFKG acquisition batch def run_mfkg(async_return, nworkers, is_manager, libE_specs):