From 22dd7e0428adf704f79d023c8b584e3d641b3070 Mon Sep 17 00:00:00 2001 From: Robert Nishihara Date: Mon, 22 Oct 2018 23:16:55 -0700 Subject: [PATCH] Add test for wait reconstruction. (#3110) --- test/runtest.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/test/runtest.py b/test/runtest.py index 4ddeb57f8..a9efb2fb3 100644 --- a/test/runtest.py +++ b/test/runtest.py @@ -2551,3 +2551,22 @@ def test_initialized_local_mode(shutdown_only_with_initialization_check): assert not ray.is_initialized() ray.init(num_cpus=0, local_mode=True) assert ray.is_initialized() + + +@pytest.mark.skipif( + os.environ.get("RAY_USE_XRAY") != "1", + reason="This test only works with xray.") +def test_wait_reconstruction(shutdown_only): + ray.init(num_cpus=1, object_store_memory=10**8) + + @ray.remote + def f(): + return np.zeros(6 * 10**7, dtype=np.uint8) + + x_id = f.remote() + ray.wait([x_id]) + ray.wait([f.remote()]) + assert not ray.worker.global_worker.plasma_client.contains( + ray.pyarrow.plasma.ObjectID(x_id.id())) + ready_ids, _ = ray.wait([x_id]) + assert len(ready_ids) == 1