Testing/increasing coverage

tests/test_scheduler.py

@@ -10,6 +10,7 @@
     _construct_machine_list_slurm,
     _parse_host_info,
     _parse_machine_data,
+    _restrict_machines_to_core_count,
     load_machines,
 )
 from ansys.fluent.core.scheduler.machine_list import Machine, MachineList
@@ -157,6 +158,31 @@ def setUp(self):
     def tearDown(self):
         self._machineList.reset()
 
+    def test_machine_info(self):
+        info = [
+            {"machine-name": "M0", "core-count": 1},
+            {"machine-name": "M1", "core-count": 6},
+        ]
+        machineList = load_machines(machine_info=info)
+        self.assertEqual(machineList.number_of_cores, 7)
+
+    def test_restrict_machines(self):
+        info = [
+            {"machine-name": "M0", "core-count": 1},
+            {"machine-name": "M1", "core-count": 1},
+        ]
+        machineList = load_machines(machine_info=info)
+        old_machine_list = _restrict_machines_to_core_count(
+            machineList, ncores=machineList.number_of_cores
+        )
+        self.assertEqual(machineList, old_machine_list)
+
+    def test_lsb_mcpu(self):
+        os.environ["LSB_MCPU_HOSTS"] = "m1 3 m2 3"
+        machineList = load_machines()
+        del os.environ["LSB_MCPU_HOSTS"]
+        self.assertEqual(machineList.number_of_cores, 6)
+
     def test_no_environment(self):
         machineList = load_machines()
         self.assertEqual(machineList[0].host_name, socket.gethostname())
@@ -381,6 +407,12 @@ def test_cmd_string(self):
                     machine.number_of_cores, self._expectedValues[machine.host_name]
                 )
 
+    def test_no_machine_name(self):
+        hostList = "M0:2,M1:2,"
+        with self.assertRaises(RuntimeError) as cm:
+            _parse_host_info(hostList)
+        self.assertEqual(str(cm.exception), "Problem with machine list format.")
+
     def test_host_file(self):
         import os.path