code: def testCheckpointLargeBuffer(self):
      if (
          pywrap_sanitizers.is_asan_enabled()
          or pywrap_sanitizers.is_tsan_enabled()
          or pywrap_sanitizers.is_msan_enabled()
      ):
          self.skipTest("Skip to avoid OOM when using sanitizers.")
      if sys.platform == "darwin":
          self.skipTest("Skip to avoid memory issues on mac.")
      dataset = dataset_ops.Dataset.range(12).batch(2)
      dataset = dataset.map(
          # Create tensors of size 512M.
          lambda seed: stateless_random_ops.stateless_random_uniform(
              (128, 1024, 1024), seed, dtype=dtypes.float32
          )
      )
      dataset = dataset.shuffle(buffer_size=6)
      iterator = iter(dataset)
      next(iterator)  # Request an element to fill the shuffle buffer
      ckpt = trackable_utils.Checkpoint(iterator=iterator)
      manager = checkpoint_management.CheckpointManager(
          ckpt, self.get_temp_dir(), max_to_keep=1
      )
      manager.save()
      del dataset
      del iterator
      manager.restore_or_initialize()
  
imports: import collections
  import functools
  import sys
  from absl.testing import parameterized
  import numpy
  from tensorflow.python import pywrap_sanitizers
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.checkpoint import checkpoint_management
  from tensorflow.python.data.experimental.ops import iterator_ops
  from tensorflow.python.data.experimental.ops import random_access
  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 options
  from tensorflow.python.eager import def_function
  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.framework import random_seed
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import check_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.training import saver
label_desc: Use image node_id 20 for calling the ShuffleTest obj's underlying member method code with example usage: obj.testCheckpointLargeBuffer() without return types