From d571d38fb9ae68c2e64fbc3b1f312e2c4990fd51 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20Jan=C3=9Fen?= <janssen@mpie.de> Date: Sun, 27 Oct 2024 19:34:19 +0100 Subject: [PATCH] Cache fix test --- tests/test_cache_shared.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/tests/test_cache_shared.py b/tests/test_cache_shared.py index 993553cd..e1c26ca9 100644 --- a/tests/test_cache_shared.py +++ b/tests/test_cache_shared.py @@ -5,11 +5,8 @@ try: - from executorlib.cache.shared import ( - execute_task_in_file, - _check_task_output, - ) - from executorlib.cache.backend import FutureItem + from executorlib.cache.backend import backend_execute_task_in_file + from executorlib.cache.shared import _check_task_output, FutureItem from executorlib.standalone.hdf import dump from executorlib.standalone.serialize import serialize_funct_h5 @@ -36,7 +33,7 @@ def test_execute_function_mixed(self): ) file_name = os.path.join(cache_directory, task_key + ".h5in") dump(file_name=file_name, data_dict=data_dict) - execute_task_in_file(file_name=file_name) + backend_execute_task_in_file(file_name=file_name) future_obj = Future() _check_task_output( task_key=task_key, future_obj=future_obj, cache_directory=cache_directory @@ -59,7 +56,7 @@ def test_execute_function_args(self): ) file_name = os.path.join(cache_directory, task_key + ".h5in") dump(file_name=file_name, data_dict=data_dict) - execute_task_in_file(file_name=file_name) + backend_execute_task_in_file(file_name=file_name) future_obj = Future() _check_task_output( task_key=task_key, future_obj=future_obj, cache_directory=cache_directory @@ -82,7 +79,7 @@ def test_execute_function_kwargs(self): ) file_name = os.path.join(cache_directory, task_key + ".h5in") dump(file_name=file_name, data_dict=data_dict) - execute_task_in_file(file_name=file_name) + backend_execute_task_in_file(file_name=file_name) future_obj = Future() _check_task_output( task_key=task_key, future_obj=future_obj, cache_directory=cache_directory