code: def testUnbatched(self, seed, seed_dtype, max_index, index_dtype):
      if max_index > 200:
          self.skipTest("Too slow in graph mode.")
      seen = (max_index + 1) * [False]
      seed = math_ops.cast(seed, seed_dtype)
      for index in range(max_index + 1):
          new_index = stateless.index_shuffle(
              math_ops.cast(index, index_dtype),
              seed,
              max_index=math_ops.cast(max_index, index_dtype),
          )
          self.assertEqual(new_index.dtype, index_dtype)
          new_index = self.evaluate(new_index)
          self.assertGreaterEqual(new_index, 0)
          self.assertLessEqual(new_index, max_index)
          self.assertFalse(seen[new_index])
          seen[new_index] = True
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/random/random_index_shuffle_test.py
imports: import itertools
  from absl.testing import parameterized
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_random_index_shuffle_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the StatelessOpsTest obj's underlying member method code with example usage: obj.testUnbatched(seed, seed_dtype, max_index, index_dtype) without return types