Create tests/test_retrieval.py

tests/test_retrieval.py

@@ -0,0 +1,17 @@
+import unittest
+from src.retrieval import RetrievalSystem
+
+class TestRetrieval(unittest.TestCase):
+    def setUp(self):
+        self.retrieval = RetrievalSystem("data/rupeia_document.json", "all-MiniLM-L6-v2")
+
+    def test_retrieval_sip(self):
+        context = self.retrieval.get_context("How do I start an SIP?")
+        self.assertIn("SIP allows regular investments", context)
+
+    def test_retrieval_empty(self):
+        context = self.retrieval.get_context("")
+        self.assertEqual(context, "")
+
+if __name__ == "__main__":
+    unittest.main()