code: def test_seed_reproducibility(self):
      first_fake_kernel = test_fixtures.RandomTransitionKernel()
      second_fake_kernel = test_fixtures.RandomTransitionKernel()
      seed = samplers.sanitize_seed(test_util.test_seed())
      last_state_t = step_kernel(
          num_steps=1,
          current_state=0.0,
          kernel=test_fixtures.RandomTransitionKernel(),
          seed=seed,
      )
      for num_steps in range(2, 5):
          first_final_state_t = step_kernel(
              num_steps=num_steps,
              current_state=0.0,
              kernel=first_fake_kernel,
              seed=seed,
          )
          second_final_state_t = step_kernel(
              num_steps=num_steps,
              current_state=1.0,  # difference should be irrelevant
              kernel=second_fake_kernel,
              seed=seed,
          )
          (
              last_state,
              first_final_state,
              second_final_state,
          ) = self.evaluate(
              [last_state_t, first_final_state_t, second_final_state_t]
          )
          self.assertEqual(first_final_state, second_final_state)
          self.assertNotEqual(first_final_state, last_state)
          last_state_t = first_final_state_t
  
file_path: probability/tensorflow_probability/python/experimental/mcmc/step_test.py
imports: import collections
  import warnings
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.mcmc.internal import test_fixtures
  from tensorflow_probability.python.experimental.mcmc.step import step_kernel
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import hmc
  from tensorflow_probability.python.mcmc import sample
label_desc: Use image node_id 5 for calling the StepKernelTest obj's underlying member method code with example usage: obj.test_seed_reproducibility() without return types