class_docstr: Tests for garage.envs.multi_env_wrapper
class_docstr_tok: ["Tests","for","garage.envs.multi_env_wrapper"]
docstr: test if output of step function is correct
docstr_tok: ["test","if","output","of","step","function","is","correct"]
code: def test_one_hot_observation(self):
      """test if output of step function is correct"""
      envs = ["CartPole-v0", "CartPole-v0"]
      mt_env = self._init_multi_env_wrapper(
          envs, sample_strategy=round_robin_strategy
      )
  
      obs, _ = mt_env.reset()
      assert (obs[-2:] == np.array([1.0, 0.0])).all()
      obs = mt_env.step(1).observation
      assert (obs[-2:] == np.array([1.0, 0.0])).all()
  
      obs, _ = mt_env.reset()
      assert (obs[-2:] == np.array([0.0, 1.0])).all()
      obs = mt_env.step(1).observation
      assert (obs[-2:] == np.array([0.0, 1.0])).all()
  
file_path: garage/tests/garage/envs/test_multi_env_wrapper.py
imports: import akro
  import numpy
  import pytest
  from garage.envs import GymEnv
  from garage.envs.multi_env_wrapper import MultiEnvWrapper, round_robin_strategy, uniform_random_strategy
label_desc: Use image node_id 13 for calling the TestMultiEnvWrapper obj's underlying member method code with example usage: obj.test_one_hot_observation() without return types