✅ all tests passed

Author: jmeisele

tests/test_service/test_models.py

@@ -0,0 +1,33 @@
+import pytest
+
+from fastapi_scaffolding.core import config
+from fastapi_scaffolding.models.payload import HousePredictionPayload
+from fastapi_scaffolding.models.prediction import HousePredictionResult
+from fastapi_scaffolding.services.models import HousePriceModel
+
+
+def test_prediction(test_client) -> None:
+    model_path = config.DEFAULT_MODEL_PATH
+    hpp = HousePredictionPayload.parse_obj({
+        "median_income_in_block": 8.3252,
+        "median_house_age_in_block": 41,
+        "average_rooms": 6,
+        "average_bedrooms": 1,
+        "population_per_block": 322,
+        "average_house_occupancy": 2.55,
+        "block_latitude": 37.88,
+        "block_longitude": -122.23
+    })
+
+    hpm = HousePriceModel(model_path)
+    result = hpm.predict(hpp)
+    assert isinstance(result, HousePredictionResult)
+
+
+def test_prediction_no_payload(test_client) -> None:
+    model_path = config.DEFAULT_MODEL_PATH
+
+    hpm = HousePriceModel(model_path)
+    with pytest.raises(ValueError):
+        result = hpm.predict(None)
+        assert isinstance(result, HousePredictionResult)