File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
libensemble/tests/unit_tests Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 33import numpy as np
44
55import libensemble .tests .unit_tests .setup as setup
6- from libensemble .resources .platforms import PerlmutterGPU
76from libensemble .utils .misc import pydanticV1 , specs_dump
87
98
@@ -168,8 +167,9 @@ def test_flakey_workflow():
168167
169168
170169def 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
189193if __name__ == "__main__" :
190194 test_ensemble_init ()
You can’t perform that action at this time.
0 commit comments