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