def METHOD_NAME(self):
    self._env = suite_highway.load(environment_name='highway-v0')
    self.assertIsInstance(self._env, alf_environment.AlfEnvironment)
    self.assertEqual(torch.float32, self._env.observation_spec().dtype)
    actions = self._env.action_spec().sample()
    for _ in range(10):
        # unwrapped env (not thread_env or parallel_env) needs to convert
        # from tensor to array
        time_step = self._env.step(actions.cpu().numpy())