fix mercury/mask-mpi test library paths

LeiHuang-LM · May 10, 2024 · 6a5e7a1 · 6a5e7a1
1 parent 0517182
commit 6a5e7a1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 13 deletions.
diff --git a/src/sst/elements/mask-mpi/tests/testsuite_default_mask_mpi.py b/src/sst/elements/mask-mpi/tests/testsuite_default_mask_mpi.py
@@ -20,15 +20,14 @@ def tearDown(self):
 
 #####
 
-    def test_testme(self):
+    def test_sendrecv(self):
         testdir = self.get_testsuite_dir()
-
-        paths = os.environ.get("SST_LIB_PATH")
-        if paths is None:
-            os.environ["SST_LIB_PATH"] = testdir
+        libdir = sstsimulator_conf_get_value_str("SST_ELEMENT_LIBRARY","SST_ELEMENT_LIBRARY_LIBDIR")
+        path = os.environ.get("SST_LIB_PATH")
+        if path is None or path == "":
+            os.environ["SST_LIB_PATH"] = libdir
         else:
-            os.environ["SST_LIB_PATH"] = paths + ":" + testdir
-
+            os.environ["SST_LIB_PATH"] = path + ":" + libdir
         self.mask_mpi_template("test_sendrecv")
 
 #####

diff --git a/src/sst/elements/mercury/tests/testsuite_default_hg.py b/src/sst/elements/mercury/tests/testsuite_default_hg.py
@@ -21,13 +21,12 @@ def tearDown(self):
 
     def test_testme(self):
         testdir = self.get_testsuite_dir()
-
-        paths = os.environ.get("SST_LIB_PATH")
-        if paths is None:
-            os.environ["SST_LIB_PATH"] = testdir
+        libdir = sstsimulator_conf_get_value_str("SST_ELEMENT_LIBRARY","SST_ELEMENT_LIBRARY_LIBDIR")
+        path = os.environ.get("SST_LIB_PATH")
+        if path is None or path == "":
+            os.environ["SST_LIB_PATH"] = libdir
         else:
-            os.environ["SST_LIB_PATH"] = paths + ":" + testdir
-
+            os.environ["SST_LIB_PATH"] = path + ":" + libdir
         self.simple_components_template("ostest")
 
 #####