starting to create unit test

Author: jlnav

libensemble/tests/unit_tests/test_ensemble.py

@@ -3,6 +3,7 @@
 import numpy as np
 
 import libensemble.tests.unit_tests.setup as setup
+from libensemble.resources.platforms import PerlmutterGPU
 from libensemble.utils.misc import pydanticV1, specs_dump
 
 
@@ -166,10 +167,30 @@ def test_flakey_workflow():
     assert not flag, "should've caught input errors"
 
 
+def test_ensemble_specs_update_libE_specs():
+
+    from libensemble.ensemble import Ensemble
+    from libensemble.specs import LibeSpecs
+
+    platform_specs = PerlmutterGPU()
+
+    ensemble = Ensemble(
+        libE_specs=LibeSpecs(comms="local", nworkers=4),
+    )
+
+    ensemble.libE_specs = LibeSpecs(
+        num_resource_sets=ensemble.nworkers - 1,
+        resource_info={"gpus_on_node": 4},
+        use_workflow_dir=True,
+        platform_specs=platform_specs,
+    )
+
+
 if __name__ == "__main__":
     test_ensemble_init()
     test_ensemble_parse_args_false()
     test_from_files()
     test_bad_func_loads()
     test_full_workflow()
     test_flakey_workflow()
+    test_ensemble_specs_update_libE_specs()