wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 19:49:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Shard unit tests into medium sized files for test stability (#6398)

Browse Source
This commit is contained in:
Eric Liang
2019-12-09 13:15:29 -08:00
committed by GitHub
parent a6bc2b1842
commit 304b4f0d3d
27 changed files with 3049 additions and 2832 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_multi_node_2.py
-37
View File
@@ -154,43 +154,6 @@ def test_heartbeats_single(ray_start_cluster_head):
ray.get(work_handle)
@pytest.mark.flaky(reruns=4)
def test_heartbeats_cluster(ray_start_cluster_head):
"""Unit test for `Cluster.wait_for_nodes`.
Test proper metrics.
"""
cluster = ray_start_cluster_head
timeout = 8
num_workers_nodes = 3
num_nodes_total = int(num_workers_nodes + 1)
[cluster.add_node() for i in range(num_workers_nodes)]
cluster.wait_for_nodes()
monitor = setup_monitor(cluster.address)
verify_load_metrics(monitor, (0.0, {"CPU": 0.0}, {"CPU": num_nodes_total}))
@ray.remote
class Actor(object):
def work(self, timeout):
time.sleep(timeout)
return True
test_actors = [Actor.remote() for i in range(num_nodes_total)]
work_handles = [actor.work.remote(timeout * 2) for actor in test_actors]
verify_load_metrics(monitor, (num_nodes_total, {
"CPU": num_nodes_total
}, {
"CPU": num_nodes_total
}))
ray.get(work_handles)
verify_load_metrics(monitor, (0.0, {"CPU": 0.0}, {"CPU": num_nodes_total}))
ray.shutdown()
def test_wait_for_nodes(ray_start_cluster_head):
"""Unit test for `Cluster.wait_for_nodes`.