code: def testShuffleV2InFunction(self):
      counter_var = variables.Variable(0)
  
      @def_function.function
      def consume():
          ds = dataset_ops.Dataset.range(10)
          ds = ds.shuffle(1)
          for _ in ds:
              counter_var.assign(counter_var + 1)
  
      consume()
      self.assertAllEqual(self.evaluate(counter_var), 10)
  
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 17 for calling the ShuffleTest obj's underlying member method code with example usage: obj.testShuffleV2InFunction() without return types