code: def test_save_load_checkpoint(self, verify_fn: Callable[..., None]):
      test_snapshot = TestSnapshot()
      cluster = data_service_test_base.TestCluster(num_workers=1)
      dataset = dataset_ops.Dataset.range(10)
      self.evaluate(
          distributed_save_op.distributed_save(
              dataset, test_snapshot.path, cluster.dispatcher_address()
          )
      )
  
      def _build_ds() -> dataset_ops.Dataset:
          return load_op._load_distributed_snapshot_v2(
              test_snapshot.path
          )
  
      verify_fn(self, _build_ds, num_outputs=10)
  
imports: import os
  import shutil
  import tempfile
  import threading
  import time
  from typing import Callable, Optional
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.data.experimental.kernel_tests.service import test_base
  from tensorflow.python.data.experimental.ops import data_service_ops
  from tensorflow.python.data.experimental.ops import distributed_save_op
  from tensorflow.python.data.kernel_tests import checkpoint_test_base
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import load_op
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import errors
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import googletest
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the SaveLoadCheckpointTest obj's underlying member method code with example usage: obj.test_save_load_checkpoint(verify_fn) and returns: load_op