wehub-resource-sync
47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
import os
|
|
import pathlib
|
|
|
|
import pytest
|
|
|
|
import ray.dashboard.modules # noqa
|
|
from ray.tests.conftest import * # noqa
|
|
|
|
|
|
@pytest.fixture
|
|
def enable_test_module():
|
|
os.environ["RAY_DASHBOARD_MODULE_TEST"] = "true"
|
|
import ray.dashboard.tests
|
|
|
|
p = pathlib.Path(ray.dashboard.modules.__path__[0]) / "tests" / "__init__.py"
|
|
p.touch()
|
|
yield
|
|
os.environ.pop("RAY_DASHBOARD_MODULE_TEST", None)
|
|
p.unlink()
|
|
|
|
|
|
@pytest.fixture
|
|
def disable_aiohttp_cache():
|
|
os.environ["RAY_DASHBOARD_NO_CACHE"] = "true"
|
|
yield
|
|
os.environ.pop("RAY_DASHBOARD_NO_CACHE", None)
|
|
|
|
|
|
@pytest.fixture
|
|
def small_event_line_limit():
|
|
os.environ["EVENT_READ_LINE_LENGTH_LIMIT"] = "1024"
|
|
yield 1024
|
|
os.environ.pop("EVENT_READ_LINE_LENGTH_LIMIT", None)
|
|
|
|
|
|
@pytest.fixture
|
|
def fast_gcs_failure_detection(monkeypatch):
|
|
monkeypatch.setenv("RAY_gcs_rpc_server_reconnect_timeout_s", "2")
|
|
monkeypatch.setenv("GCS_CHECK_ALIVE_INTERVAL_SECONDS", "1")
|
|
|
|
|
|
@pytest.fixture
|
|
def reduce_actor_cache():
|
|
os.environ["RAY_maximum_gcs_destroyed_actor_cached_count"] = "3"
|
|
yield
|
|
os.environ.pop("RAY_maximum_gcs_destroyed_actor_cached_count", None)
|