code: def test_from_env_step_time_step(sample_data):
      print(sample_data)
      agent_info = sample_data["agent_info"]
      last_observation = sample_data["observation"]
      observation = sample_data["next_observation"]
      episode_info = sample_data["episode_info"]
      time_step = TimeStep(**sample_data)
      del sample_data["agent_info"]
      del sample_data["next_observation"]
      del sample_data["episode_info"]
      sample_data["observation"] = observation
      env_step = EnvStep(**sample_data)
      time_step_new = TimeStep.from_env_step(
          env_step=env_step,
          last_observation=last_observation,
          agent_info=agent_info,
          episode_info=episode_info,
      )
      assert time_step == time_step_new
  
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 28 for calling a global function with example usage: test_from_env_step_time_step(sample_data) without return types