code: def test_time_limit_wrapper():
      """time limit wrapper unit tests"""
      env = build_dummy_maze_env()
      env = TimeLimitWrapper.wrap(env, max_episode_steps=5)
      env.set_max_episode_steps(max_episode_steps=5)
  
      env.seed(1234)
      env.reset()
      for i in range(5):
          obs, rew, done, info = env.step(env.action_space.sample())
          if i >= 4:
              assert done
      env.close()
  
file_path: maze/maze/test/core/wrappers/test_time_limit_wrapper.py
imports: from maze.core.wrappers.maze_gym_env_wrapper import GymMazeEnv
  from maze.core.wrappers.time_limit_wrapper import TimeLimitWrapper
  from maze.test.shared_test_utils.helper_functions import build_dummy_base_env, build_dummy_maze_env
  from maze.test.shared_test_utils.wrappers import assert_wrapper_clone_from
label_desc: Use image node_id 1 for calling a global function with example usage: test_time_limit_wrapper() without return types