From 4be6c26de2a7eac9579076425c32af27710e551b Mon Sep 17 00:00:00 2001
From: Oleksandr Bezdieniezhnykh <oleksandr.bezdieniezhnykh@pwc.com>
Date: Sat, 28 Mar 2026 17:36:18 +0200
Subject: [PATCH] [AZ-171] Add CoreML inference test

Made-with: Cursor
---
 tests/test_export.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/test_export.py b/tests/test_export.py
index 1e6f58f..bb91e01 100644
--- a/tests/test_export.py
+++ b/tests/test_export.py
@@ -104,3 +104,19 @@ class TestCoremlExport:
         assert len(pkgs) >= 1
         model_file = pkgs[0] / "Data" / "com.apple.CoreML" / "model.mlmodel"
         assert model_file.exists()
+
+    def test_coreml_inference_produces_detections(self, exported_models):
+        # Arrange
+        pkgs = list(exported_models["model_dir"].glob("*.mlpackage"))
+        assert len(pkgs) >= 1
+        imgs = sorted(_DATASET_IMAGES.glob("*.jpg"))
+        if not imgs:
+            pytest.skip("no test images")
+        model = YOLO(str(pkgs[0]))
+
+        # Act
+        results = model.predict(source=str(imgs[0]), imgsz=c.config.export.onnx_imgsz, verbose=False)
+
+        # Assert
+        assert len(results) == 1
+        assert results[0].boxes is not None