code: def test_basic_operation(self):
      kernel = test_fixtures.TestTransitionKernel()
      seed1, seed2 = samplers.split_seed(
          test_util.test_seed(sampler_type="stateless")
      )
      result = sample_chain_with_burnin(
          num_results=2, current_state=0.0, kernel=kernel, seed=seed1
      )
      samples = result.trace
      kernel_results = result.final_kernel_results
      self.assertAllClose([2], tensorshape_util.as_list(samples.shape))
  
      samples, kernel_results = self.evaluate([samples, kernel_results])
      self.assertAllClose([1, 2], samples)
      self.assertAllClose(2, kernel_results.counter_1)
      self.assertAllClose(4, kernel_results.counter_2)
  
      # Warm-restart the underlying kernel.  The Trace does not support warm
      # restart.
      result_2 = sample_chain_with_burnin(
          num_results=2, seed=seed2, **result.resume_kwargs
      )
      samples_2, kernel_results_2 = self.evaluate(
          [result_2.trace, result_2.final_kernel_results]
      )
      self.assertAllClose([3, 4], samples_2)
      self.assertAllClose(4, kernel_results_2.counter_1)
      self.assertAllClose(8, kernel_results_2.counter_2)
  
imports: import collections
  import warnings
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.experimental.mcmc import covariance_reducer
  from tensorflow_probability.python.experimental.mcmc import expectations_reducer
  from tensorflow_probability.python.experimental.mcmc.internal import test_fixtures
  from tensorflow_probability.python.experimental.mcmc.sample_fold import sample_chain_with_burnin
  from tensorflow_probability.python.experimental.mcmc.sample_fold import sample_fold
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import hmc
label_desc: Use image node_id 2 for calling the SampleChainTest obj's underlying member method code with example usage: obj.test_basic_operation() without return types