Update test_pipelines.py

test_pipelines.py

@@ -1,18 +1,17 @@
-import io
 from app import rag_bot
 
-def test_rag_with_pdf():
-    # Load a small dummy PDF for testing
-    sample_pdf_content = b"%PDF-1.4 sample minimal pdf"
-    pdf_file = io.BytesIO(sample_pdf_content)
-    pdf_file.name = "test.pdf"
+def test_rag_bot():
+    # Use a sample small PDF (must exist in repo for testing)
+    sample_pdf = "manual.pdf"  # Replace with a small test PDF
+    question = "What is the purpose of this document?"
 
     try:
-        answer = rag_bot("What is inside this document?", pdf_file)
+        answer = rag_bot(question, sample_pdf)
         assert isinstance(answer, str)
-        print("✅ Bot responded successfully:", answer[:200])
+        print("✅ Test Passed: Bot returned an answer")
+        print("Sample Answer:", answer[:200])
     except Exception as e:
-        print("❌ Test failed:", e)
+        print("❌ Test Failed:", e)
 
 if __name__ == "__main__":
-    test_rag_with_pdf()
+    test_rag_bot()