fixed unit tests

NREL · Nov 30, 2016 · decd6b3 · decd6b3
1 parent 2a7b30e
commit decd6b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/units/test_tracknodes.py b/test/units/test_tracknodes.py
@@ -43,14 +43,14 @@ def test_detect_resourcemanager_slurm_shortpath(self):
         tn.detect_resourcemanager()
         assert( tn.resourcemanager == "slurm" )
 
-    @mock.patch('tracknodes.tracknodes.TrackNodes.which', return_value="pbsnodes")
+    @mock.patch('tracknodes.tracknodes.TrackNodes.which', return_value="sinfo")
     def test_run(self, mock_which):
         out = StringIO()
         orig_stdout = sys.stdout
         sys.stdout = out
 
-        tn = TrackNodes(nodes_cmd="pbsnodes")
-
+        tn = TrackNodes(nodes_cmd="sinfo")
+        tn.resourcemanager="slurm"
         tn.run()
 
         sys.stdout = orig_stdout