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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
from pathlib import Path
import fpdf
import pytest
from fpdf.errors import FPDFException, FPDFUnicodeEncodingException
from fpdf.image_parsing import get_img_info
from PIL import Image
from test.conftest import assert_same_file
HERE = Path(__file__).resolve().parent
def test_add_page_throws_without_page():
pdf = fpdf.FPDF()
with pytest.raises(FPDFException) as e:
pdf.text(1, 2, "ok")
msg = "No page open, you need to call add_page() first"
assert str(e.value) == msg
def test_encoding_exception():
pdf = fpdf.FPDF()
pdf.add_page()
pdf.set_font("Helvetica", size=15)
with pytest.raises(FPDFUnicodeEncodingException) as error:
pdf.cell(text="Joséō")
# This should through an error since Helvetica is a latin-1 encoder and the ō is out of range.
msg = (
'Character "ō" at index 4 in text is outside the range of characters supported by the font '
'used: "helvetica". Please consider using a Unicode font.'
)
assert str(error.value) == msg
def test_orientation_portrait_landscape():
l = fpdf.FPDF(orientation="l")
landscape = fpdf.FPDF(orientation="landscape")
p = fpdf.FPDF(orientation="p")
portrait = fpdf.FPDF(orientation="portrait")
assert l.w_pt == landscape.w_pt
assert l.h_pt == landscape.h_pt
assert l.def_orientation == landscape.def_orientation
assert p.w_pt == portrait.w_pt
assert p.h_pt == portrait.h_pt
assert p.def_orientation == portrait.def_orientation
assert landscape.w_pt == portrait.h_pt
assert landscape.h_pt == portrait.w_pt
def test_incorrect_orientation():
with pytest.raises(ValueError) as error:
fpdf.FPDF(orientation="hello")
assert str(error.value) == "HELLO is not a valid PageOrientation"
def test_units():
with pytest.raises(ValueError) as e:
fpdf.FPDF(unit="smiles")
assert str(e.value) == "Incorrect unit: smiles"
assert fpdf.FPDF(unit="pt").k == pytest.approx(1)
assert fpdf.FPDF(unit="mm").k == pytest.approx(2.83464566929)
assert fpdf.FPDF(unit="cm").k == pytest.approx(28.3464566929)
assert fpdf.FPDF(unit="in").k == pytest.approx(72)
def test_doc_option_only_core_fonts_encoding():
pdf = fpdf.FPDF()
with pytest.warns(DeprecationWarning) as record:
pdf.set_doc_option("core_fonts_encoding", 4)
assert pdf.core_fonts_encoding == 4
with pytest.raises(FPDFException) as e:
pdf.set_doc_option("not core_fonts_encoding", None)
msg = 'Unknown document option "not core_fonts_encoding"'
assert str(e.value) == msg
for r in record:
assert_same_file(r.filename, __file__)
def test_adding_content_after_closing():
pdf = fpdf.FPDF()
pdf.set_font("helvetica", size=24)
pdf.add_page()
pdf.cell(w=pdf.epw, text="Hello fpdf2!", align="C")
pdf.output()
with pytest.raises(FPDFException) as error:
pdf.add_page()
assert (
str(error.value)
== "A page cannot be added on a closed document, after calling output()"
)
with pytest.raises(FPDFException) as error:
pdf.cell(w=pdf.epw, text="Hello again!", align="C")
assert (
str(error.value)
== "Content cannot be added on a finalized document, after calling output()"
)
def test_unsupported_image_filter_error():
image_filter = "N/A"
with pytest.raises(FPDFException) as error:
get_img_info(
HERE / "flowers.png",
Image.open(HERE / "flowers.png"),
image_filter=image_filter,
)
assert str(error.value) == f'Unsupported image filter: "{image_filter}"'
def test_incorrect_number_of_pages_toc():
pdf = fpdf.FPDF()
pdf.add_page()
pdf.insert_toc_placeholder(lambda a, b: None, 10)
with pytest.raises(FPDFException):
pdf.output()
def test_invalid_page_background():
pdf = fpdf.FPDF()
i = 0
with pytest.raises(TypeError) as error:
pdf.set_page_background(i)
msg = f"""background must be of type str, io.BytesIO, PIL.Image.Image, drawing.DeviceRGB, tuple or None
got: {type(i)}"""
assert str(error.value) == msg
def test_instantiating_fpdf_module(): # issue 683
# pylint: disable=import-outside-toplevel,not-callable,redefined-outer-name
from fpdf import fpdf
with pytest.raises(TypeError) as error:
fpdf()
assert str(error.value) == (
"You tried to instantied the fpdf module."
" You probably want to import the FPDF class instead:"
" from fpdf import FPDF"
)
|
