Skip to content

Commit aa4db8a

Browse files
committed
finish up unit test
1 parent 2bc504c commit aa4db8a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

libensemble/tests/unit_tests/test_ensemble.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
import numpy as np
44

55
import libensemble.tests.unit_tests.setup as setup
6-
from libensemble.resources.platforms import PerlmutterGPU
76
from libensemble.utils.misc import pydanticV1, specs_dump
87

98

@@ -168,8 +167,9 @@ def test_flakey_workflow():
168167

169168

170169
def test_ensemble_specs_update_libE_specs():
171-
170+
"""Test that libE_specs is updated as expected with .attribute setting"""
172171
from libensemble.ensemble import Ensemble
172+
from libensemble.resources.platforms import PerlmutterGPU
173173
from libensemble.specs import LibeSpecs
174174

175175
platform_specs = PerlmutterGPU()
@@ -185,6 +185,10 @@ def test_ensemble_specs_update_libE_specs():
185185
platform_specs=platform_specs,
186186
)
187187

188+
assert ensemble.libE_specs.num_resource_sets == ensemble.nworkers - 1
189+
assert len(str(ensemble.libE_specs.workflow_dir_path)) > 1
190+
assert ensemble.libE_specs.platform_specs == specs_dump(platform_specs, exclude_none=True)
191+
188192

189193
if __name__ == "__main__":
190194
test_ensemble_init()

0 commit comments

Comments
 (0)