wehub-resource-sync
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import sys
|
|
|
|
import pytest
|
|
|
|
from ray._raylet import NodeID
|
|
from ray.runtime_context import RuntimeContext
|
|
from ray.util.client.ray_client_helpers import ray_start_client_server
|
|
|
|
|
|
def test_get_ray_metadata(ray_start_regular_shared):
|
|
"""Test the ClusterInfo client data pathway and API surface"""
|
|
with ray_start_client_server() as ray:
|
|
ip_address = ray_start_regular_shared["node_ip_address"]
|
|
|
|
initialized = ray.is_initialized()
|
|
assert initialized
|
|
|
|
nodes = ray.nodes()
|
|
assert len(nodes) == 1, nodes
|
|
assert nodes[0]["NodeManagerAddress"] == ip_address
|
|
|
|
current_node_id = "node:" + ip_address
|
|
|
|
cluster_resources = ray.cluster_resources()
|
|
available_resources = ray.available_resources()
|
|
|
|
assert cluster_resources["CPU"] == 1.0
|
|
assert current_node_id in cluster_resources
|
|
assert current_node_id in available_resources
|
|
|
|
|
|
def test_get_runtime_context(ray_start_regular_shared):
|
|
"""Test the get_runtime_context data through the metadata API"""
|
|
with ray_start_client_server() as ray:
|
|
rtc = ray.get_runtime_context()
|
|
assert isinstance(rtc, RuntimeContext)
|
|
assert isinstance(rtc.node_id, NodeID)
|
|
assert len(rtc.node_id.hex()) == 56
|
|
assert isinstance(rtc.namespace, str)
|
|
|
|
wid = ray.get_runtime_context().get_worker_id()
|
|
assert isinstance(wid, str)
|
|
assert len(wid) == 56
|
|
|
|
# Ensure this doesn't throw
|
|
ray.get_runtime_context().get()
|
|
|
|
with pytest.raises(Exception):
|
|
_ = rtc.task_id
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|