Fixing the test

pitt-crc · Aug 27, 2024 · 7f1591a · 7f1591a
1 parent 08e8d8f
commit 7f1591a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_crc_idle.py b/tests/test_crc_idle.py
@@ -95,12 +95,12 @@ def test_count_idle_gpu_resources(self, mock_run_command: Mock) -> None:
 
         cluster = 'gpu'
         partition = 'default'
-        mock_run_command.return_value = "node1_4_2_idle_3500\nnode2_4_4_drain_4000"
+        mock_run_command.return_value = "node1_4_2_idle_3500\nnode2_4_4_drain_N/A"
 
         app = CrcIdle()
         result = app.count_idle_resources(cluster, partition)
         expected = {2: {'count': 1, 'min_free_mem': 3500, 'max_free_mem': 3500},
-                    0: {'count': 1, 'min_free_mem': 4000, 'max_free_mem': 4000}
+                    0: {'count': 1, 'min_free_mem': 0, 'max_free_mem': 0}
                     }
         self.assertEqual(expected, result)