Skip to content

Commit 67b1dec

Browse files
authored
Fix blockallocation test call sites
1 parent 807036a commit 67b1dec

4 files changed

Lines changed: 13 additions & 0 deletions

File tree

tests/unit/executor/test_single_shell_executor.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def test_execute_single_task(self):
3434
self.assertFalse(f.done())
3535
_execute_multiple_tasks(
3636
future_queue=test_queue,
37+
return_queue=queue.Queue(),
3738
cores=1,
3839
openmpi_oversubscribe=False,
3940
spawner=MpiExecSpawner,
@@ -60,6 +61,7 @@ def test_wrong_error(self):
6061
with self.assertRaises(TypeError):
6162
_execute_multiple_tasks(
6263
future_queue=test_queue,
64+
return_queue=queue.Queue(),
6365
cores=1,
6466
openmpi_oversubscribe=False,
6567
spawner=MpiExecSpawner,
@@ -87,6 +89,7 @@ def test_broken_executable(self):
8789
with self.assertRaises(FileNotFoundError):
8890
_execute_multiple_tasks(
8991
future_queue=test_queue,
92+
return_queue=queue.Queue(),
9093
cores=1,
9194
openmpi_oversubscribe=False,
9295
spawner=MpiExecSpawner,

tests/unit/executor/test_single_shell_interactive.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ def test_execute_single_task(self):
9090
self.assertFalse(future_pattern.done())
9191
_execute_multiple_tasks(
9292
future_queue=test_queue,
93+
return_queue=queue.Queue(),
9394
cores=1,
9495
openmpi_oversubscribe=False,
9596
spawner=MpiExecSpawner,

tests/unit/standalone/interactive/test_spawner.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,7 @@ def test_execute_task(self):
263263
cloudpickle_register(ind=1)
264264
_execute_multiple_tasks(
265265
future_queue=q,
266+
return_queue=Queue(),
266267
cores=1,
267268
openmpi_oversubscribe=False,
268269
spawner=MpiExecSpawner,
@@ -445,6 +446,7 @@ def test_execute_task_failed_no_argument(self):
445446
cloudpickle_register(ind=1)
446447
_execute_multiple_tasks(
447448
future_queue=q,
449+
return_queue=Queue(),
448450
cores=1,
449451
openmpi_oversubscribe=False,
450452
spawner=MpiExecSpawner,
@@ -461,6 +463,7 @@ def test_execute_task_failed_wrong_argument(self):
461463
cloudpickle_register(ind=1)
462464
_execute_multiple_tasks(
463465
future_queue=q,
466+
return_queue=Queue(),
464467
cores=1,
465468
openmpi_oversubscribe=False,
466469
spawner=MpiExecSpawner,
@@ -477,6 +480,7 @@ def test_execute_task(self):
477480
cloudpickle_register(ind=1)
478481
_execute_multiple_tasks(
479482
future_queue=q,
483+
return_queue=Queue(),
480484
cores=1,
481485
openmpi_oversubscribe=False,
482486
spawner=MpiExecSpawner,
@@ -495,6 +499,7 @@ def test_execute_task_parallel(self):
495499
cloudpickle_register(ind=1)
496500
_execute_multiple_tasks(
497501
future_queue=q,
502+
return_queue=Queue(),
498503
cores=2,
499504
openmpi_oversubscribe=False,
500505
spawner=MpiExecSpawner,
@@ -518,6 +523,7 @@ def test_execute_task_cache(self):
518523
cloudpickle_register(ind=1)
519524
_execute_multiple_tasks(
520525
future_queue=q,
526+
return_queue=Queue(),
521527
cores=1,
522528
openmpi_oversubscribe=False,
523529
spawner=MpiExecSpawner,
@@ -537,6 +543,7 @@ def test_execute_task_cache_failed_no_argument(self):
537543
cloudpickle_register(ind=1)
538544
_execute_multiple_tasks(
539545
future_queue=q,
546+
return_queue=Queue(),
540547
cores=1,
541548
openmpi_oversubscribe=False,
542549
spawner=MpiExecSpawner,

tests/unit/task_scheduler/interactive/test_spawner_flux.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def test_execute_task(self):
113113
cloudpickle_register(ind=1)
114114
_execute_multiple_tasks(
115115
future_queue=q,
116+
return_queue=Queue(),
116117
cores=1,
117118
flux_executor=self.flux_executor,
118119
spawner=FluxPythonSpawner,
@@ -128,6 +129,7 @@ def test_execute_task_threads(self):
128129
cloudpickle_register(ind=1)
129130
_execute_multiple_tasks(
130131
future_queue=q,
132+
return_queue=Queue(),
131133
cores=1,
132134
threads_per_core=1,
133135
flux_executor=self.flux_executor,

0 commit comments

Comments
 (0)