class_docstr: Wraps a tf.train.Checkpoint to make it more convenient for testing.
code: def restore_latest(self, assert_consumed):
      status = self._checkpoint.restore(
          tf.train.latest_checkpoint(self._root)
      )
      if assert_consumed:
          # Ensures that all values in the checkpoint have been consumed by some
          # checkpointable Python object.
          status.assert_consumed()
      return status
  
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 3 for calling the TestCheckpoint obj's underlying member method code with example usage: obj.restore_latest(assert_consumed) and returns: status