docstr: Tests save and restore.
      
code: def testRestore(self):
      """Tests save and restore."""
      fname = os.path.join(self.get_temp_dir(), "checkpoint")
      g = random.Generator.from_seed(1)
      cp = tracking_util.Checkpoint(g=g)
  
      def write_restore_compare():
          cp.write(fname)
          r1 = g.uniform([], dtype=dtypes.uint32, minval=None)
          cp.restore(fname)
          r2 = g.uniform([], dtype=dtypes.uint32, minval=None)
          self.assertAllEqual(r1, r2)
  
      # Run multiple times so that cp.write is called in various RNG states
      for _ in range(2):
          write_restore_compare()
  
imports: import os
  import re
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.kernel_tests.random import util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_random_ops
  from tensorflow.python.ops import gen_stateful_random_ops
  from tensorflow.python.ops import logging_ops
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.ops import stateful_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 36 for calling the StatefulRandomOpsTest obj's underlying member method code with example usage: obj.testRestore() without return types