Skip to content

Commit 96e9cc4

Browse files
committed
fix tests
1 parent f24e444 commit 96e9cc4

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

tests/unit/executor/test_single_dependencies.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,7 @@ def test_dependency_steps(self):
116116
fs1 = Future()
117117
fs2 = Future()
118118
q = Queue()
119+
qr = Queue()
119120
q.put(
120121
{
121122
"fn": add_function,
@@ -150,6 +151,7 @@ def test_dependency_steps(self):
150151
target=_execute_tasks_with_dependencies,
151152
kwargs={
152153
"future_queue": q,
154+
"return_queue": qr,
153155
"executor_queue": executor._future_queue,
154156
"executor": executor,
155157
"refresh_rate": 0.01,
@@ -168,6 +170,7 @@ def test_dependency_steps_error(self):
168170
fs1 = Future()
169171
fs2 = Future()
170172
q = Queue()
173+
qr = Queue()
171174
q.put(
172175
{
173176
"fn": raise_error,
@@ -202,6 +205,7 @@ def test_dependency_steps_error(self):
202205
target=_execute_tasks_with_dependencies,
203206
kwargs={
204207
"future_queue": q,
208+
205209
"executor_queue": executor._future_queue,
206210
"executor": executor,
207211
"refresh_rate": 0.01,
@@ -222,6 +226,7 @@ def test_dependency_steps_error_before(self):
222226
fs1.set_exception(RuntimeError())
223227
fs2 = Future()
224228
q = Queue()
229+
qr = Queue()
225230
q.put(
226231
{
227232
"fn": add_function,
@@ -256,6 +261,7 @@ def test_dependency_steps_error_before(self):
256261
target=_execute_tasks_with_dependencies,
257262
kwargs={
258263
"future_queue": q,
264+
"return_queue": qr,
259265
"executor_queue": executor._future_queue,
260266
"executor": executor,
261267
"refresh_rate": 0.01,

0 commit comments

Comments
 (0)