Enable more new scheduler tests (#12421)

This commit is contained in:
Eric Liang
2020-11-27 16:10:38 -08:00
committed by GitHub
parent affb0b776c
commit 569eee5e71
5 changed files with 3 additions and 10 deletions
+1 -4
View File
@@ -21,7 +21,7 @@ def increase(x):
return x + 1
@pytest.mark.skipif(new_scheduler_enabled(), reason="broken")
@pytest.mark.skipif(new_scheduler_enabled(), reason="notimpl")
@pytest.mark.parametrize(
"ray_start_regular", [
generate_system_config_map(
@@ -47,7 +47,6 @@ def test_gcs_server_restart(ray_start_regular):
assert result == 2
@pytest.mark.skipif(new_scheduler_enabled(), reason="broken")
@pytest.mark.parametrize(
"ray_start_regular", [
generate_system_config_map(
@@ -71,7 +70,6 @@ def test_gcs_server_restart_during_actor_creation(ray_start_regular):
assert len(unready) == 0
@pytest.mark.skipif(new_scheduler_enabled(), reason="broken")
@pytest.mark.parametrize(
"ray_start_cluster_head", [
generate_system_config_map(
@@ -132,7 +130,6 @@ def test_node_failure_detector_when_gcs_server_restart(ray_start_cluster_head):
wait_for_condition(condition, timeout=10)
@pytest.mark.skipif(new_scheduler_enabled(), reason="broken")
@pytest.mark.parametrize(
"ray_start_regular", [
generate_system_config_map(
+1 -3
View File
@@ -9,7 +9,7 @@ from ray.test_utils import (
RayTestTimeoutException, check_call_ray, run_string_as_driver,
run_string_as_driver_nonblocking, wait_for_children_of_pid,
wait_for_children_of_pid_to_exit, wait_for_condition, kill_process_by_name,
Semaphore, init_error_pubsub, get_error_message, new_scheduler_enabled)
Semaphore, init_error_pubsub, get_error_message)
def test_remote_raylet_cleanup(ray_start_cluster):
@@ -139,7 +139,6 @@ print("success")
assert "success" in out
@pytest.mark.skipif(new_scheduler_enabled(), reason="hangs")
def test_driver_exiting_quickly(call_ray_start):
# This test will create some drivers that submit some tasks and then
# exit without waiting for the tasks to complete.
@@ -305,7 +304,6 @@ ray.get([a.log.remote(), f.remote()])
"--min-worker-port=0 --max-worker-port=0 --port 0"
],
indirect=True)
@pytest.mark.skipif(new_scheduler_enabled(), reason="hangs")
def test_drivers_release_resources(call_ray_start):
address = call_ray_start
+1 -1
View File
@@ -118,7 +118,7 @@ def verify_load_metrics(monitor, expected_resource_usage=None, timeout=30):
"num_cpus": 2,
}],
indirect=True)
@pytest.mark.skipif(new_scheduler_enabled(), reason="fails")
@pytest.mark.skipif(new_scheduler_enabled(), reason="fails in travis?")
def test_heartbeats_single(ray_start_cluster_head):
"""Unit test for `Cluster.wait_for_nodes`.
-1
View File
@@ -251,7 +251,6 @@ 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)