wehub-resource-sync
72 lines
1.7 KiB
Python
72 lines
1.7 KiB
Python
"""Test that runtime_env raises good errors if ray[default] is not installed.
|
|
|
|
|
|
In CI, this file is run with a minimal ray installation (`pip install ray`.)
|
|
|
|
To run this test file locally, remove some dependency that appears in
|
|
ray[default] but not in ray (e.g., `pip uninstall aiohttp`) and set
|
|
`export RAY_MINIMAL=1`.
|
|
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
from ray.exceptions import RuntimeEnvSetupError
|
|
|
|
|
|
def _test_task_and_actor():
|
|
@ray.remote
|
|
def f():
|
|
return 1
|
|
|
|
with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
|
|
ray.get(f.options(runtime_env={"pip": ["requests"]}).remote())
|
|
|
|
@ray.remote
|
|
class A:
|
|
def task(self):
|
|
return 1
|
|
|
|
a = A.options(runtime_env={"pip": ["requests"]}).remote()
|
|
|
|
with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
|
|
ray.get(a.task.remote())
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform == "win32", reason="runtime_env unsupported on Windows."
|
|
)
|
|
@pytest.mark.skipif(
|
|
os.environ.get("RAY_MINIMAL") != "1",
|
|
reason="This test is only run in CI with a minimal Ray installation.",
|
|
)
|
|
def test_task_actor(shutdown_only):
|
|
ray.init()
|
|
_test_task_and_actor()
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform == "win32", reason="runtime_env unsupported on Windows."
|
|
)
|
|
@pytest.mark.skipif(
|
|
os.environ.get("RAY_MINIMAL") != "1",
|
|
reason="This test is only run in CI with a minimal Ray installation.",
|
|
)
|
|
def test_ray_init(shutdown_only):
|
|
ray.init(runtime_env={"pip": ["requests"]})
|
|
|
|
@ray.remote
|
|
def f():
|
|
return 1
|
|
|
|
with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
|
|
ray.get(f.remote())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|