code: def test_new_eps(eps_data):
      t = EpisodeBatch(**eps_data)
      assert t.env_spec is eps_data["env_spec"]
      assert t.observations is eps_data["observations"]
      assert t.last_observations is eps_data["last_observations"]
      assert t.actions is eps_data["actions"]
      assert t.rewards is eps_data["rewards"]
      assert t.env_infos is eps_data["env_infos"]
      assert t.agent_infos is eps_data["agent_infos"]
      assert t.step_types is eps_data["step_types"]
      assert t.lengths is eps_data["lengths"]
      assert t.episode_infos_by_episode is eps_data["episode_infos"]
      assert (
          t.episode_infos["task_one_hot"][0].shape
          == eps_data["episode_infos"]["task_one_hot"][0].shape
      )
  
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 2 for calling a global function with example usage: test_new_eps(eps_data) without return types