More test fixes

User Support Ticket(s) or Story Referenced: TRILFRAME-522
trilinos · Dec 15, 2022 · 39775a0 · 39775a0
1 parent 63d425f
commit 39775a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/packages/framework/pr_tools/trilinosprhelpers/sysinfo/unittests/test_gpu_utils.py b/packages/framework/pr_tools/trilinosprhelpers/sysinfo/unittests/test_gpu_utils.py
@@ -30,8 +30,8 @@ def mock_nvidia_smi():
             "GPU 2: Tesla V100S-PCIE-32GB (UUID: GPU-somehash3)",
             "GPU 3: Tesla V100S-PCIE-32GB (UUID: GPU-somehash4)"]
 
-def mock_which(thing_to_find):
-    return os.path.join(os.getcwd(), thing_to_find)
+def mock_which_not_found(_):
+    return None
 
 
 #==============================================================================
@@ -70,6 +70,6 @@ def test_nvidia_smi_output_without_smi(self):
         Test that without nvidia-smi available the smi interface returns an empty list of output.
         """
         print("")
-        with patch("trilinosprhelpers.sysinfo.gpu_utils.which", side_effect=mock_which):
+        with patch("trilinosprhelpers.sysinfo.gpu_utils.which", side_effect=mock_which_not_found):
             ret = sysinfo.gpu_utils._nvidia_smi()
         self.assertEqual([], ret)
diff --git a/packages/framework/pr_tools/unittests/test_LaunchDriver.py b/packages/framework/pr_tools/unittests/test_LaunchDriver.py
@@ -27,12 +27,12 @@ def setUp(self):
 
     ## Test LaunchDriver methods
     def testUnitGetLaunchEnv(self):
-      env = ld.get_launch_env(self.build_name, "dne")
+      env = ld.get_launch_env("dne")
       self.assertEqual(env, "")
 
 
     def testUnitGetLaunchCmd(self):
-      cmd = ld.get_launch_cmd(self.build_name, "dne")
+      cmd = ld.get_launch_cmd("dne")
       self.assertEqual(cmd, " ")