code: def testWrapTransitionKernel(self):
      class TestKernel(tfp.mcmc.TransitionKernel):
          def one_step(self, current_state, previous_kernel_results):
              return [
                  x + 1 for x in current_state
              ], previous_kernel_results + 1
  
          def bootstrap_results(self, current_state):
              return sum(current_state)
  
          def is_calibrated(self):
              return True
  
      def kernel(state, pkr):
          return util_tfp.transition_kernel_wrapper(
              state, pkr, TestKernel()
          )
  
      state = {"x": self._constant(0.0), "y": self._constant(1.0)}
      kr = 1.0
      (final_state, final_kr), _ = fun_mc.trace(
          (state, kr),
          kernel,
          2,
          trace_fn=lambda *args: (),
      )
      self.assertAllEqual(
          {"x": 2.0, "y": 3.0}, util.map_tree(np.array, final_state)
      )
      self.assertAllEqual(1.0 + 2.0, final_kr)
  
imports: from absl.testing import parameterized
  from jax import config
  import numpy
  import tensorflow.compat.v2
  from fun_mc import backend
  from fun_mc import fun_mc_lib
  from fun_mc import test_util
  from fun_mc import util_tfp
label_desc: Use image node_id 3 for calling the UtilTFPTest obj's underlying member method code with example usage: obj.testWrapTransitionKernel() and returns: util_tfp, unknown,  unknown, sum, True