Skip to content

Commit 5d057e8

Browse files
committed
fix test
1 parent 7fdbe1c commit 5d057e8

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

tests/unit/task_scheduler/file/test_backend.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def test_execute_function_mixed(self):
5050
backend_execute_task_in_file(file_name=file_name)
5151
future_obj = Future()
5252
_check_task_output(
53-
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory
53+
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory, duplicate_dict={},
5454
)
5555
self.assertTrue(future_obj.done())
5656
self.assertEqual(future_obj.result(), 3)
@@ -101,7 +101,7 @@ def test_execute_function_mixed_selector_convert(self):
101101
backend_execute_task_in_file(file_name=file_name_1)
102102
f1 = Future()
103103
_check_task_output(
104-
task_key=task_key_1, future_obj=f1, cache_directory=cache_directory
104+
task_key=task_key_1, future_obj=f1, cache_directory=cache_directory, duplicate_dict={},
105105
)
106106
task_key_2, data_dict = serialize_funct(
107107
fn=return_list,
@@ -113,7 +113,7 @@ def test_execute_function_mixed_selector_convert(self):
113113
backend_execute_task_in_file(file_name=file_name_2)
114114
f2 = Future()
115115
_check_task_output(
116-
task_key=task_key_2, future_obj=f2, cache_directory=cache_directory
116+
task_key=task_key_2, future_obj=f2, cache_directory=cache_directory, duplicate_dict={},
117117
)
118118
fs1 = FutureSelector(future=f1, selector="a")
119119
fs2 = FutureSelector(future=f2, selector=1)
@@ -143,7 +143,7 @@ def test_execute_function_args(self):
143143
backend_execute_task_in_file(file_name=file_name)
144144
future_obj = Future()
145145
_check_task_output(
146-
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory
146+
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory, duplicate_dict={},
147147
)
148148
self.assertTrue(future_obj.done())
149149
self.assertEqual(future_obj.result(), 3)
@@ -170,7 +170,7 @@ def test_execute_function_kwargs(self):
170170
backend_execute_task_in_file(file_name=file_name)
171171
future_obj = Future()
172172
_check_task_output(
173-
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory
173+
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory, duplicate_dict={},
174174
)
175175
self.assertTrue(future_obj.done())
176176
self.assertEqual(future_obj.result(), 3)
@@ -198,7 +198,7 @@ def test_execute_function_error(self):
198198
backend_execute_task_in_file(file_name=file_name)
199199
future_obj = Future()
200200
_check_task_output(
201-
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory
201+
task_key=task_key, future_obj=future_obj, cache_directory=cache_directory, duplicate_dict={},
202202
)
203203
self.assertTrue(future_obj.done())
204204
with self.assertRaises(ValueError):

0 commit comments

Comments
 (0)