def METHOD_NAME(self):
    self.assertEqual({
        'level_seed': TensorSpec(shape=(), dtype=torch.int32),
        'prev_level_complete': TensorSpec(shape=(), dtype=torch.uint8),
        'prev_level_seed': TensorSpec(shape=(), dtype=torch.int32)
    }, self._env.env_info_spec())
    self.assertEqual({
        'rgb':
            BoundedTensorSpec(
                shape=(3, 64, 64),
                dtype=torch.uint8,
                minimum=np.array(0),
                maximum=np.array(255))
    }, self._env.observation_spec())
    self.assertEqual(
        BoundedTensorSpec(
            shape=(),
            dtype=torch.int32,
            minimum=np.array(0),
            maximum=np.array(14)), self._env.action_spec())