code: def test_embed_image(self):
      image = np.zeros((100, 50, 3), dtype=np.uint8)
      (
          image_features,
          image_class_embeddings,
          pred_boxes,
      ) = self.model.embed_image(image)
  
      self.assertIsInstance(image_features, np.ndarray)
      self.assertIsInstance(image_class_embeddings, np.ndarray)
      self.assertIsInstance(pred_boxes, np.ndarray)
  
      chex.assert_shape(image_features, (self.num_instances, 768))
      chex.assert_shape(
          image_class_embeddings, (self.num_instances, 512)
      )
      chex.assert_shape(pred_boxes, (self.num_instances, 4))
  
file_path: scenic/scenic/projects/owl_vit/notebooks/tests/inference_test.py
imports: from unittest import mock
  from absl.testing import absltest
  import chex
  import jax
  import jax.numpy
  import numpy
  from scenic.projects.owl_vit import models
  from scenic.projects.owl_vit.configs import clip_b32
  from scenic.projects.owl_vit.notebooks import inference
label_desc: Use image node_id 4 for calling the InferenceTest obj's underlying member method code with example usage: obj.test_embed_image() without return types