code: def testSaveRestoreReshuffleDataset(self):
      dataset = dataset_ops.Dataset.range(10)
      dataset = dataset.shuffle(10, reshuffle_each_iteration=True)
      iterator = iter(dataset)
      ckpt = trackable_utils.Checkpoint(
          step=variables.Variable(0), iterator=iterator
      )
      manager = checkpoint_management.CheckpointManager(
          ckpt, self.get_temp_dir(), max_to_keep=3
      )
  
      iter1 = [next(iterator).numpy() for _ in range(5)]
  
      manager.save()
      iter2 = [next(iterator).numpy() for _ in range(5)]
  
      ckpt.restore(manager.latest_checkpoint)
      iter3 = [next(iterator).numpy() for _ in range(5)]
  
      self.assertNotEqual(iter1, iter2)
      self.assertCountEqual(iter2, iter3)
  
imports: import os
  from absl.testing import parameterized
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.checkpoint import checkpoint_management
  from tensorflow.python.checkpoint import checkpoint_options
  from tensorflow.python.data.experimental.ops import grouping
  from tensorflow.python.data.experimental.ops import interleave_ops
  from tensorflow.python.data.experimental.ops import scan_ops
  from tensorflow.python.data.experimental.ops import take_while_ops
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import options
  from tensorflow.python.eager import test
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import gen_dataset_ops
  from tensorflow.python.ops import io_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import parsing_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import script_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import gfile
label_desc: Use image node_id 14 for calling the CheckpointTest obj's underlying member method code with example usage: obj.testSaveRestoreReshuffleDataset() without return types