Skip to content

Commit ebca858

Browse files
committed
fix unit test
1 parent 4369cf2 commit ebca858

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

libensemble/tests/unit_tests/test_ensemble.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,11 @@ def test_ensemble_parse_args_false():
3838

3939
def test_full_workflow():
4040
"""Test initializing a workflow via Specs and Ensemble.run()"""
41+
from libensemble.alloc_funcs.give_sim_work_first import give_sim_work_first
4142
from libensemble.ensemble import Ensemble
4243
from libensemble.gen_funcs.sampling import latin_hypercube_sample
4344
from libensemble.sim_funcs.simple_sim import norm_eval
44-
from libensemble.specs import ExitCriteria, GenSpecs, LibeSpecs, SimSpecs
45+
from libensemble.specs import AllocSpecs, ExitCriteria, GenSpecs, LibeSpecs, SimSpecs
4546

4647
LS = LibeSpecs(comms="local", nworkers=4)
4748

@@ -58,6 +59,7 @@ def test_full_workflow():
5859
},
5960
),
6061
exit_criteria=ExitCriteria(gen_max=101),
62+
alloc_specs=AllocSpecs(alloc_f=give_sim_work_first),
6163
)
6264

6365
ens.add_random_streams()

0 commit comments

Comments
 (0)