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