tests/test_retrieval.py

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()