code: def test_registered_saver_checkpoint(self):
      p1 = Part([1, 4])
      p2 = Part([2, 5])
      p3 = Part([3, 6])
      s = Stack([p1, p2, p3])
      s2 = Stack([p3, p1, p2])
  
      expected_value_s = s.value()
      expected_value_s2 = s2.value()
  
      ckpt_path = os.path.join(self.get_temp_dir(), "ckpt")
      util.Checkpoint(s=s, s2=s2).write(ckpt_path)
  
      del s, s2, p1, p2, p3
  
      restore_s = Stack([Part([0, 0]) for _ in range(3)])
      util.Checkpoint(s=restore_s).read(ckpt_path).expect_partial()
      self.assertAllEqual(expected_value_s, restore_s.value())
      util.Checkpoint(s2=restore_s).read(ckpt_path).expect_partial()
      self.assertAllEqual(expected_value_s2, restore_s.value())
  
imports: import os
  import tempfile
  from absl.testing import parameterized
  from google.protobuf import wrappers_pb2
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.client import session
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import io_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import gfile
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import loader
  from tensorflow.python.saved_model import registration
  from tensorflow.python.saved_model import save
  from tensorflow.python.trackable import autotrackable
label_desc: Use image node_id 2 for calling the SingleCycleTest obj's underlying member method code with example usage: obj.test_registered_saver_checkpoint() without return types