olmocr/tests/test_filter.py

import os
import unittest

from pypdf import PdfReader

from pdelfin.filter import PdfFilter


class PdfFilterTest(unittest.TestCase):
    def testFormLaterPages(self):
        self.filter = PdfFilter(apply_form_check=True)

        self.assertTrue(self.filter.filter_out_pdf(os.path.join(
            os.path.dirname(__file__),
            "gnarly_pdfs",
            "form_on_later_pages.pdf"
        )))

        self.filter = PdfFilter(apply_form_check=False)

        self.assertFalse(self.filter.filter_out_pdf(os.path.join(
            os.path.dirname(__file__),
            "gnarly_pdfs",
            "form_on_later_pages.pdf"
        )))