def test_simple(self):
with TemporaryDirectory() as tmp:
booklets.generate_booklets(input_pdf=pdf_path,
target_directory=str(tmp))
split_result = analyze_split(tmp)
self.assertEqual(
split_result,
"""booklet01.pdf
1
2
10 3
4 9
8 5
6 7""",
)
def test_2booklets(self):
with TemporaryDirectory() as tmp:
booklets.generate_booklets(input_pdf=pdf_path,
target_directory=str(tmp),
max_size=8)
split_result = analyze_split(tmp)
self.assertEqual(
split_result,
"""booklet01.pdf
8 1
2 7
6 3
4 5
booklet02.pdf
9
10""",
)
def test_layout(self):
with TemporaryDirectory() as tmp:
booklets.generate_booklets(input_pdf=pdf_path,
target_directory=str(tmp),
layout=False)
split_result = analyze_split(tmp)
self.assertEqual(
split_result,
"""booklet01.pdf
1
2
3
4
5
6
7
8
9
10""",
)
def test_cover(self):
with TemporaryDirectory() as tmp:
booklets.generate_booklets(input_pdf=pdf_path,
target_directory=str(tmp),
max_size=8,
cover=True)
split_result = analyze_split(tmp)
self.assertEqual(
split_result,
"""booklet01.pdf
6
5
4 1
2 3
booklet02.pdf
7
8
9
10""",
)
def test_not_positive(self):
with TemporaryDirectory() as tmp:
booklets.generate_booklets(input_pdf=pdf_path,
target_directory=str(tmp),
max_size=0)