Fix test for available resources (ray-project#2914)

rsepassi · Sep 26, 2018 · fcdca6d · fcdca6d
1 parent 971df5e
commit fcdca6d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/ray/test/test_global_state.py b/python/ray/test/test_global_state.py
@@ -26,7 +26,7 @@ class TestAvailableResources(object):
 
     def test_no_tasks(self):
         cluster_resources = ray.global_state.cluster_resources()
-        available_resources = ray.global_state.cluster_resources()
+        available_resources = ray.global_state.available_resources()
         assert cluster_resources == available_resources
 
     def test_replenish_resources(self):
@@ -37,8 +37,8 @@ def test_replenish_resources(self):
         resources_reset = False
 
         while not resources_reset and time.time() - start < self.timeout:
-            resources_reset = (
-                cluster_resources == ray.global_state.available_resources())
+            available_resources = ray.global_state.available_resources()
+            resources_reset = (cluster_resources == available_resources)
 
         assert resources_reset