Skip to content

Commit f38f016

Browse files
authored
Add test for _convert_args_and_kwargs() (#742)
1 parent 69ccf00 commit f38f016

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

tests/test_cache_fileexecutor_serial.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
terminate_subprocess,
1212
)
1313
from executorlib.task_scheduler.file.task_scheduler import FileTaskScheduler, create_file_executor
14-
from executorlib.task_scheduler.file.shared import execute_tasks_h5
14+
from executorlib.task_scheduler.file.shared import execute_tasks_h5, _convert_args_and_kwargs
1515

1616
skip_h5py_test = False
1717
except ImportError:
@@ -234,5 +234,19 @@ def test_execute_in_subprocess_errors(self):
234234
backend="flux",
235235
)
236236

237+
def test_convert_args_and_kwargs(self):
238+
f1 = Future()
239+
f1.set_result(1)
240+
f2 = Future()
241+
f2.set_result(2)
242+
task_args, task_kwargs, future_wait_key_lst = _convert_args_and_kwargs(
243+
task_dict={"fn": 1, "args": (f1,), "kwargs": {"a": f2}},
244+
memory_dict={},
245+
file_name_dict={},
246+
)
247+
self.assertEqual(task_args, [1])
248+
self.assertEqual(task_kwargs, {"a": 2})
249+
self.assertTrue(len(future_wait_key_lst) == 0)
250+
237251
def tearDown(self):
238252
shutil.rmtree("executorlib_cache", ignore_errors=True)

0 commit comments

Comments
 (0)