code: def test_new_time_step(sample_data):
      s = TimeStep(**sample_data)
      assert s.env_spec is sample_data["env_spec"]
      assert s.observation is sample_data["observation"]
      assert s.action is sample_data["action"]
      assert s.reward is sample_data["reward"]
      assert s.step_type is sample_data["step_type"]
      assert s.env_info is sample_data["env_info"]
      assert s.agent_info is sample_data["agent_info"]
      assert s.episode_info is sample_data["episode_info"]
      del s
  
      obs_space = akro.Box(
          low=-1, high=10, shape=(4, 3, 2), dtype=np.float32
      )
      act_space = akro.Box(
          low=-1, high=10, shape=(4, 2), dtype=np.float32
      )
      env_spec = EnvSpec(obs_space, act_space)
      sample_data["env_spec"] = env_spec
      obs_space = akro.Box(
          low=-1000, high=1000, shape=(4, 3, 2), dtype=np.float32
      )
      act_space = akro.Box(
          low=-1000, high=1000, shape=(4, 2), dtype=np.float32
      )
      sample_data["observation"] = obs_space.sample()
      sample_data["next_observation"] = obs_space.sample()
      sample_data["action"] = act_space.sample()
      s = TimeStep(**sample_data)
  
      assert s.observation is sample_data["observation"]
      assert s.next_observation is sample_data["next_observation"]
      assert s.action is sample_data["action"]
  
imports: import akro
  import gym.spaces
  import numpy
  import pytest
  from garage import EnvSpec, EnvStep, EpisodeBatch, StepType, TimeStep, TimeStepBatch
label_desc: Use image node_id 26 for calling a global function with example usage: test_new_time_step(sample_data) without return types