code: def testSymbolicCheckpointSize(self):
      examples_per_flat_map = 100
      example_len = 10_000
  
      def flat_map_fn(_):
          data = []
          for _ in range(examples_per_flat_map):
              data.append(
                  stateless_random_ops.stateless_random_uniform(
                      [example_len], seed=(42, 42)
                  )
              )
          return dataset_ops.Dataset.from_tensor_slices(data)
  
      ds = dataset_ops.Dataset.range(10)
      # Inputs to flat_map are >1MB
      ds = ds.map(
          lambda x: stateless_random_ops.stateless_random_uniform(
              [1_000_000], seed=(42, 42)
          )
      )
      ds = ds.flat_map(flat_map_fn)
  
      options = options_lib.Options()
      options.experimental_symbolic_checkpoint = True
      ds = ds.with_options(options)
  
      it = ds.as_numpy_iterator()
      for _ in range(30):
          next(it)
  
      ckpt = it._save()
      # Make sure the checkpoint is smaller than the element sizes, i.e. no
      # elements are being stored in the checkpoint.
      self.assertLess(len(ckpt.numpy()), 10_000)
  
imports: import random
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.client import session
  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.framework import combinations
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import function
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import sparse_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import tensor_array_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops.ragged import ragged_conversion_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.platform import test
  from tensorflow.python.training import server_lib
label_desc: Use image node_id 9 for calling the FlatMapTest obj's underlying member method code with example usage: obj.testSymbolicCheckpointSize() and returns: dataset_ops