code: def test_gym_discrete_environment(atari_env):
      # observation space
      assert (
          type(atari_env.state_space["observation"])
          == ImageObservationSpace
      )
      assert np.all(
          atari_env.state_space["observation"].shape == [210, 160, 3]
      )
      assert np.all(
          atari_env.last_env_response.next_state["observation"].shape
          == (210, 160, 3)
      )
  
      # action space
      assert type(atari_env.action_space) == DiscreteActionSpace
      assert np.all(atari_env.action_space.high == 3)
  
      # make sure that the seed is working properly
      assert (
          np.sum(atari_env.last_env_response.next_state["observation"])
          == 4115856
      )
  
file_path: coach/rl_coach/tests/environments/test_gym_environment.py
imports: import os
  import sys
  import pytest
  from rl_coach.environments.gym_environment import GymEnvironment
  from rl_coach.base_parameters import VisualizationParameters
  import numpy
  from rl_coach.spaces import DiscreteActionSpace, BoxActionSpace, ImageObservationSpace, VectorObservationSpace
label_desc: Use image node_id 3 for calling a global function with example usage: test_gym_discrete_environment(atari_env) without return types