code: def __init__(self):
      self.test_observation_spec = _OBSERVATION_SPEC
      self.test_observations = [
          {
              "RGB_INTERLEAVED": np.array(
                  [
                      [[255, 0, 0], [128, 0, 0], [0, 0, 255]],
                      [[0, 255, 0], [128, 0, 0], [0, 255, 0]],
                  ],
                  dtype=np.uint8,
              )
          },
          {
              "RGB_INTERLEAVED": np.array(
                  [[[0, 255, 0], [0, 128, 0]]], dtype=np.uint8
              )
          },
          {
              "RGB_INTERLEAVED": np.array(
                  [[[0, 0, 255], [0, 0, 128]]], dtype=np.uint8
              )
          },
      ]
      self.test_rewards = [0, 1, 2, 3]
      self._frame_index = 0
      self.last_actions = None
      self.last_steps = None
      self.events_return = None
      self.is_running_return = None
      self.action_spec_return = None
      self.reset_return = None
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import tempfile
  from absl.testing import absltest
  import numpy
  import six
  from PIL import Image
  from python.tests.utils import test_environment_decorator
label_desc: Use image node_id 1 to create a new EnvironmentStub object from inherited base classes: object with example: obj = EnvironmentStub()