def METHOD_NAME(self):
    self._env = suite_highway.load(environment_name="highway-v0")
    actions = self._env.action_spec().sample()
    time_step = self._env.step(actions.cpu().numpy())
    env_info = time_step.env_info
    for field in env_info:
        self.assertEqual(env_info[field].size, 1)
    self.assertFalse('action' in time_step.env_info.keys())