Initial Commit

.github/workflows/test.yml

@@ -0,0 +1,25 @@
+name: TestBackend
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  test:
+    name: Run pytest
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: action/checkout@v2
+
+    - name: Install python to perform testing
+    - uses: actions/setup-python@v2
+      with:
+        python-version: '3.9'
+    
+    - name: Install Dependencies
+    - run: python install -r requirement.txt
+
+    - name: Run Pytest
+    - run: pytest

.gitignore

@@ -0,0 +1,2 @@
+bin
+__pycache__

models.py

@@ -0,0 +1,11 @@
+from pydantic import BaseModel
+
+class PredictReq(BaseModel):
+	medInc: float
+	houseAge: float
+	avgRooms: float
+	avgBdrms: float
+	population: float
+	avgOccup: float
+	latitude: float
+	longitude: float

server.py

@@ -0,0 +1,38 @@
+from fastapi import FastAPI
+from models import PredictReq
+import pickle
+import json
+import numpy as np
+
+app = FastAPI()
+
+ml_model = None
+# Load the XGBoost model
+with open('xgb_model.pkl', 'rb') as model_file:
+	ml_model = pickle.load(model_file)
+
+# This is just the normal getter api to check if working of the entire backend
+@app.get("/")
+def foo():
+	return {
+		"status": "House Price Prediction"
+	}
+
+@app.post("/predict_house_price")
+def predice_house_price(req: PredictReq):
+	inp = np.array([[
+				req.medInc,
+				req.houseAge,
+				req.avgRooms,
+				req.avgBdrms,
+				req.population,
+				req.avgOccup,
+				req.latitude,
+				req.longitude,
+			]])
+
+	prediction = ml_model.predict(inp)[0]
+
+	return {
+		'price': str(prediction)
+	}

test_server.py

@@ -0,0 +1,28 @@
+from fastapi.testclient import TestClient
+import json
+
+from server import app
+
+client = TestClient(app)
+
+def test_read_app():
+    response = client.get("/")
+    assert response.status_code == 200
+    assert response.json() == {"status": "House Price Prediction"}
+
+def test_predict_house_price():
+	response = client.post(
+			"/predict_house_price",
+			json={
+				'medInc': 8.3252,
+				'houseAge': 41,
+				'avgRooms': 6.0,
+				'avgBdrms': 1.0,
+				'population': 1.0,
+				'avgOccup': 2.5,
+				'latitude': 37.88,
+				'longitude': -122.23,
+			}
+		)
+
+	assert response.status_code == 200