add unit tests

Author: Pandede

tests/unit/wpodnet/backend/test_prediction.py

@@ -0,0 +1,17 @@
+import pytest
+
+from wpodnet import Prediction
+
+
+class TestPrediction:
+    def test_validation(self):
+        # Valid
+        Prediction(bounds=[(1, 2), (3, 4), (5, 6), (7, 8)], confidence=0.95)
+
+        # Invalid bounds length
+        with pytest.raises(ValueError):
+            Prediction(bounds=[(1, 2), (3, 4), (5, 6)], confidence=0.95)
+
+        # Invalid confidence value
+        with pytest.raises(ValueError):
+            Prediction(bounds=[(1, 2), (3, 4), (5, 6), (7, 8)], confidence=1.01)

wpodnet/backend.py

@@ -9,7 +9,7 @@
 from .model import WPODNet
 
 
-@dataclass
+@dataclass(frozen=True)
 class Prediction:
     """
     The prediction result from WPODNet.