bigocrpdf/tests/test_pdf_assembly.py at main · biglinux/bigocrpdf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
"""Tests for PDF text escaping and assembly utilities."""

import pikepdf

from bigocrpdf.services.rapidocr_service.pdf_assembly import (
    escape_pdf_text,
    set_root_page_layout,
)


class TestEscapePdfText:
    """Tests for escape_pdf_text."""

    def test_plain_ascii(self):
        assert escape_pdf_text("Hello World") == "Hello World"

    def test_escapes_backslash(self):
        assert escape_pdf_text("a\\b") == "a\\\\b"

    def test_escapes_parentheses(self):
        assert escape_pdf_text("(test)") == "\\(test\\)"

    def test_em_dash_replaced(self):
        assert escape_pdf_text("word\u2014word") == "word-word"

    def test_en_dash_replaced(self):
        assert escape_pdf_text("1\u20135") == "1-5"

    def test_smart_quotes_replaced(self):
        text = "\u201cHello\u201d \u2018world\u2019"
        assert escape_pdf_text(text) == "\"Hello\" 'world'"

    def test_ellipsis_replaced(self):
        assert escape_pdf_text("wait\u2026") == "wait..."

    def test_bullet_replaced(self):
        assert escape_pdf_text("\u2022 item") == "* item"

    def test_fi_ligature(self):
        assert escape_pdf_text("\ufb01nd") == "find"

    def test_fl_ligature(self):
        assert escape_pdf_text("\ufb02ow") == "flow"

    def test_zero_width_chars_removed(self):
        assert escape_pdf_text("a\u200bb\u200cc\ufeffd") == "abcd"

    def test_non_latin1_replaced_with_question(self):
        # Chinese character can't be encoded to latin-1
        assert escape_pdf_text("\u4e16") == "?"

    def test_combined_escaping(self):
        text = "(test\\) \u2014 \u201cquote\u201d"
        result = escape_pdf_text(text)
        assert "\\\\" in result
        assert "\\(" in result
        assert "\\)" in result
        assert "-" in result
        assert '"' in result

    def test_empty_string(self):
        assert escape_pdf_text("") == ""

    def test_only_latin1_passes(self):
        text = "\xe9\xe0\xfc"  # é à ü
        assert escape_pdf_text(text) == text


class TestSetRootPageLayout:
    """Tests for the viewer /PageLayout catalog setting."""

    @staticmethod
    def _blank_pdf() -> pikepdf.Pdf:
        pdf = pikepdf.Pdf.new()
        pdf.add_blank_page(page_size=(200, 300))
        return pdf

    def test_default_omits_page_layout(self):
        pdf = self._blank_pdf()
        assert set_root_page_layout(pdf, "default") is False
        assert "/PageLayout" not in pdf.Root

    def test_unknown_value_omits_page_layout(self):
        pdf = self._blank_pdf()
        assert set_root_page_layout(pdf, "garbage") is False
        assert "/PageLayout" not in pdf.Root

    def test_single_maps_to_singlepage(self):
        pdf = self._blank_pdf()
        assert set_root_page_layout(pdf, "single") is True
        assert str(pdf.Root.PageLayout) == "/SinglePage"

    def test_continuous_maps_to_onecolumn(self):
        pdf = self._blank_pdf()
        assert set_root_page_layout(pdf, "continuous") is True
        assert str(pdf.Root.PageLayout) == "/OneColumn"

    def test_two_page_maps_to_twocolumnleft(self):
        pdf = self._blank_pdf()
        assert set_root_page_layout(pdf, "two_page") is True
        assert str(pdf.Root.PageLayout) == "/TwoColumnLeft"