Add: unittests

weni-ai · Mar 22, 2024 · 9cc429b · 9cc429b
1 parent b14f2de
commit 9cc429b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/app/tests/test_document_loader.py b/app/tests/test_document_loader.py
@@ -146,6 +146,17 @@ def test_load_file_url_and_split_text(self):
         docs = load_file_url_and_split_text(file_path, file_type, self.text_splitter)
         self.assertEqual(list, type(docs))
 
+    def test_load_file_url_and_split_text_pdf_miner(self):
+        file_path = f'{self.path}/{self.file_name}.pdf'
+        file_type = "pdf"
+        docs = load_file_url_and_split_text(
+            file_path,
+            file_type,
+            self.text_splitter,
+            load_type="pdfminer"
+        )
+        self.assertEqual(list, type(docs))
+
     def test_load_file_url_and_get_pages_text(self):  # this function is deprecated
         file_path = f'{self.path}/{self.file_name}.pdf'
         file_type = "pdf"