mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 11:21:15 +08:00
SangBin Cho
8cedcdf2df
* Trial 1. * Fix. * Revert "Fix." This reverts commit 26ad970f753d581f340857be30054d6954df8255. * Revert "Trial 1." This reverts commit 63f7aca5162bb40f2d5e28fb9647598cbde7ad41. * Another fix try. * Last trial. * Remove unnecessary comment. * Small fix. * Use better units. * Lint.
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
import re
|
|
import subprocess
|
|
import sys
|
|
import pytest
|
|
|
|
import ray
|
|
|
|
from ray.test_utils import run_string_as_driver_nonblocking
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == "win32", reason="Failing on Windows.")
|
|
def test_worker_stdout():
|
|
script = """
|
|
import ray
|
|
import sys
|
|
|
|
ray.init(num_cpus=2)
|
|
|
|
@ray.remote
|
|
def foo(out_str, err_str):
|
|
print(out_str)
|
|
print(err_str, file=sys.stderr)
|
|
|
|
ray.get(foo.remote("abc", "def"))
|
|
"""
|
|
|
|
proc = run_string_as_driver_nonblocking(script)
|
|
out_str = proc.stdout.read().decode("ascii")
|
|
err_str = proc.stderr.read().decode("ascii")
|
|
|
|
assert out_str.endswith("abc\n")
|
|
assert err_str.split("\n")[-2].endswith("def")
|
|
|
|
|
|
def test_output():
|
|
# Use subprocess to execute the __main__ below.
|
|
outputs = subprocess.check_output(
|
|
[sys.executable, __file__, "_ray_instance"],
|
|
stderr=subprocess.STDOUT).decode()
|
|
lines = outputs.split("\n")
|
|
for line in lines:
|
|
print(line)
|
|
assert len(lines) == 3, lines
|
|
logging_header = r"\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2},\d{3}\sINFO\s"
|
|
assert re.match(
|
|
logging_header + r"resource_spec.py:\d+ -- Starting Ray with [0-9\.]+ "
|
|
r"GiB memory available for workers and up to [0-9\.]+ GiB "
|
|
r"for objects. You can adjust these settings with .*?.", lines[0])
|
|
assert re.match(
|
|
logging_header +
|
|
r"services.py:\d+ -- View the Ray dashboard at .*?localhost:\d+?.*",
|
|
lines[1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1 and sys.argv[1] == "_ray_instance":
|
|
# Set object store memory very low so that it won't complain
|
|
# about low shm memory in Linux environment.
|
|
# The test failures currently complain it only has 2 GB memory,
|
|
# so let's set it much lower than that.
|
|
MB = 1000**2
|
|
ray.init(num_cpus=1, object_store_memory=(100 * MB))
|
|
ray.shutdown()
|
|
else:
|
|
sys.exit(pytest.main(["-v", __file__]))
|