Enable even more new scheduler tests (#12096)

This commit is contained in:
Eric Liang
2020-11-19 16:47:18 -08:00
committed by GitHub
parent dac09bd569
commit e72abcd0aa
11 changed files with 48 additions and 35 deletions
+5 -3
View File
@@ -10,9 +10,9 @@ import ray
import ray.test_utils
from ray.core.generated import common_pb2
from ray.core.generated import node_manager_pb2, node_manager_pb2_grpc
from ray.test_utils import (wait_for_condition, wait_for_pid_to_exit,
run_string_as_driver,
run_string_as_driver_nonblocking)
from ray.test_utils import (
wait_for_condition, wait_for_pid_to_exit, run_string_as_driver,
run_string_as_driver_nonblocking, new_scheduler_enabled)
def get_workers():
@@ -207,6 +207,7 @@ def test_worker_capping_run_chained_tasks(shutdown_only):
assert len(get_workers()) == 2
@pytest.mark.skipif(new_scheduler_enabled(), reason="fails")
def test_worker_capping_fifo(shutdown_only):
# Start 2 initial workers by setting num_cpus to 2.
info = ray.init(num_cpus=2)
@@ -250,6 +251,7 @@ ray.shutdown()
assert worker2.pid == get_workers()[0].pid
@pytest.mark.skipif(new_scheduler_enabled(), reason="raylet hang 100% cpu")
def test_worker_registration_failure_after_driver_exit(shutdown_only):
info = ray.init(num_cpus=1)