code: def test_crop_orientation_model(mock_text_box):
      text_box_0 = cv2.imread(mock_text_box)
      text_box_90 = np.rot90(text_box_0, 1)
      text_box_180 = np.rot90(text_box_0, 2)
      text_box_270 = np.rot90(text_box_0, 3)
      classifier = classification.crop_orientation_predictor(
          "mobilenet_v3_small_orientation", pretrained=True
      )
      assert classifier(
          [text_box_0, text_box_90, text_box_180, text_box_270]
      ) == [0, 1, 2, 3]
  
file_path: doctr/tests/tensorflow/test_models_classification_tf.py
imports: import os
  import tempfile
  import cv2
  import numpy
  import onnxruntime
  import psutil
  import pytest
  import tensorflow
  from doctr.models import classification
  from doctr.models.classification.predictor import CropOrientationPredictor
  from doctr.models.utils import export_model_to_onnx
label_desc: Use image node_id 4 for calling a global function with example usage: test_crop_orientation_model(mock_text_box) without return types