Refactor unit test paths to use arcann_training instead of deepmd_ite…

…rative
arcann-chem · May 14, 2024 · 02b170b · 02b170b
1 parent 1e31242
commit 02b170b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arcann_training/unittests/test_machine.py b/arcann_training/unittests/test_machine.py
@@ -211,7 +211,7 @@ class TestGetMachineFromConfigs(unittest.TestCase):
         Test case where no matching configuration is found.
     """
 
-    @patch("deepmd_iterative.common.machine.get_host_name", return_value="my_machine")
+    @patch("arcann_training.common.machine.get_host_name", return_value="my_machine")
     def test_matching_hostname(self, mock_get_host_name):
         """
         Test case where a matching hostname is found.
@@ -234,7 +234,7 @@ def test_matching_machine_short_name(self):
         result = get_machine_from_configs(machine_configs, machine_short_name="machine2")
         self.assertEqual(result, "machine2")
 
-    @patch("deepmd_iterative.common.machine.get_host_name", return_value="unknown_machine")
+    @patch("arcann_training.common.machine.get_host_name", return_value="unknown_machine")
     def test_no_matching_config(self, mock_get_host_name):
         """
         Test case where no matching configuration is found.