code: def test_wrapped_cartpole_transition(self):
      cartpole_env = gym.spec("CartPole-v1").make()
      env = gym_wrapper.GymnasiumWrapper(cartpole_env)
      env.reset()
      transition_time_step = env.step(np.array(0, dtype=np.int32))
  
      self.assertTrue(transition_time_step.is_mid())
      self.assertNotEqual(None, transition_time_step.reward)
      self.assertEqual(1.0, transition_time_step.discount)
      self.assertEqual((4,), transition_time_step.observation.shape)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import math
  from absl.testing.absltest import mock
  import gymnasium
  import numpy
  from tf_agents.environments import gymnasium_wrapper
  from tf_agents.utils import test_utils
label_desc: Use image node_id 3 for calling the GymnasiumWrapperOnCartpoleTest obj's underlying member method code with example usage: obj.test_wrapped_cartpole_transition() without return types