code: def test_sort_boxes(input_boxes, sorted_idxs):
      doc_builder = builder.DocumentBuilder()
      assert (
          doc_builder._sort_boxes(np.asarray(input_boxes))[0].tolist()
          == sorted_idxs
      )
  
file_path: doctr/tests/common/test_models_builder.py
imports: import numpy
  import pytest
  from doctr.file_utils import CLASS_NAME
  from doctr.io import Document
  from doctr.io.elements import KIEDocument
  from doctr.models import builder
label_desc: Use image node_id 3 for calling a global function with example usage: test_sort_boxes(input_boxes, sorted_idxs) without return types