code: def test_produces_correct_onehots(self):
      obs, _ = self.wrapped.reset()
      assert len(obs) == self.base_len + self.n_total_tasks
      assert (
          obs[-self.n_total_tasks :] == np.array([0, 1, 0, 0, 0])
      ).all()
  
file_path: garage/tests/garage/envs/test_task_onehot_wrapper.py
imports: import numpy
  from garage.envs import PointEnv, TaskOnehotWrapper
label_desc: Use image node_id 2 for calling the TestSingleWrappedEnv obj's underlying member method code with example usage: obj.test_produces_correct_onehots() without return types