code: def test_new_ts_batch(batch_data):
      s = TimeStepBatch(**batch_data)
      assert s.episode_infos is batch_data["episode_infos"]
      assert s.env_spec is batch_data["env_spec"]
      assert s.observations is batch_data["observations"]
      assert s.next_observations is batch_data["next_observations"]
      assert s.actions is batch_data["actions"]
      assert s.rewards is batch_data["rewards"]
      assert s.env_infos is batch_data["env_infos"]
      assert s.agent_infos is batch_data["agent_infos"]
      assert s.step_types is batch_data["step_types"]
  
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 30 for calling a global function with example usage: test_new_ts_batch(batch_data) without return types