Backup PYTHONPATH for use with run_external (#157)

Author: berland

src/subscript/runrms/runrms.py

@@ -218,6 +218,14 @@ def __init__(self):
         if "PYTHONPATH" in os.environ:
             self.oldpythonpath = os.environ["PYTHONPATH"]
 
+        # Set environment variables for use by `run_external`
+        # from equinor/equilibrium:
+        if "PYTHONPATH" in os.environ:
+            os.environ["_PRE_RMS_PYTHONPATH"] = os.environ["PYTHONPATH"]
+        else:
+            os.environ["_PRE_RMS_PYTHONPATH"] = ""
+        os.environ["_PRE_RMS_BACKUP"] = "1"
+
         print(
             _BColors.BOLD,
             "\nRunning <{0}>. Type <{0} -h> for help\n".format(THISSCRIPT),
@@ -500,7 +508,7 @@ def get_pythonpath(self):
             )
 
     def launch_rms(self, empty=False):
-        """Lauch RMS with correct pythonpath"""
+        """Launch RMS with correct pythonpath"""
 
         if self.exe is None:
             self.exe = "rms -v " + self.version_requested

tests/test_runrms.py

@@ -41,6 +41,12 @@ def test_do_parse_args(tmpdir):
     assert "dryrun=True" in str(runner.args)
 
 
+def test_backuppythonpath():
+    rr.RunRMS()
+    assert "_PRE_RMS_BACKUP" in os.environ
+    assert "_PRE_RMS_PYTHONPATH" in os.environ
+
+
 @pytest.mark.integration
 def test_integration():
     """Test that the endpoint is installed"""