code: def test_gym_continuous_environment(continuous_env):
      # observation space
      assert (
          type(continuous_env.state_space["observation"])
          == VectorObservationSpace
      )
      assert np.all(
          continuous_env.state_space["observation"].shape == [3]
      )
      assert np.all(
          continuous_env.last_env_response.next_state[
              "observation"
          ].shape
          == (3,)
      )
  
      # action space
      assert type(continuous_env.action_space) == BoxActionSpace
      assert np.all(continuous_env.action_space.shape == np.array([1]))
  
      # make sure that the seed is working properly
      assert (
          np.sum(
              continuous_env.last_env_response.next_state["observation"]
          )
          == 0.6118565010687202
      )
  
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 4 for calling a global function with example usage: test_gym_continuous_environment(continuous_env) without return types