Create tests/test_inference.py

tests/test_inference.py

@@ -0,0 +1,8 @@
+# tests/test_inference.py
+
+def test_inference():
+    frame = np.zeros((640, 640, 3), dtype=np.uint8)
+    model = YOLOv8Model()
+    result = model.predict(frame)
+    assert isinstance(result, pd.DataFrame), "Expected results in DataFrame format"
+    assert "name" in result.columns, "Violation class (name) missing in results"