👮 updating tests to simulate yolo8.1

tests/test_yolo.py

@@ -8,12 +8,13 @@
 
 class MockBoxes():
     def __init__(self, boxes):
-        self.boxes = torch.tensor(boxes)
+        self.xyxy = torch.tensor([boxes[0:4]])
+        self.cls = torch.tensor(boxes[-1])
 
 
 class MockImageResult():
     def __init__(self, boxes):
-        self.boxes = MockBoxes(boxes)
+        self.boxes = [MockBoxes(box) for box in boxes]
 
 
 class MockYoloOutput():
@@ -47,22 +48,22 @@ def __init__(self, *args):
     def setup_model(self, model):
         pass
 
+
 class MockYoloModel():
     predictor = None
     model = None
     names = [str(x) for x in range(4)]
 
     def predict(self, source, show=False, save=True, batch=4, imgsz=1280):
         for index, image in enumerate(source):
-            shutil.copy(image, self.predictor.save_dir/f"image{index}.jpg" )
+            shutil.copy(image, self.predictor.save_dir/image.name )
 
         return MockYoloOutput(source)
 
     def to(self, device):
         self.device = device
 
 
-
 @patch('hespi.yolo.DetectionPredictor', return_value=MockYoloPredictor())
 def test_yolo_output(mock_detection_predictor):
     base_dir = Path(__file__).parent/"testdata"