From eb891e6c944bf98e63283d948ba7a621b77b4a1d Mon Sep 17 00:00:00 2001 From: Tao Wang Date: Thu, 17 Sep 2020 01:12:30 +0800 Subject: [PATCH] [TEST]make retry counts (#10463) * make retry counts * lint * use function * move methond inside --- python/ray/tests/test_advanced_2.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/python/ray/tests/test_advanced_2.py b/python/ray/tests/test_advanced_2.py index da20a974d..1dc0b895d 100644 --- a/python/ray/tests/test_advanced_2.py +++ b/python/ray/tests/test_advanced_2.py @@ -585,25 +585,30 @@ def test_many_custom_resources(shutdown_only): def test_zero_capacity_deletion_semantics(shutdown_only): ray.init(num_cpus=2, num_gpus=1, resources={"test_resource": 1}) - def test(): - resources = ray.available_resources() - MAX_RETRY_ATTEMPTS = 5 - retry_count = 0 - + def delete_miscellaneous_item(resources): del resources["memory"] del resources["object_store_memory"] for key in list(resources.keys()): if key.startswith("node:"): del resources[key] + def test(): + resources = ray.available_resources() + MAX_RETRY_ATTEMPTS = 5 + retry_count = 0 + + delete_miscellaneous_item(resources) + while resources and retry_count < MAX_RETRY_ATTEMPTS: time.sleep(0.1) resources = ray.available_resources() + delete_miscellaneous_item(resources) retry_count += 1 if retry_count >= MAX_RETRY_ATTEMPTS: raise RuntimeError( - "Resources were available even after five retries.", resources) + "Resources were available even after {} retries.".format( + MAX_RETRY_ATTEMPTS), resources) return resources