Ensure HOME env variable is corrected for OpenMPI 5+

If exists KHIOPS_MPI_HOME is prepended to HOME

Author: Thierry RAMORASOAVINA

khiops/core/internals/runner.py

@@ -994,8 +994,12 @@ def _initialize_khiops_environment(self):
                     var_value = ""
                 else:
                     continue
+
+                if var_name == "KHIOPS_MPI_HOME":
+                    # Prepend the corresponding value to HOME for OpenMPI 5+
+                    os.environ["HOME"] = var_value + os.pathsep + os.environ["HOME"]
                 # Set paths to Khiops binaries
-                if var_name == "KHIOPS_PATH":
+                elif var_name == "KHIOPS_PATH":
                     self.khiops_path = var_value
                     os.environ["KHIOPS_PATH"] = var_value
                 elif var_name == "KHIOPS_COCLUSTERING_PATH":

tests/test_khiops_integrations.py

@@ -282,12 +282,93 @@ def test_environment_error_on_bogus_khiops_env_script(self):
         output_msg = str(context.exception)
         self.assertEqual(output_msg, expected_msg)
 
+    def test_runner_environment_for_openmpi5(self):
+        """Test if HOME is actually corrected for OpenMPI 5+"""
+
+        with tempfile.TemporaryDirectory() as temp_dir:
+
+            # Save current HOME
+            old_home = os.environ["HOME"]
+
+            # Create temporary khiops_env binary dir
+            # Note: The "bin" subdir is needed for Windows.
+            temp_khiops_env_dir = os.path.join(temp_dir, "bin")
+            os.makedirs(temp_khiops_env_dir)
+
+            # Create temporary khiops_env path
+            temp_khiops_env_file_path = os.path.join(temp_khiops_env_dir, "khiops_env")
+
+            # On Windows, set KHIOPS_HOME to the temp dir
+            original_khiops_home_env_var = os.environ.get("KHIOPS_HOME")
+            if platform.system() == "Windows":
+                os.environ["KHIOPS_HOME"] = temp_dir
+                temp_khiops_env_file_path += ".cmd"
+
+            # Replace the khiops_env with a crafted one
+            with open(
+                temp_khiops_env_file_path, "w", encoding="utf-8"
+            ) as temp_khiops_env_file:
+                if platform.system() == "Windows":
+                    temp_khiops_env_file.write(
+                        "echo KHIOPS_PATH /usr/bin/MODL_openmpi\r\n"
+                    )
+                    temp_khiops_env_file.write(
+                        "echo KHIOPS_COCLUSTERING_PATH "
+                        "/usr/bin/MODL_Coclustering_openmpi\r\n"
+                    )
+                    temp_khiops_env_file.write("echo KHIOPS_MPI_HOME \tmp" + "\r\n")
+                else:
+                    temp_khiops_env_file.write("#!/bin/bash\n")
+                    temp_khiops_env_file.write(
+                        "echo KHIOPS_PATH /usr/bin/MODL_openmpi\n"
+                    )
+                    temp_khiops_env_file.write(
+                        "echo KHIOPS_COCLUSTERING_PATH "
+                        "/usr/bin/MODL_Coclustering_openmpi\n"
+                    )
+                    temp_khiops_env_file.write("echo KHIOPS_MPI_HOME /tmp\n")
+
+            # Make the temporary khiops_env file executable
+            os.chmod(
+                temp_khiops_env_file_path,
+                os.stat(temp_khiops_env_file_path).st_mode | stat.S_IEXEC,
+            )
+
+            # Store initial PATH
+            original_path_env_var = os.environ["PATH"]
+
+            # Prepend path to temporary khiops_env to PATH
+            os.environ["PATH"] = (
+                temp_khiops_env_dir + os.pathsep + original_path_env_var
+            )
+
+            _ = KhiopsLocalRunner()
+
+            # Check `HOME` is corrected for OpenMPI 5+
+            self.assertEqual(
+                os.path.join("/", "tmp") + os.pathsep + old_home, os.environ["HOME"]
+            )
+
+            # Restore HOME
+            os.environ["HOME"] = old_home
+
+            # Restore initial PATH
+            os.environ["PATH"] = original_path_env_var
+
+            # On Windows, restore initial KHIOPS_HOME
+            if (
+                platform.system() == "Windows"
+                and original_khiops_home_env_var is not None
+            ):
+                os.environ["KHIOPS_HOME"] = original_khiops_home_env_var
+
     def test_runner_environment_initialization(self):
         """Test that local runner initializes/ed its environment properly
 
         .. note::
             To test a real initialization this test should be executed alone.
         """
+
         # Obtain the current runner
         runner = kh.get_runner()