mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 03:50:55 +08:00
Sven
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
|
|
|
|
@pytest.fixture
|
|
def ray_gdb_start():
|
|
# Setup environment and start ray
|
|
_environ = os.environ.copy()
|
|
for process_name in ["RAYLET", "PLASMA_STORE"]:
|
|
os.environ["RAY_{}_GDB".format(process_name)] = "1"
|
|
os.environ["RAY_{}_TMUX".format(process_name)] = "1"
|
|
|
|
yield None
|
|
|
|
# Restore original environment and stop ray
|
|
os.environ.clear()
|
|
os.environ.update(_environ)
|
|
ray.shutdown()
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform != "linux" and sys.platform != "linux2",
|
|
reason="This test requires Linux.")
|
|
def test_raylet_gdb(ray_gdb_start):
|
|
# ray_gdb_start yields the expected process name
|
|
ray.init(num_cpus=1)
|
|
|
|
@ray.remote
|
|
def f():
|
|
return 42
|
|
|
|
assert ray.get(f.remote()) == 42
|
|
|
|
# Check process name in `ps aux | grep gdb`
|
|
for process_name in ["raylet/raylet", "plasma/plasma_store_server"]:
|
|
pgrep_command = subprocess.Popen(
|
|
["pgrep", "-f", "gdb.*{}".format(process_name)],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
assert pgrep_command.communicate()[0]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
# Make subprocess happy in bazel.
|
|
os.environ["LC_ALL"] = "en_US.UTF-8"
|
|
os.environ["LANG"] = "en_US.UTF-8"
|
|
sys.exit(pytest.main(["-v", __file__]))
|