code: def test_wrap_gym_render_kwargs(self):
      gym_env = gym.make("CartPole-v1")
      env = suite_gym.wrap_env(
          gym_env, render_kwargs={"width": 96, "height": 128}
      )
      gym_env.render = mock.MagicMock()
      # render_kwargs should be passed to them underlying gym env's render().
      env.render()
      gym_env.render.assert_called_with(
          "rgb_array", width=96, height=128
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  from absl.testing.absltest import mock
  import gin
  import gym
  from tf_agents.environments import py_environment
  from tf_agents.environments import suite_gym
  from tf_agents.environments import wrappers
  from tf_agents.utils import test_utils
label_desc: Use image node_id 9 for calling the SuiteGymTest obj's underlying member method code with example usage: obj.test_wrap_gym_render_kwargs() without return types