Unit Test Added

Lagstill · Jun 5, 2023 · 30fbed7 · 30fbed7
1 parent 967957a
commit 30fbed7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test_app.py b/test_app.py
@@ -0,0 +1,15 @@
+from flask import Flask, request, render_template
+from src.pipeline.predict_pipeline import CustomData, PredictPipeline
+import pytest
+
+from app import app
+
+@pytest.fixture
+def client():
+    with app.test_client() as client:
+        yield client
+
+def test_index(client):
+    response = client.get('/')
+    assert response.status_code == 200
+    assert b'Hello, World!' in response.data