tests/test_pdf_parser.py

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