tests/test_checkpoint_restore.py: fixup

Commit 0dceab0 did not define the work_dir for one of the cases.

Fixes: 0dceab0 ("commit tests: add --work-path to criu test")
Signed-off-by: Kir Kolyshkin <[email protected]>

Author: kolyshkin

tests/test_checkpoint_restore.py

@@ -161,6 +161,7 @@ def _get_pre_dump_size(cr_dir):
 
     cid = None
     cr_dir = os.path.join(get_tests_root(), 'pre-dump')
+    work_dir = 'work-dir'
     try:
         _, cid = run_and_get_output(
             conf,
@@ -187,7 +188,6 @@ def _get_pre_dump_size(cr_dir):
 
         # Do the final dump. This dump should be much smaller.
         cr_dir = os.path.join(get_tests_root(), 'checkpoint')
-        work_dir = 'work-dir'
         run_crun_command([
             "checkpoint",
             "--parent-path=../pre-dump",