class_docstr: Adds test methods running standard checkpointing tests.
docstr: Test restoring from a golden checkpoint still works.
docstr_tok: ["Test","restoring","from","a","golden","checkpoint","still","works","."]
code: def test_restore_golden(self, golden):
      """Test restoring from a golden checkpoint still works."""
      module = golden.create_module()
      checkpoint = TestCheckpoint(golden=golden, module=module)
      variables = golden.create_all_variables(module)
      for variable in variables:
          variable.assign(tf.zeros_like(variable))
      checkpoint.restore_latest(assert_consumed=True)
      for variable in variables:
          self.assertAllEqual(
              variable.read_value(),
              goldens.range_like(variable),
              msg=variable.name,
          )
  
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 4 for calling the GoldenCheckpointsTest obj's underlying member method code with example usage: obj.test_restore_golden(golden) without return types