fix(test_execution_checkpoint): new invocaton ids won't work across P…

…ython versions
ZettaAI · Jan 5, 2024 · 16ba21d · 16ba21d
1 parent 7af7735
commit 16ba21d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/assets/reference/task_checkpoint.zstd b/tests/assets/reference/task_checkpoint.zstd
diff --git a/tests/unit/mazepa/test_execution.py b/tests/unit/mazepa/test_execution.py
@@ -239,7 +239,13 @@ def test_local_execution_backup_write(reset_task_count, mocker):
     record_execution_checkpoint_m.assert_called()
 
 
-def test_local_execution_backup_read(reset_task_count):
+def test_local_execution_backup_read(reset_task_count, mocker):
+    task1 = dummy_task.make_task(argument="f1-x1")
+    mocker.patch(
+        "zetta_utils.mazepa.execution_state.read_execution_checkpoint",
+        return_value=set([task1.id_]),
+    )
+
     execute(
         concurrent_flow(
             [
@@ -251,7 +257,7 @@ def test_local_execution_backup_read(reset_task_count):
         batch_gap_sleep_sec=0,
         max_batch_len=2,
         do_dryrun_estimation=False,
-        checkpoint="tests/assets/reference/task_checkpoint.zstd",
+        checkpoint="MOCKED_PATH",
     )