code: def test_flatten_observation(env_id):
      env = gym.make(env_id, disable_env_checker=True)
      wrapped_env = FlattenObservation(env)
  
      obs, info = env.reset()
      wrapped_obs, wrapped_obs_info = wrapped_env.reset()
  
      space = spaces.Tuple(
          (spaces.Discrete(32), spaces.Discrete(11), spaces.Discrete(2))
      )
      wrapped_space = spaces.Box(0, 1, [32 + 11 + 2], dtype=np.int64)
  
      assert space.contains(obs)
      assert wrapped_space.contains(wrapped_obs)
      assert isinstance(info, dict)
      assert isinstance(wrapped_obs_info, dict)
  
imports: import numpy
  import pytest
  import gym
  from gym import spaces
  from gym.wrappers import FlattenObservation
label_desc: Use image node_id 1 for calling a global function with example usage: test_flatten_observation(env_id) without return types