code: def _test_infer_batch(batch):
      assert len(batch) == 1
  
      batch = batch[0]
      assert len(batch) == 2
  
      for x in batch:
          assert x.shape == (3, 375, 500)
          assert x.dtype == torch.float
  
imports: import pytest
  from icevision.all import *
  from icevision.models.torchvision import faster_rcnn, mask_rcnn, keypoint_rcnn
label_desc: Use image node_id 5 for calling a global function with example usage: _test_infer_batch(batch) without return types