class_docstr: Wraps a tf.train.Checkpoint to make it more convenient for testing.
code: def __init__(self, golden=None, **kwargs):
      if golden is None:
          root = absltest.get_default_test_tmpdir()
      else:
          root = os.path.join(
              "sonnet/src/conformance/checkpoints/", golden.name
          )
      self._root = root
      self._prefix = os.path.join(self._root, "checkpoint")
      self._checkpoint = tf.train.Checkpoint(**kwargs)
  
imports: import os
  from absl import logging
  from absl.testing import absltest
  from absl.testing import parameterized
  from sonnet.src import test_utils
  from sonnet.src.conformance import goldens
  from sonnet.src.distribute import replicator
  from sonnet.src.distribute import replicator_test_utils
  import tensorflow
  import tree
label_desc: Use image node_id 1 to create a new TestCheckpoint object  with example: obj = TestCheckpoint(golden)