code: def test_atari_preprocessing_grayscale(env, obs_shape):
      assert env.observation_space.shape == obs_shape
  
      # It is not possible to test the outputs as we are not using actual observations.
      # todo: update when ale-py is compatible with the ci
  
      env = StepAPICompatibility(
          env, output_truncation_bool=True
      )  # using compatibility wrapper since ale-py uses old step API
  
      obs, _ = env.reset(seed=0)
      assert obs in env.observation_space
  
      obs, _, _, _, _ = env.step(env.action_space.sample())
      assert obs in env.observation_space
  
      env.close()
  
file_path: gym/tests/wrappers/test_atari_preprocessing.py
imports: import numpy
  import pytest
  from gym.spaces import Box, Discrete
  from gym.wrappers import AtariPreprocessing, StepAPICompatibility
  from tests.testing_env import GenericTestEnv, old_step_fn
label_desc: Use image node_id 1 for calling a global function with example usage: test_atari_preprocessing_grayscale(env, obs_shape) without return types