Create test_model.py

test_model.py

@@ -0,0 +1,18 @@
+import pytest
+from inference import predict
+from utils import preprocess_text
+
+def test_preprocessing():
+    text = "  Hello   World  "
+    processed = preprocess_text(text)
+    assert processed == "Hello World"
+
+def test_prediction():
+    # Test with simple input
+    result = predict("Test input")
+    assert isinstance(result, list)
+    print("Basic tests passed!")
+
+if __name__ == "__main__":
+    test_preprocessing()
+    test_prediction()