diff --git a/python/ray/experimental/test/async_test.py b/python/ray/experimental/test/async_test.py index bbe706eb6..3bdca4118 100644 --- a/python/ray/experimental/test/async_test.py +++ b/python/ray/experimental/test/async_test.py @@ -46,42 +46,6 @@ def test_gather(init): assert all(a == b for a, b in zip(results, ray.get(tasks))) -def test_gather_benchmark(init): - @ray.remote - def f(n): - time.sleep(0.001 * n) - return 42 - - async def test_async(): - sum_time = 0. - for _ in range(50): - tasks = [f.remote(n) for n in range(20)] - start = time.time() - futures = [async_api.as_future(obj_id) for obj_id in tasks] - await asyncio.gather(*futures) - sum_time += time.time() - start - return sum_time - - def baseline(): - sum_time = 0. - for _ in range(50): - tasks = [f.remote(n) for n in range(20)] - start = time.time() - ray.get(tasks) - sum_time += time.time() - start - return sum_time - - # warm up - baseline() - # async get - sum_time_1 = asyncio.get_event_loop().run_until_complete(test_async()) - # get - sum_time_2 = baseline() - - # Ensure the new implementation is not too slow. - assert sum_time_2 * 1.2 > sum_time_1 - - def test_wait(init): loop = asyncio.get_event_loop() tasks = gen_tasks()