fix: Test code (#3)

* first commit

* add model code

* add git ignore

* set up frontend+backend

* set up frontend+backend

* remove test

* test code

* fix test code

---------

Co-authored-by: Neang-Rothmny <neangrothmony@gmail.com>
Co-authored-by: lyheang <lyheang.ky@khmerenterprise.info>

Author: 3 people

backend/tests/conftest.py

@@ -20,22 +20,43 @@ def _install_fake_detector_module():
     """
     fake_detector = types.ModuleType("detector")
 
+    class _TensorLike:
+        """Mimics a tensor with tolist() method"""
+        def __init__(self, data):
+            self._data = data
+        
+        def tolist(self):
+            return self._data if isinstance(self._data, list) else [self._data]
+        
+        def __float__(self):
+            return float(self._data)
+        
+        def __int__(self):
+            return int(self._data)
+
     class _SeqWithToList:
         def __init__(self, data):
             self._data = data
 
         def tolist(self):
             return list(self._data)
+        
+        def __iter__(self):
+            # Wrap each item in _TensorLike so it has .tolist()
+            for item in self._data:
+                yield _TensorLike(item)
 
     class _FakeBoxes:
         def __init__(self):
-            # minimal shape similar to ultralytics result
-            self.xywh = _SeqWithToList([[10, 20, 30, 40]])
-            self.cls = _SeqWithToList([0])
+            # Match the new API: xyxy format (x1, y1, x2, y2), conf, cls
+            self.xyxy = _SeqWithToList([[10, 20, 40, 60]])  # x1, y1, x2, y2
+            self.conf = _SeqWithToList([0.95])  # confidence score
+            self.cls = _SeqWithToList([0])  # class id
 
     class _FakeResult:
         def __init__(self):
             self.boxes = _FakeBoxes()
+            self.names = {0: "ក"}  # Fake class names mapping
 
     class Detector:  # noqa: N801 - match name imported in app.py
         instance = None

backend/tests/test_detect.py

@@ -25,8 +25,18 @@ def test_detect_endpoint_returns_boxes_and_classes(client):
     assert resp.status_code == 200
     body = resp.json()
 
-    # From our fake detector we expect one box and one class id
-    assert "boxes" in body and isinstance(body["boxes"], list)
-    assert "classes" in body and isinstance(body["classes"], list)
-    assert body["boxes"] == [[10, 20, 30, 40]]
-    assert body["classes"] == [0]
+    # Check response structure matches new API
+    assert "width" in body and body["width"] == 50
+    assert "height" in body and body["height"] == 50
+    assert "detections" in body and isinstance(body["detections"], list)
+    assert "inference_ms" in body and isinstance(body["inference_ms"], (int, float))
+
+    # From our fake detector we expect one detection
+    assert len(body["detections"]) == 1
+    detection = body["detections"][0]
+    
+    assert "box" in detection
+    assert detection["box"] == {"x1": 10, "y1": 20, "x2": 40, "y2": 60}
+    assert detection["confidence"] == 0.95
+    assert detection["class_id"] == 0
+    assert detection["class_name"] == "ក"