hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Reduce verbosity of component_failures_test.py. (#440)

Browse source
This commit is contained in:
Robert Nishihara 2017-04-07 23:05:29 -07:00 • committed by Philipp Moritz
parent 05fd4c2c37
commit 0eac3ccdd0
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
test/component_failures_test.py
View file

@ -24,7 +24,8 @@ class ComponentFailureTest(unittest.TestCase):
ray.worker._init(num_workers=1,
driver_mode=ray.SILENT_MODE,
start_workers_from_local_scheduler=False,
start_ray_local=True)
start_ray_local=True,
redirect_output=True)
# Have the worker wait in a get call.
f.remote()
@ -56,7 +57,8 @@ class ComponentFailureTest(unittest.TestCase):
ray.worker._init(num_workers=1,
driver_mode=ray.SILENT_MODE,
start_workers_from_local_scheduler=False,
start_ray_local=True)
start_ray_local=True,
redirect_output=True)
# Have the worker wait in a get call.
f.remote()
@ -87,7 +89,8 @@ class ComponentFailureTest(unittest.TestCase):
num_local_schedulers=num_local_schedulers,
start_workers_from_local_scheduler=False,
start_ray_local=True,
num_cpus=[num_initial_workers] * num_local_schedulers)
num_cpus=[num_initial_workers] * num_local_schedulers,
redirect_output=True)
# Submit more tasks than there are workers so that all workers and cores
# are utilized.
object_ids = [f.remote(i) for i
@ -123,7 +126,8 @@ class ComponentFailureTest(unittest.TestCase):
num_workers=num_local_schedulers * num_workers_per_scheduler,
num_local_schedulers=num_local_schedulers,
start_ray_local=True,
num_cpus=[num_workers_per_scheduler] * num_local_schedulers)
num_cpus=[num_workers_per_scheduler] * num_local_schedulers,
redirect_output=True)
# Submit more tasks than there are workers so that all workers and cores
# are utilized.
@ -196,7 +200,7 @@ class ComponentFailureTest(unittest.TestCase):
False)
def testDriverLivesSequential(self):
ray.worker.init()
ray.worker.init(redirect_output=True)
all_processes = ray.services.all_processes
processes = [
all_processes[ray.services.PROCESS_TYPE_PLASMA_STORE][0],
@ -214,7 +218,7 @@ class ComponentFailureTest(unittest.TestCase):
# If the driver can reach the tearDown method, then it is still alive.
def testDriverLivesParallel(self):
ray.worker.init()
ray.worker.init(redirect_output=True)
all_processes = ray.services.all_processes
processes = [
all_processes[ray.services.PROCESS_TYPE_PLASMA_STORE][0],