code: def test_should_return_batches_equivalent_to_number_of_frames(self):
      from evadb.functions.decorators.yolo_object_detection_decorators import (
          Yolo,
      )
  
      frame_dog = {
          "id": 1,
          "data": self._load_image(
              os.path.join(self.base_path, "data", "dog.jpeg")
          ),
      }
      frame_dog_cat = {
          "id": 2,
          "data": self._load_image(
              os.path.join(self.base_path, "data", "dog_cat.jpg")
          ),
      }
      test_df_dog = pd.DataFrame([frame_dog])
      test_df_cat = pd.DataFrame([frame_dog_cat])
      frame_dog = numpy_to_yolo_format(test_df_dog["data"].values[0])
      frame_cat = numpy_to_yolo_format(test_df_cat["data"].values[0])
      detector = Yolo()
      result = []
      result.append(detector.forward(frame_dog))
      result.append(detector.forward(frame_cat))
  
      self.assertEqual(["dog"], result[0]["labels"].tolist()[0])
      self.assertEqual(["cat", "dog"], result[1]["labels"].tolist()[0])
  
file_path: evadb/test/integration_tests/long/functions/test_yolo_object_detector.py
imports: import os
  import sys
  import unittest
  import mock
  import numpy
  import pandas
  from evadb.utils.generic_utils import try_to_import_cv2, try_to_import_torch
label_desc: Use image node_id 4 for calling the YoloTest obj's underlying member method code with example usage: obj.test_should_return_batches_equivalent_to_number_of_frames() without return types