backend/tests/ingest_documents.py

import unittest
import os
from ..data.preprocessing.ingest_documents import PDFProcessor

class TestIngestDocuments(unittest.TestCase):
    def test_ingest_documents(self):
        file = os.path.join(os.path.dirname(__file__), "../data", "combined_forms", "ACTC-Patient-Packet.pdf")
        print(file)
        processor = PDFProcessor(file)
        output = processor.extract_questions()
        output = [message.content for message in output]
        print('********output********', output)
        #results = ingest_documents.invoke({"text":"test"})
        #self.assertEqual(results, "test")
    def test_create_questions(self):
        file = os.path.join(os.path.dirname(__file__), "../data", "combined_forms", "ACTC-Patient-Packet.pdf")

if __name__ == "__main__":
    unittest.main()