| """Tests for file ingestion parsers.""" | |
| from __future__ import annotations | |
| import unittest | |
| from plane_mode_scholar.ingestion.pdf_parser import parse_file | |
| class PdfParserTests(unittest.TestCase): | |
| def test_demo_notes_txt(self) -> None: | |
| text, pages = parse_file("plane_mode_scholar/data/demo_notes.txt") | |
| self.assertTrue(text.strip()) | |
| self.assertEqual(len(pages), 1) | |
| def test_unsupported_suffix(self) -> None: | |
| text, pages = parse_file("readme.docx") | |
| self.assertEqual(text, "") | |
| self.assertEqual(pages, []) | |
| if __name__ == "__main__": | |
| unittest.main() | |