chore: import upstream snapshot with attribution

python/ray/tests/test_runtime_env_py_executable.py

@@ -0,0 +1,53 @@
+import os
+import sys
+import tempfile
+from pathlib import Path
+
+import pytest
+
+import ray
+
+
+@pytest.fixture(scope="function")
+def tmp_working_dir():
+    """A test fixture, which writes a test file."""
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        path = Path(tmp_dir)
+
+        script_file = path / "start.sh"
+        with script_file.open(mode="w") as f:
+            f.write(sys.executable + " $@")
+
+        yield tmp_dir
+
+
+@pytest.mark.skipif(sys.platform == "win32", reason="Not ported to Windows yet.")
+def test_simple_py_executable(shutdown_only):
+    runtime_env = {
+        "py_executable": "env RAY_TEST_PY_EXECUTABLE_ENV_EXAMPLE=1 " + sys.executable
+    }
+    ray.init(runtime_env=runtime_env)
+
+    @ray.remote
+    def f():
+        return os.environ["RAY_TEST_PY_EXECUTABLE_ENV_EXAMPLE"]
+
+    assert ray.get(f.remote()) == "1"
+
+
+@pytest.mark.skipif(sys.platform == "win32", reason="Not ported to Windows yet.")
+def test_py_executable_with_working_dir(shutdown_only, tmp_working_dir):
+    tmp_dir = tmp_working_dir
+
+    runtime_env = {"working_dir": tmp_dir, "py_executable": "bash start.sh"}
+    ray.init(runtime_env=runtime_env)
+
+    @ray.remote
+    def f():
+        return "hello"
+
+    assert ray.get(f.remote()) == "hello"
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-sv", __file__]))