wehub-resource-sync
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
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__]))
|